updating test utils

elastic · Dec 8, 2020 · 4cb05a0 · 4cb05a0
1 parent c8b8dfe
commit 4cb05a0
Showing 1 changed file with 33 additions and 22 deletions.
diff --git a/src/core/server/http/test_utils.ts b/src/core/server/http/test_utils.ts
@@ -32,28 +32,39 @@ const env = Env.createDefault(REPO_ROOT, getEnvOptions());
 const logger = loggingSystemMock.create();
 
 const configService = configServiceMock.create();
-configService.atPath.mockReturnValue(
-  new BehaviorSubject({
-    hosts: ['localhost'],
-    maxPayload: new ByteSizeValue(1024),
-    autoListen: true,
-    ssl: {
-      enabled: false,
-    },
-    compression: { enabled: true },
-    xsrf: {
-      disableProtection: true,
-      allowlist: [],
-    },
-    customResponseHeaders: {},
-    requestId: {
-      allowFromAnyIp: true,
-      ipAllowlist: [],
-    },
-    keepaliveTimeout: 120_000,
-    socketTimeout: 120_000,
-  } as any)
-);
+configService.atPath.mockImplementation((path) => {
+  if (path === 'server') {
+    return new BehaviorSubject({
+      hosts: ['localhost'],
+      maxPayload: new ByteSizeValue(1024),
+      autoListen: true,
+      ssl: {
+        enabled: false,
+      },
+      compression: { enabled: true },
+      xsrf: {
+        disableProtection: true,
+        allowlist: [],
+      },
+      customResponseHeaders: {},
+      requestId: {
+        allowFromAnyIp: true,
+        ipAllowlist: [],
+      },
+      keepaliveTimeout: 120_000,
+      socketTimeout: 120_000,
+    } as any);
+  }
+  if (path === 'externalUrl') {
+    return new BehaviorSubject({
+      policy: [],
+    } as any);
+  }
+  if (path === 'csp') {
+    return new BehaviorSubject({} as any);
+  }
+  throw new Error(`Unexpected config path: ${path}`);
+});
 
 const defaultContext: CoreContext = {
   coreId,