enttest options

f0ssel · f0ssel · commit f031b10b491e · 2022-09-27T20:49:12.000Z
diff --git a/enterprise/coderd/coderdenttest/coderdenttest.go b/enterprise/coderd/coderdenttest/coderdenttest.go
@@ -82,14 +82,15 @@ func NewWithAPI(t *testing.T, options *Options) (*codersdk.Client, io.Closer, *c
 }
 
 type LicenseOptions struct {
-	AccountType string
-	AccountID   string
-	GraceAt     time.Time
-	ExpiresAt   time.Time
-	UserLimit   int64
-	AuditLog    bool
-	BrowserOnly bool
-	SCIM        bool
+	AccountType    string
+	AccountID      string
+	GraceAt        time.Time
+	ExpiresAt      time.Time
+	UserLimit      int64
+	AuditLog       bool
+	BrowserOnly    bool
+	SCIM           bool
+	WorkspaceQuota bool
 }
 
 // AddLicense generates a new license with the options provided and inserts it.
@@ -121,6 +122,10 @@ func GenerateLicense(t *testing.T, options LicenseOptions) string {
 	if options.SCIM {
 		scim = 1
 	}
+	workspaceQuota := int64(0)
+	if options.WorkspaceQuota {
+		workspaceQuota = 1
+	}
 
 	c := &coderd.Claims{
 		RegisteredClaims: jwt.RegisteredClaims{
@@ -134,10 +139,11 @@ func GenerateLicense(t *testing.T, options LicenseOptions) string {
 		AccountID:      options.AccountID,
 		Version:        coderd.CurrentVersion,
 		Features: coderd.Features{
-			UserLimit:   options.UserLimit,
-			AuditLog:    auditLog,
-			BrowserOnly: browserOnly,
-			SCIM:        scim,
+			UserLimit:      options.UserLimit,
+			AuditLog:       auditLog,
+			BrowserOnly:    browserOnly,
+			SCIM:           scim,
+			WorkspaceQuota: workspaceQuota,
 		},
 	}
 	tok := jwt.NewWithClaims(jwt.SigningMethodEdDSA, c)
