add read yam

pulumitest/pulumiYAML.go

@@ -0,0 +1,37 @@
+package pulumitest
+
+import (
+	"os"
+	"path/filepath"
+	"strings"
+)
+
+// YAML does not allow tabs, so this function will error if the program contains tabs.
+func (a *PulumiTest) WritePulumiYaml(t PT, program string) {
+	t.Helper()
+
+	// find the line of the program that contains tabs
+	lines := strings.Split(program, "\n")
+	for i, line := range lines {
+		if strings.Contains(line, "\t") {
+			ptFatalF(t, "program contains tabs on line %d: %s\nTry replacing it with:\n%s", i+1, line, strings.ReplaceAll(program, "\t", "    "))
+		}
+	}
+
+	pulumiYamlPath := filepath.Join(a.CurrentStack().Workspace().WorkDir(), "Pulumi.yaml")
+	err := os.WriteFile(pulumiYamlPath, []byte(program), 0o600)
+	if err != nil {
+		ptFatalF(t, "failed to replace program %s", err)
+	}
+}
+
+func (a *PulumiTest) ReadPulumiYaml(t PT) string {
+	t.Helper()
+
+	pulumiYamlPath := filepath.Join(a.CurrentStack().Workspace().WorkDir(), "Pulumi.yaml")
+	program, err := os.ReadFile(pulumiYamlPath)
+	if err != nil {
+		ptFatalF(t, "failed to read program %s", err)
+	}
+	return string(program)
+}

pulumitest/writePulumiYAML_test.go → pulumitest/pulumiYAML_test.go

@@ -12,14 +12,29 @@ func TestReplaceProgram(t *testing.T) {
 
 	test := pulumitest.NewPulumiTest(t, "testdata/yaml_empty")
 
-	// Note the forbidden tab character in the program
 	test.WritePulumiYaml(t, `
 name: yaml_empty
 runtime: yaml
 outputs:
-	output: "output"`)
+    output: "output"`)
 
 	res := test.Up(t)
 
 	require.Equal(t, "output", res.Outputs["output"].Value)
 }
+
+func TestReadProgram(t *testing.T) {
+	t.Parallel()
+
+	test := pulumitest.NewPulumiTest(t, "testdata/yaml_empty")
+
+	program := `
+name: yaml_prog
+runtime: yaml
+outputs:
+	output: "output"`
+
+	test.WritePulumiYaml(t, program)
+
+	require.Equal(t, program, test.ReadPulumiYaml(t))
+}