Tidy up test initialisation

- Extract pulumiTestInit method so we're not copying the same code into 3 places.
- Extract testContext method to create context with deadline if required.

pulumitest/convert.go

@@ -48,12 +48,7 @@ func (a *PulumiTest) Convert(language string, opts ...opttest.Option) ConvertRes
 		source:  targetDir,
 		options: options,
 	}
-	if !options.SkipInstall {
-		convertedTest.Install()
-	}
-	if !options.SkipStackCreate {
-		convertedTest.NewStack(options.StackName, options.NewStackOpts...)
-	}
+	pulumiTestInit(convertedTest, options)
 	return ConvertResult{
 		PulumiTest: convertedTest,
 		Output:     string(out),

pulumitest/copy.go

@@ -48,12 +48,7 @@ func (a *PulumiTest) CopyTo(dir string, opts ...opttest.Option) *PulumiTest {
 		source:  dir,
 		options: options,
 	}
-	if !options.SkipInstall {
-		newTest.Install()
-	}
-	if !options.SkipStackCreate {
-		newTest.NewStack(options.StackName, options.NewStackOpts...)
-	}
+	pulumiTestInit(newTest, options)
 	return newTest
 }
 

pulumitest/pulumiTest.go

@@ -23,14 +23,7 @@ type PulumiTest struct {
 // 3. Create a new stack called "test" with state stored to a local temporary directory and a fixed passphrase for encryption.
 func NewPulumiTest(t *testing.T, source string, opts ...opttest.Option) *PulumiTest {
 	t.Helper()
-	var ctx context.Context
-	var cancel context.CancelFunc
-	if deadline, ok := t.Deadline(); ok {
-		ctx, cancel = context.WithDeadline(context.Background(), deadline)
-	} else {
-		ctx, cancel = context.WithCancel(context.Background())
-	}
-	t.Cleanup(cancel)
+	ctx := testContext(t)
 	options := opttest.DefaultOptions()
 	for _, opt := range opts {
 		opt.Apply(options)
@@ -43,14 +36,34 @@ func NewPulumiTest(t *testing.T, source string, opts ...opttest.Option) *PulumiT
 	}
 	if !options.TestInPlace {
 		pt = pt.CopyToTempDir()
+	} else {
+		pulumiTestInit(pt, options)
+	}
+	return pt
+}
+
+func testContext(t *testing.T) context.Context {
+	t.Helper()
+	var ctx context.Context
+	var cancel context.CancelFunc
+	if deadline, ok := t.Deadline(); ok {
+		ctx, cancel = context.WithDeadline(context.Background(), deadline)
+	} else {
+		ctx, cancel = context.WithCancel(context.Background())
 	}
+	t.Cleanup(cancel)
+	return ctx
+}
+
+// Perform the common initialization steps for a PulumiTest instance.
+func pulumiTestInit(pt *PulumiTest, options *opttest.Options) {
+	pt.t.Helper()
 	if !options.SkipInstall {
 		pt.Install()
 	}
 	if !options.SkipStackCreate {
 		pt.NewStack(options.StackName, options.NewStackOpts...)
 	}
-	return pt
 }
 
 // Source returns the current source directory.