mirror of
https://github.com/coder/coder.git
synced 2026-09-24 15:04:27 +08:00
refactor: use specific error for agpl and prebuilds (#17591)
Follow-up PR to https://github.com/coder/coder/pull/17458 Addresses this discussion: https://github.com/coder/coder/pull/17458#discussion_r2055940797
This commit is contained in:
@@ -111,6 +111,11 @@ func TestClaimPrebuild(t *testing.T) {
|
||||
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,
|
||||
@@ -224,8 +229,11 @@ func TestClaimPrebuild(t *testing.T) {
|
||||
TemplateVersionPresetID: presets[0].ID,
|
||||
})
|
||||
|
||||
isNoPrebuiltWorkspaces := errors.Is(tc.claimingErr, agplprebuilds.ErrNoClaimablePrebuiltWorkspaces)
|
||||
isUnsupported := errors.Is(tc.claimingErr, agplprebuilds.ErrAGPLDoesNotSupportPrebuiltWorkspaces)
|
||||
|
||||
switch {
|
||||
case tc.claimingErr != nil && errors.Is(tc.claimingErr, agplprebuilds.ErrNoClaimablePrebuiltWorkspaces):
|
||||
case tc.claimingErr != nil && (isNoPrebuiltWorkspaces || isUnsupported):
|
||||
require.NoError(t, err)
|
||||
coderdtest.AwaitWorkspaceBuildJobCompleted(t, userClient, userWorkspace.LatestBuild.ID)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user