update unit tests

Signed-off-by: Joshua Li <[email protected]>

src/plugins/query_enhancements/server/routes/index.test.ts

@@ -0,0 +1,21 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+import { coerceStatusCode } from '.';
+
+describe('coerceStatusCode', () => {
+  it('should return 503 when input is 500', () => {
+    expect(coerceStatusCode(500)).toBe(503);
+  });
+
+  it('should return the input status code when it is not 500', () => {
+    expect(coerceStatusCode(404)).toBe(404);
+  });
+
+  it('should return 503 when input is undefined or null', () => {
+    expect(coerceStatusCode((undefined as unknown) as number)).toBe(503);
+    expect(coerceStatusCode((null as unknown) as number)).toBe(503);
+  });
+});

src/plugins/query_enhancements/server/routes/index.ts

@@ -16,6 +16,15 @@ import { API } from '../../common';
 import { registerQueryAssistRoutes } from './query_assist';
 import { registerDataSourceConnectionsRoutes } from './data_source_connection';
 
+/**
+ * Coerce status code to 503 for 500 errors from dependency services. Only use
+ * this function to handle errors throw by other services, and not from OSD.
+ */
+export const coerceStatusCode = (statusCode: number) => {
+  if (statusCode === 500) return 503;
+  return statusCode || 503;
+};
+
 /**
  * @experimental
  *
@@ -91,9 +100,8 @@ export function defineSearchStrategyRouteProvider(logger: Logger, router: IRoute
           } catch (e) {
             error = err;
           }
-          const statusCode = error.status || err.status;
           return res.custom({
-            statusCode: statusCode === 500 ? 503 : statusCode || 503,
+            statusCode: coerceStatusCode(error.status || err.status),
             body: err.message,
           });
         }