Function TestWorkspaceProvisionerdServerMetrics

enterprise/coderd/workspaces_test.go:3018–3170

(t *testing.T)

}

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

	// Setup
	clock := quartz.NewMock(t)
	ctx := testutil.Context(t, testutil.WaitSuperLong)
	db, pb := dbtestutil.NewDB(t, dbtestutil.WithDumpOnFailure())
	logger := testutil.Logger(t)
	reg := prometheus.NewRegistry()
	provisionerdserverMetrics := provisionerdserver.NewMetrics(logger)
	err := provisionerdserverMetrics.Register(reg)
	require.NoError(t, err)
	client, _, api, owner := coderdenttest.NewWithAPI(t, &coderdenttest.Options{
		Options: &coderdtest.Options{
			Database:                  db,
			Pubsub:                    pb,
			IncludeProvisionerDaemon:  true,
			Clock:                     clock,
			ProvisionerdServerMetrics: provisionerdserverMetrics,
		},
	})

	// Setup Prebuild reconciler
	cache := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})
	reconciler := prebuilds.NewStoreReconciler(
		db, pb, cache,
		codersdk.PrebuildsConfig{},
		logger,
		clock,
		prometheus.NewRegistry(),
		notifications.NewNoopEnqueuer(),
		api.AGPL.BuildUsageChecker,
		noop.NewTracerProvider(),
		10,
		nil,
	)
	var claimer agplprebuilds.Claimer = prebuilds.NewEnterpriseClaimer()
	api.AGPL.PrebuildsClaimer.Store(&claimer)

	organizationName, err := client.Organization(ctx, owner.OrganizationID)
	require.NoError(t, err)
	userClient, user := coderdtest.CreateAnotherUser(t, client, owner.OrganizationID, rbac.RoleMember())

	// Setup template and template version with a preset with 1 prebuild instance
	versionPrebuild := coderdtest.CreateTemplateVersion(t, client, owner.OrganizationID, templateWithAgentAndPresetsWithPrebuilds(1))
	coderdtest.AwaitTemplateVersionJobCompleted(t, client, versionPrebuild.ID)
	templatePrebuild := coderdtest.CreateTemplate(t, client, owner.OrganizationID, versionPrebuild.ID)
	presetsPrebuild, err := client.TemplateVersionPresets(ctx, versionPrebuild.ID)
	require.NoError(t, err)
	require.Len(t, presetsPrebuild, 1)

	// Setup template and template version with a preset without prebuild instances
	versionNoPrebuild := coderdtest.CreateTemplateVersion(t, client, owner.OrganizationID, templateWithAgentAndPresetsWithPrebuilds(0))
	coderdtest.AwaitTemplateVersionJobCompleted(t, client, versionNoPrebuild.ID)
	templateNoPrebuild := coderdtest.CreateTemplate(t, client, owner.OrganizationID, versionNoPrebuild.ID)
	presetsNoPrebuild, err := client.TemplateVersionPresets(ctx, versionNoPrebuild.ID)
	require.NoError(t, err)
	require.Len(t, presetsNoPrebuild, 1)