fixes

Author: mkondratek

vscode/test/e2e/auth.test.ts

@@ -102,7 +102,6 @@ test
     'logs out the user when userShouldUseEnterprise is set to true',
     async ({ page, sidebar, server }) => {
         await sidebarSignin(page, sidebar, { skipAssertions: true })
-        await server.setUserShouldUseEnterprise(true)
         await expectSignInPage(page)
         await expect(
             page

vscode/test/e2e/helpers.ts

@@ -161,11 +161,8 @@ export const test = base
     })
     .extend<{ server: MockServer }>({
         server: [
-            async ({ shouldUseEnterprise }, use) => {
+            async (_, use) => {
                 MockServer.run(async server => {
-                    if (shouldUseEnterprise !== undefined) {
-                        server.setUserShouldUseEnterprise(shouldUseEnterprise)
-                    }
                     await use(server)
                 })
             },

vscode/test/fixtures/mock-server.ts

@@ -176,7 +176,6 @@ class GraphQlMock {
 export class MockServer {
     graphQlMocks: Map<string, GraphQlMock> = new Map();
     availableLLMs: ServerModelConfiguration | undefined = getServerSentModelsMock();
-    userShouldUseEnterprise: boolean = false;
     featureFlags: Map<string, boolean> = new Map([]);
 
     constructor(public readonly express: express.Express) {}
@@ -190,10 +189,6 @@ export class MockServer {
         return mock;
     }
 
-    public setUserShouldUseEnterprise(value: boolean) {
-        this.userShouldUseEnterprise = value;
-    }
-
     public setAvailableLLMs(config: ServerModelConfiguration) {
         this.availableLLMs = config;
     }
@@ -416,7 +411,7 @@ export class MockServer {
                     attribution,
                     // When server-sent LLMs have been set, we enable the models api
                     modelsAPIEnabled: !!controller.availableLLMs,
-                    userShouldUseEnterprise: controller.userShouldUseEnterprise,
+                    userShouldUseEnterprise: true,
                 }),
             );
         });