Function TestRunner_Cleanup

scaletest/taskstatus/run_internal_test.go:656–707

(t *testing.T)

}

func TestRunner_Cleanup(t *testing.T) {
	t.Parallel()

	ctx := testutil.Context(t, testutil.WaitMedium)

	fakeClient := &fakeClientWithCleanupTracking{
		fakeClient:           newFakeClient(t),
		deleteWorkspaceCalls: make([]uuid.UUID, 0),
	}
	fakeClient.initialize(slog.Make(sloghuman.Sink(testutil.NewTestLogWriter(t))).Leveled(slog.LevelDebug))

	cfg := Config{
		AppSlug:              "test-app",
		TemplateID:           uuid.UUID{5, 6, 7, 8},
		WorkspaceName:        "test-workspace",
		MetricLabelValues:    []string{"test"},
		Metrics:              NewMetrics(prometheus.NewRegistry(), "test"),
		ReportStatusPeriod:   100 * time.Millisecond,
		ReportStatusDuration: 200 * time.Millisecond,
		StartReporting:       make(chan struct{}),
		ConnectedWaitGroup:   &sync.WaitGroup{},
	}

	runner := &Runner{
		client:      fakeClient,
		updater:     newFakeAppStatusUpdater(t),
		cfg:         cfg,
		clock:       quartz.NewMock(t),
		randFloat64: func() float64 { return 0.5 }, // not random in tests
	}

	logWriter := testutil.NewTestLogWriter(t)

	// Case 1: No workspace created - Cleanup should do nothing
	err := runner.Cleanup(ctx, "test-runner", logWriter)
	require.NoError(t, err)
	require.Len(t, fakeClient.deleteWorkspaceCalls, 0, "deleteWorkspace should not be called when no workspace was created")

	// Case 2: Workspace created - Cleanup should delete it
	runner.workspaceID = uuid.UUID{1, 2, 3, 4}
	err = runner.Cleanup(ctx, "test-runner", logWriter)
	require.NoError(t, err)
	require.Len(t, fakeClient.deleteWorkspaceCalls, 1, "deleteWorkspace should be called once")
	require.Equal(t, runner.workspaceID, fakeClient.deleteWorkspaceCalls[0], "deleteWorkspace should be called with correct workspace ID")

	// Case 3: Cleanup with error
	fakeClient.deleteError = xerrors.New("delete failed")
	runner.workspaceID = uuid.UUID{5, 6, 7, 8}
	err = runner.Cleanup(ctx, "test-runner", logWriter)
	require.Error(t, err)
	require.Contains(t, err.Error(), "delete external workspace")
}

// fakeClientWithCleanupTracking extends fakeClient to track deleteWorkspace calls
type fakeClientWithCleanupTracking struct {