[Bug] Allow creating queues with preempt permission verb

Signed-off-by: JamesMurkin <[email protected]>
armadaproject · Jun 17, 2024 · 437045f · 437045f
1 parent 5530e50
commit 437045f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/pkg/client/queue/permission_verb.go b/pkg/client/queue/permission_verb.go
@@ -18,10 +18,10 @@ const (
 )
 
 // NewPermissionVerb returns PermissionVerb from input string. If input string doesn't match
-// one of allowed verb values ["submit", "cancel", "reprioritize", "watch"], and error is returned.
+// one of allowed verb values ["submit", "cancel", "preempt", "reprioritize", "watch"], and error is returned.
 func NewPermissionVerb(in string) (PermissionVerb, error) {
 	switch verb := PermissionVerb(in); verb {
-	case PermissionVerbSubmit, PermissionVerbCancel, PermissionVerbReprioritize, PermissionVerbWatch:
+	case PermissionVerbSubmit, PermissionVerbCancel, PermissionVerbPreempt, PermissionVerbReprioritize, PermissionVerbWatch:
 		return verb, nil
 	default:
 		return "", fmt.Errorf("invalid queue permission verb: %s", in)
@@ -76,6 +76,7 @@ func AllPermissionVerbs() PermissionVerbs {
 	return []PermissionVerb{
 		PermissionVerbSubmit,
 		PermissionVerbCancel,
+		PermissionVerbPreempt,
 		PermissionVerbReprioritize,
 		PermissionVerbWatch,
 	}

diff --git a/pkg/client/queue/permission_verb_test.go b/pkg/client/queue/permission_verb_test.go
@@ -14,6 +14,7 @@ func TestPermissionVerbUnmarshal(t *testing.T) {
 			Verbs: []PermissionVerb{
 				PermissionVerbCancel,
 				PermissionVerbReprioritize,
+				PermissionVerbPreempt,
 				PermissionVerbSubmit,
 				PermissionVerbWatch,
 			},