Function TestClaimPrebuild

enterprise/coderd/prebuilds/claim_test.go:85–390

(t *testing.T)

}

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

	const (
		desiredInstances = 1
		presetCount      = 2
	)

	unexpectedClaimingError := xerrors.New("unexpected claiming error")

	cases := map[string]struct {
		expectPrebuildClaimed  bool
		markPrebuildsClaimable bool
		// if claimingErr is not nil - error will be returned when ClaimPrebuiltWorkspace is called
		claimingErr error
	}{
		"no eligible prebuilds to claim": {
			expectPrebuildClaimed:  false,
			markPrebuildsClaimable: false,
		},
		"claiming an eligible prebuild should succeed": {
			expectPrebuildClaimed:  true,
			markPrebuildsClaimable: true,
		},
		"no claimable prebuilt workspaces error is returned": {
			expectPrebuildClaimed:  false,
			markPrebuildsClaimable: true,
			claimingErr:            agplprebuilds.ErrNoClaimablePrebuiltWorkspaces,
		},
		"AGPL does not support prebuilds error is returned": {
			expectPrebuildClaimed:  false,
			markPrebuildsClaimable: true,
			claimingErr:            agplprebuilds.ErrAGPLDoesNotSupportPrebuiltWorkspaces,
		},
		"unexpected claiming error is returned": {
			expectPrebuildClaimed:  false,
			markPrebuildsClaimable: true,
			claimingErr:            unexpectedClaimingError,
		},
	}

	for name, tc := range cases {
		// Ensure that prebuilt workspaces can be claimed in non-default organizations:
		for _, useDefaultOrg := range []bool{true, false} {
			t.Run(name, func(t *testing.T) {
				t.Parallel()

				// Setup
				clock := quartz.NewMock(t)
				clock.Set(dbtime.Now())
				ctx := testutil.Context(t, testutil.WaitSuperLong)
				db, pubsub := dbtestutil.NewDB(t)

				spy := newStoreSpy(db, tc.claimingErr)
				expectedPrebuildsCount := desiredInstances * presetCount

				logger := testutil.Logger(t)
				client, _, api, owner := coderdenttest.NewWithAPI(t, &coderdenttest.Options{