Fix columns duplication on MongoDB Query Runner

getredash · Feb 6, 2024 · 4632886 · 4632886
1 parent 6041750
commit 4632886
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/query_runner/test_mongodb.py b/tests/query_runner/test_mongodb.py
@@ -12,7 +12,7 @@
     parse_query_json,
     parse_results,
 )
-from redash.utils import json_dumps, json_loads, parse_human_time
+from redash.utils import json_dumps, parse_human_time
 
 
 @patch("redash.query_runner.mongodb.pymongo.MongoClient")
@@ -63,7 +63,7 @@ def test_run_query_with_fields(self, mongo_client):
         result, err = mongo_qr.run_query(json_dumps(query), None)
 
         self.assertIsNone(err)
-        self.assertEqual(expected, json_loads(result))
+        self.assertEqual(expected, result)
 
     def test_run_query_with_aggregate(self, mongo_client):
         config = {
@@ -96,7 +96,7 @@ def test_run_query_with_aggregate(self, mongo_client):
         mongo_client().__getitem__().__getitem__().aggregate.return_value = return_value
         result, err = mongo_qr.run_query(json_dumps(query), None)
         self.assertIsNone(err)
-        self.assertEqual(expected, json_loads(result))
+        self.assertEqual(expected, result)
 
 
 class TestParseQueryJson(TestCase):