Add test to verify dormancy in lifecycle executor

BrunoQuaresma · BrunoQuaresma · commit aea55aa01c26 · 2024-07-22T20:34:10.000Z
diff --git a/coderd/autobuild/lifecycle_executor_test.go b/coderd/autobuild/lifecycle_executor_test.go
@@ -3,6 +3,7 @@ package autobuild_test
 import (
 	"context"
 	"os"
+	"sync/atomic"
 	"testing"
 	"time"
 
@@ -25,6 +26,8 @@ import (
 	"github.com/coder/coder/v2/provisioner/echo"
 	"github.com/coder/coder/v2/provisionersdk/proto"
 	"github.com/coder/coder/v2/testutil"
+
+	enterpriseSchedule "github.com/coder/coder/v2/enterprise/coderd/schedule"
 )
 
 func TestExecutorAutostartOK(t *testing.T) {
@@ -1062,6 +1065,52 @@ func TestExecutorInactiveWorkspace(t *testing.T) {
 	})
 }
 
+func TestNotifications(t *testing.T) {
+	t.Parallel()
+
+	t.Run("Dormancy", func(t *testing.T) {
+		t.Parallel()
+
+		// Setup template with dormancy and create a workspace with it
+		var (
+			ticker                = make(chan time.Time)
+			statCh                = make(chan autobuild.Stats)
+			logger                = slogtest.Make(t, &slogtest.Options{})
+			notificationsEnqueuer = testutil.FakeNotificationsEnqueuer{}
+			client                = coderdtest.New(t, &coderdtest.Options{
+				AutobuildTicker:          ticker,
+				AutobuildStats:           statCh,
+				IncludeProvisionerDaemon: true,
+				NotificationsEnqueuer:    &notificationsEnqueuer,
+				TemplateScheduleStore:    enterpriseSchedule.NewEnterpriseTemplateScheduleStore(userQuietHoursScheduleStore(), &notificationsEnqueuer, logger),
+			})
+			admin          = coderdtest.CreateFirstUser(t, client)
+			version        = coderdtest.CreateTemplateVersion(t, client, admin.OrganizationID, nil)
+			timeTilDormant = 1000
+		)
+
+		coderdtest.AwaitTemplateVersionJobCompleted(t, client, version.ID)
+		template := coderdtest.CreateTemplate(t, client, admin.OrganizationID, version.ID, func(ctr *codersdk.CreateTemplateRequest) {
+			ctr.TimeTilDormantMillis = ptr.Ref(int64(timeTilDormant))
+		})
+		userClient, _ := coderdtest.CreateAnotherUser(t, client, admin.OrganizationID)
+		workspace := coderdtest.CreateWorkspace(t, userClient, admin.OrganizationID, template.ID)
+		coderdtest.AwaitWorkspaceBuildJobCompleted(t, userClient, workspace.LatestBuild.ID)
+
+		// Stop workspace
+		workspace = coderdtest.MustTransitionWorkspace(t, client, workspace.ID, database.WorkspaceTransitionStart, database.WorkspaceTransitionStop)
+		build := coderdtest.AwaitWorkspaceBuildJobCompleted(t, userClient, workspace.LatestBuild.ID)
+
+		// Wait for workspace to become dormant
+		ticker <- build.Job.CompletedAt.Add(time.Millisecond * time.Duration(timeTilDormant) * 2)
+		<-statCh
+
+		// Check that the workspace is dormant
+		workspace = coderdtest.MustWorkspace(t, client, workspace.ID)
+		require.NotNil(t, workspace.DormantAt)
+	})
+}
+
 func mustProvisionWorkspace(t *testing.T, client *codersdk.Client, mut ...func(*codersdk.CreateWorkspaceRequest)) codersdk.Workspace {
 	t.Helper()
 	user := coderdtest.CreateFirstUser(t, client)
@@ -1113,3 +1162,10 @@ func mustWorkspaceParameters(t *testing.T, client *codersdk.Client, workspaceID
 func TestMain(m *testing.M) {
 	goleak.VerifyTestMain(m)
 }
+
+func userQuietHoursScheduleStore() *atomic.Pointer[schedule.UserQuietHoursScheduleStore] {
+	store := schedule.NewAGPLUserQuietHoursScheduleStore()
+	p := &atomic.Pointer[schedule.UserQuietHoursScheduleStore]{}
+	p.Store(&store)
+	return p
+}
