Fix refresh on agent pools

CHANGELOG_PENDING.md

@@ -5,5 +5,6 @@
 
 - Fixes a bug where the `Environment` resource would print its contents on error. [#390](https://github.com/pulumi/pulumi-pulumiservice/pull/390)
 - Fixes a bug where the `Environment` resource would error if a FileAsset was used. [#391](https://github.com/pulumi/pulumi-pulumiservice/pull/391)
+- Fixes a bug where the `AgentPool` resource lost some outputs on `refresh`. [#395](https://github.com/pulumi/pulumi-pulumiservice/pull/395)
 
 ### Miscellaneous

examples/examples_yaml_test.go

@@ -277,6 +277,20 @@ func TestYamlEnvironmentsExample(t *testing.T) {
 	})
 }
 
+func TestYamlAgentPoolsExample(t *testing.T) {
+	cwd := getCwd(t)
+	digits := generateRandomFiveDigits()
+	integration.ProgramTest(t, &integration.ProgramTestOptions{
+		Dir: path.Join(cwd, ".", "yaml-agent-pools"),
+		Config: map[string]string{
+			"digits": digits,
+		},
+		PrepareProject: func(p *engine.Projinfo) error {
+			return nil
+		},
+	})
+}
+
 func TestYamlTemplateSourcesExample(t *testing.T) {
 	cwd := getCwd(t)
 	integration.ProgramTest(t, &integration.ProgramTestOptions{

examples/yaml-agent-pools/Pulumi.yaml

@@ -7,7 +7,7 @@ resources:
     type: pulumiservice:index:AgentPool
     properties:
       organizationName: service-provider-test-org
-      name: test-agent-pool
+      name: test-agent-pool-${digits}
       description: Test agent pool
 
 outputs:

provider/pkg/provider/agent_pool.go

@@ -217,6 +217,14 @@ func (ap *PulumiServiceAgentPoolResource) Read(req *pulumirpc.ReadRequest) (*pul
 		return &pulumirpc.ReadResponse{}, nil
 	}
 
+	propertyMap, err := plugin.UnmarshalProperties(req.GetProperties(), plugin.MarshalOptions{KeepSecrets: true})
+	if err != nil {
+		return nil, err
+	}
+	if propertyMap["tokenValue"].HasValue() {
+		agentPool.TokenValue = getSecretOrStringValue(propertyMap["tokenValue"])
+	}
+
 	input := PulumiServiceAgentPoolInput{
 		OrgName:     orgName,
 		Description: agentPool.Description,