cov

databrickslabs · Apr 25, 2024 · d61fe4c · d61fe4c
1 parent f20264c
commit d61fe4c
Showing 1 changed file with 14 additions and 2 deletions.
diff --git a/tests/unit/source_code/test_redash.py b/tests/unit/source_code/test_redash.py
@@ -15,14 +15,16 @@
 def redash_ws():
     def api_client_do(method: str, path: str, body: dict | None = None):
         _ = body
-        tags_lookup = {"1": [Redash.BACKUP_TAG], "2": [Redash.MIGRATED_TAG, "backup: 123"], "3": []}
+        tags_lookup = {"1": [Redash.BACKUP_TAG], "2": [Redash.MIGRATED_TAG, "backup: 123", "original"], "3": []}
         if method == "POST":
             return {}
         if path.startswith("/api/2.0/preview/sql/dashboards"):
             dashboard_id = path.removeprefix("/api/2.0/preview/sql/dashboards/")
             tags = tags_lookup[dashboard_id]
             return {
                 "widgets": [
+                    {},
+                    {"visualization": {}},
                     {
                         "visualization": {
                             "query": {
@@ -32,7 +34,17 @@ def api_client_do(method: str, path: str, body: dict | None = None):
                                 "tags": tags,
                             }
                         }
-                    }
+                    },
+                    {
+                        "visualization": {
+                            "query": {
+                                "id": "1",
+                                "name": "test_query",
+                                "query": "SELECT * FROM old.things",
+                                "tags": None,
+                            }
+                        }
+                    },
                 ]
             }
         if path.startswith("/api/2.0/preview/sql/queries"):