Update XService

presleyp · presleyp · commit aced91f82caf · 2022-08-24T19:41:07.000Z
diff --git a/site/src/testHelpers/entities.ts b/site/src/testHelpers/entities.ts
@@ -647,11 +647,15 @@ export const MockEntitlementsWithWarnings: TypesGen.Entitlements = {
   warnings: ["You are over your active user limit.", "And another thing."],
   has_license: true,
   features: {
-    activeUsers: {
+    user_limit: {
       enabled: true,
-      entitlement: "entitled",
+      entitlement: "grace_period",
       limit: 100,
       actual: 102,
     },
+    audit_log: {
+      enabled: true,
+      entitlement: "entitled",
+    }
   },
 }
diff --git a/site/src/xServices/entitlements/entitlementsXService.ts b/site/src/xServices/entitlements/entitlementsXService.ts
@@ -47,7 +47,7 @@ export const entitlementsMachine = createMachine(
         on: {
           GET_ENTITLEMENTS: "gettingEntitlements",
           SHOW_MOCK_BANNER: { actions: "assignMockEntitlements" },
-          HIDE_MOCK_BANNER: { actions: "clearMockEntitlements" },
+          HIDE_MOCK_BANNER: "gettingEntitlements"
         },
       },
       gettingEntitlements: {
@@ -81,9 +81,6 @@ export const entitlementsMachine = createMachine(
       assignMockEntitlements: assign({
         entitlements: (_) => MockEntitlementsWithWarnings,
       }),
-      clearMockEntitlements: assign({
-        entitlements: (_) => emptyEntitlements,
-      }),
     },
     services: {
       getEntitlements: () => API.getEntitlements(),
