Function TestWorkspaceBuildUsageChecker

coderd/wsbuilder/wsbuilder_test.go:1053–1164

(t *testing.T)

}

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

	t.Run("Permitted", func(t *testing.T) {
		t.Parallel()

		ctx, cancel := context.WithCancel(context.Background())
		defer cancel()

		var calls atomic.Int64
		fakeUsageChecker := &fakeUsageChecker{
			checkBuildUsageFunc: func(_ context.Context, _ database.Store, _ *database.TemplateVersion, _ *database.Task, _ database.WorkspaceTransition) (wsbuilder.UsageCheckResponse, error) {
				calls.Add(1)
				return wsbuilder.UsageCheckResponse{Permitted: true}, nil
			},
		}

		mDB := expectDB(t,
			// Inputs
			withTemplate,
			withInactiveVersion(nil),
			withLastBuildFound,
			withLastBuildState,
			withTemplateVersionVariables(inactiveVersionID, nil),
			withRichParameters(nil),
			withParameterSchemas(inactiveJobID, nil),
			withWorkspaceTags(inactiveVersionID, nil),
			withProvisionerDaemons([]database.GetEligibleProvisionerDaemonsByProvisionerJobIDsRow{}),

			// Outputs
			expectProvisionerJob(func(job database.InsertProvisionerJobParams) {}),
			withInTx,
			expectFindMatchingPresetID(uuid.Nil, sql.ErrNoRows),
			expectBuild(func(bld database.InsertWorkspaceBuildParams) {}),
			withBuild,
			withNoTask,
			expectBuildParameters(func(params database.InsertWorkspaceBuildParametersParams) {}),
		)
		fc := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})

		ws := database.Workspace{ID: workspaceID, TemplateID: templateID, OwnerID: userID}
		uut := wsbuilder.New(ws, database.WorkspaceTransitionStart, fakeUsageChecker)
		// nolint: dogsled
		_, _, _, err := uut.Build(ctx, mDB, fc, nil, audit.WorkspaceBuildBaggage{})
		require.NoError(t, err)
		require.EqualValues(t, 1, calls.Load())
	})

	// The failure cases are mostly identical from a test perspective.
	const message = "fake test message"
	cases := []struct {
		name        string
		response    wsbuilder.UsageCheckResponse
		responseErr error
		assertions  func(t *testing.T, err error)
	}{
		{
			name: "NotPermitted",