@@ -20,7 +20,6 @@ import (
2020"github.com/coder/coder/v2/coderd/database/dbgen"
2121"github.com/coder/coder/v2/coderd/database/dbtestutil"
2222"github.com/coder/coder/v2/coderd/database/dbtime"
23- agplprebuilds"github.com/coder/coder/v2/coderd/prebuilds"
2423"github.com/coder/coder/v2/codersdk"
2524"github.com/coder/coder/v2/enterprise/coderd/prebuilds"
2625"github.com/coder/coder/v2/testutil"
@@ -55,8 +54,8 @@ func TestMetricsCollector(t *testing.T) {
5554name :"prebuild provisioned but not completed" ,
5655transitions :allTransitions ,
5756jobStatuses :allJobStatusesExcept (database .ProvisionerJobStatusPending ,database .ProvisionerJobStatusRunning ,database .ProvisionerJobStatusCanceling ),
58- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
59- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID },
57+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
58+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID },
6059metrics : []metricCheck {
6160{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
6261{prebuilds .MetricClaimedCount ,ptr .To (0.0 ),true },
@@ -72,8 +71,8 @@ func TestMetricsCollector(t *testing.T) {
7271name :"prebuild running" ,
7372transitions : []database.WorkspaceTransition {database .WorkspaceTransitionStart },
7473jobStatuses : []database.ProvisionerJobStatus {database .ProvisionerJobStatusSucceeded },
75- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
76- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID },
74+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
75+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID },
7776metrics : []metricCheck {
7877{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
7978{prebuilds .MetricClaimedCount ,ptr .To (0.0 ),true },
@@ -89,8 +88,8 @@ func TestMetricsCollector(t *testing.T) {
8988name :"prebuild failed" ,
9089transitions :allTransitions ,
9190jobStatuses : []database.ProvisionerJobStatus {database .ProvisionerJobStatusFailed },
92- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
93- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID ,uuid .New ()},
91+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
92+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID ,uuid .New ()},
9493metrics : []metricCheck {
9594{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
9695{prebuilds .MetricFailedCount ,ptr .To (1.0 ),true },
@@ -105,8 +104,8 @@ func TestMetricsCollector(t *testing.T) {
105104name :"prebuild eligible" ,
106105transitions : []database.WorkspaceTransition {database .WorkspaceTransitionStart },
107106jobStatuses : []database.ProvisionerJobStatus {database .ProvisionerJobStatusSucceeded },
108- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
109- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID },
107+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
108+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID },
110109metrics : []metricCheck {
111110{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
112111{prebuilds .MetricClaimedCount ,ptr .To (0.0 ),true },
@@ -122,8 +121,8 @@ func TestMetricsCollector(t *testing.T) {
122121name :"prebuild ineligible" ,
123122transitions :allTransitions ,
124123jobStatuses :allJobStatusesExcept (database .ProvisionerJobStatusSucceeded ),
125- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
126- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID },
124+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
125+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID },
127126metrics : []metricCheck {
128127{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
129128{prebuilds .MetricClaimedCount ,ptr .To (0.0 ),true },
@@ -139,7 +138,7 @@ func TestMetricsCollector(t *testing.T) {
139138name :"prebuild claimed" ,
140139transitions :allTransitions ,
141140jobStatuses :allJobStatuses ,
142- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
141+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
143142ownerIDs : []uuid.UUID {uuid .New ()},
144143metrics : []metricCheck {
145144{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
@@ -169,8 +168,8 @@ func TestMetricsCollector(t *testing.T) {
169168name :"deleted templates should not be included in exported metrics" ,
170169transitions :allTransitions ,
171170jobStatuses :allJobStatuses ,
172- initiatorIDs : []uuid.UUID {agplprebuilds . SystemUserID },
173- ownerIDs : []uuid.UUID {agplprebuilds . SystemUserID ,uuid .New ()},
171+ initiatorIDs : []uuid.UUID {database . PrebuildsSystemUserID },
172+ ownerIDs : []uuid.UUID {database . PrebuildsSystemUserID ,uuid .New ()},
174173metrics :nil ,
175174templateDeleted : []bool {true },
176175eligible : []bool {false },
@@ -209,7 +208,7 @@ func TestMetricsCollector(t *testing.T) {
209208reconciler := prebuilds .NewStoreReconciler (db ,pubsub , codersdk.PrebuildsConfig {},logger ,quartz .NewMock (t ),prometheus .NewRegistry (),newNoopEnqueuer ())
210209ctx := testutil .Context (t ,testutil .WaitLong )
211210
212- createdUsers := []uuid.UUID {agplprebuilds . SystemUserID }
211+ createdUsers := []uuid.UUID {database . PrebuildsSystemUserID }
213212for _ ,user := range slices .Concat (test .ownerIDs ,test .initiatorIDs ) {
214213if ! slices .Contains (createdUsers ,user ) {
215214dbgen .User (t ,db , database.User {
@@ -327,8 +326,8 @@ func TestMetricsCollector_DuplicateTemplateNames(t *testing.T) {
327326test := testCase {
328327transition :database .WorkspaceTransitionStart ,
329328jobStatus :database .ProvisionerJobStatusSucceeded ,
330- initiatorID :agplprebuilds . SystemUserID ,
331- ownerID :agplprebuilds . SystemUserID ,
329+ initiatorID :database . PrebuildsSystemUserID ,
330+ ownerID :database . PrebuildsSystemUserID ,
332331metrics : []metricCheck {
333332{prebuilds .MetricCreatedCount ,ptr .To (1.0 ),true },
334333{prebuilds .MetricClaimedCount ,ptr .To (0.0 ),true },