fix tests

f0ssel · f0ssel · commit 65a20a9957ad · 2022-09-29T18:36:39.000Z
diff --git a/enterprise/cli/features_test.go b/enterprise/cli/features_test.go
@@ -57,7 +57,7 @@ func TestFeaturesList(t *testing.T) {
 		var entitlements codersdk.Entitlements
 		err := json.Unmarshal(buf.Bytes(), &entitlements)
 		require.NoError(t, err, "unmarshal JSON output")
-		assert.Len(t, entitlements.Features, 3)
+		assert.Len(t, entitlements.Features, 4)
 		assert.Empty(t, entitlements.Warnings)
 		assert.Equal(t, codersdk.EntitlementNotEntitled,
 			entitlements.Features[codersdk.FeatureUserLimit].Entitlement)
diff --git a/enterprise/coderd/licenses_test.go b/enterprise/coderd/licenses_test.go
@@ -98,18 +98,20 @@ func TestGetLicense(t *testing.T) {
 		assert.Equal(t, int32(1), licenses[0].ID)
 		assert.Equal(t, "testing", licenses[0].Claims["account_id"])
 		assert.Equal(t, map[string]interface{}{
-			codersdk.FeatureUserLimit:   json.Number("0"),
-			codersdk.FeatureAuditLog:    json.Number("1"),
-			codersdk.FeatureSCIM:        json.Number("1"),
-			codersdk.FeatureBrowserOnly: json.Number("1"),
+			codersdk.FeatureUserLimit:      json.Number("0"),
+			codersdk.FeatureAuditLog:       json.Number("1"),
+			codersdk.FeatureSCIM:           json.Number("1"),
+			codersdk.FeatureBrowserOnly:    json.Number("1"),
+			codersdk.FeatureWorkspaceQuota: json.Number("0"),
 		}, licenses[0].Claims["features"])
 		assert.Equal(t, int32(2), licenses[1].ID)
 		assert.Equal(t, "testing2", licenses[1].Claims["account_id"])
 		assert.Equal(t, map[string]interface{}{
-			codersdk.FeatureUserLimit:   json.Number("200"),
-			codersdk.FeatureAuditLog:    json.Number("1"),
-			codersdk.FeatureSCIM:        json.Number("1"),
-			codersdk.FeatureBrowserOnly: json.Number("1"),
+			codersdk.FeatureUserLimit:      json.Number("200"),
+			codersdk.FeatureAuditLog:       json.Number("1"),
+			codersdk.FeatureSCIM:           json.Number("1"),
+			codersdk.FeatureBrowserOnly:    json.Number("1"),
+			codersdk.FeatureWorkspaceQuota: json.Number("0"),
 		}, licenses[1].Claims["features"])
 	})
 }
