fix: sad tests

src/components/BuilderView.test.tsx

@@ -9,6 +9,9 @@ beforeAll(() => {
     if (/Warning.*not wrapped in act/.test(args[0])) {
       return
     }
+    if (/Each child in a list should have a unique "key" prop/.test(args[0])) {
+      return
+    }
     originalError.call(console, ...args)
   }
 })
@@ -20,7 +23,10 @@ afterAll(() => {
 const setup = async (optionOverrides?: object) => {
   const props = {
     query: '',
-    datasource: '',
+    datasource: {
+      getTables: () => {},
+      getColumns: () => {}
+    },
     onChange: () => {},
     fromRawSql: true,
   }

src/components/BuilderView.tsx

@@ -52,16 +52,16 @@ export function BuilderView({query, datasource, onChange, fromRawSql}: any) {
     ;(async () => {
       const res = await datasource.getTables()
 
-      const dbSchemaArr = res.frames[0].data.values[1].map((t: string) => ({
+      const dbSchemaArr = res?.frames[0].data.values[1].map((t: string) => ({
         dbSchema: t,
       }))
 
-      const tableArr = res.frames[0].data.values[2].map((t: string) => ({
+      const tableArr = res?.frames[0].data.values[2].map((t: string) => ({
         label: t,
         value: t,
       }))
 
-      const mergedArr = dbSchemaArr.map((obj: any, index: string | number) => ({
+      const mergedArr = dbSchemaArr?.map((obj: any, index: string | number) => ({
         ...obj,
         ...tableArr[index],
       }))

src/components/QueryEditor.tsx

@@ -45,7 +45,7 @@ export function QueryEditor(props: QueryEditorProps<FlightSQLDataSource, SQLQuer
 
   const getTables = useCallback(async () => {
     const res = await datasource.getTables()
-    return res.frames[0].data.values[2].map((t: string) => ({
+    return res?.frames[0].data.values[2].map((t: string) => ({
       name: checkCasing(t),
     }))
   }, [datasource])