From 6d86381c744e991d8bf9ac9dbb1f53b41ff7d83c Mon Sep 17 00:00:00 2001
From: sumukhswamy <sumukhhs@amazon.com>
Date: Wed, 15 May 2024 20:01:03 -0700
Subject: [PATCH] added fix for runAsync query without mds id

Signed-off-by: sumukhswamy <sumukhhs@amazon.com>
---
 public/components/Main/__snapshots__/main.test.tsx.snap | 4 ++++
 server/routes/query.ts                                  | 2 +-
 2 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/public/components/Main/__snapshots__/main.test.tsx.snap b/public/components/Main/__snapshots__/main.test.tsx.snap
index ec09cff..80b5592 100644
--- a/public/components/Main/__snapshots__/main.test.tsx.snap
+++ b/public/components/Main/__snapshots__/main.test.tsx.snap
@@ -1407,6 +1407,8 @@ exports[`<Main /> spec click run button, and response is ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="SQL"
+                    title="SQL"
                   >
                     <input
                       checked=""
@@ -1431,6 +1433,8 @@ exports[`<Main /> spec click run button, and response is ok 1`] = `
                 >
                   <span
                     class="euiButton__text euiButtonGroupButton__textShift"
+                    data-text="PPL"
+                    title="PPL"
                   >
                     <input
                       class="euiScreenReaderOnly"
diff --git a/server/routes/query.ts b/server/routes/query.ts
index 2daa3ae..fcfbe0e 100644
--- a/server/routes/query.ts
+++ b/server/routes/query.ts
@@ -186,7 +186,7 @@ export default function query(server: IRouter, service: QueryService, openSearch
 
   server.get(
     {
-      path: ROUTE_PATH_SPARK_SQL_JOB_QUERY + "/{id}" + "/{dataSourceMDSId}",
+      path: ROUTE_PATH_SPARK_SQL_JOB_QUERY + "/{id}" + "/{dataSourceMDSId?}",
       validate: {
         params: schema.object({
           id: schema.string(),