fix CircleCI test

getredash · Sep 10, 2020 · 673c556 · 673c556
1 parent 97fc91f
commit 673c556
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 8 deletions.
diff --git a/tests/handlers/test_query_results.py b/tests/handlers/test_query_results.py
@@ -79,7 +79,7 @@ def test_execute_new_query(self):
 
     def test_add_limit_change_query_sql(self):
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="databricks"
+            group=self.factory.org.default_group, type="pg"
         )
         query = self.factory.create_query(query_text="SELECT 2", data_source=ds)
         query_result = self.factory.create_query_result(data_source=ds, query_hash=query.query_hash)
@@ -100,7 +100,7 @@ def test_add_limit_change_query_sql(self):
 
     def test_add_limit_no_change_for_nonsql(self):
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="druid"
+            group=self.factory.org.default_group, type="prometheus"
         )
         query = self.factory.create_query(query_text="SELECT 5", data_source=ds)
         query_result = self.factory.create_query_result(data_source=ds, query_hash=query.query_hash)

diff --git a/tests/tasks/test_refresh_queries.py b/tests/tasks/test_refresh_queries.py
@@ -49,7 +49,7 @@ def test_enqueues_outdated_queries_for_non_sqlquery(self):
         from Query.outdated_queries().
         """
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="phoenix"
+            group=self.factory.org.default_group, type="prometheus"
         )
         query1 = self.factory.create_query(data_source=ds, options={"apply_auto_limit": True})
         query2 = self.factory.create_query(
@@ -112,7 +112,7 @@ def test_doesnt_enqueue_outdated_queries_for_paused_data_source_for_non_sqlquery
         data source is paused.
         """
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="phoenix"
+            group=self.factory.org.default_group, type="prometheus"
         )
         query = self.factory.create_query(data_source=ds, options={"apply_auto_limit": True})
         oq = staticmethod(lambda: [query])
@@ -171,7 +171,7 @@ def test_enqueues_parameterized_queries_for_non_sqlquery(self):
         Scheduled queries with parameters use saved values.
         """
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="phoenix"
+            group=self.factory.org.default_group, type="prometheus"
         )
         query = self.factory.create_query(
             query_text="select {{n}}",

diff --git a/tests/test_models.py b/tests/test_models.py
@@ -474,7 +474,7 @@ def test_order_by_relationship(self):
 
     def test_update_query_hash_basesql_with_options(self):
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="databricks"
+            group=self.factory.org.default_group, type="pg"
         )
         query = self.factory.create_query(query_text="SELECT 2", data_source=ds)
         query.options = {"apply_auto_limit": True}
@@ -484,7 +484,7 @@ def test_update_query_hash_basesql_with_options(self):
 
     def test_update_query_hash_basesql_no_options(self):
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="databricks"
+            group=self.factory.org.default_group, type="pg"
         )
         query = self.factory.create_query(query_text="SELECT 2", data_source=ds)
         query.options = {}
@@ -494,7 +494,7 @@ def test_update_query_hash_basesql_no_options(self):
 
     def test_update_query_hash_non_basesql(self):
         ds = self.factory.create_data_source(
-            group=self.factory.org.default_group, type="druid"
+            group=self.factory.org.default_group, type="prometheus"
         )
         query = self.factory.create_query(query_text="SELECT 2", data_source=ds)
         query.options = {"apply_auto_limit": True}