test_query_masking

hhyo · Dec 25, 2024 · f66e3a6 · f66e3a6
1 parent 7a023ed
commit f66e3a6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/sql/engines/tests.py b/sql/engines/tests.py
@@ -2042,6 +2042,14 @@ def test_create_instance_user(self, _mock_command):
             ],
         )
 
+    def test_query_masking(self):
+        query_result = ResultSet()
+        new_engine = MongoEngine(instance=self.ins)
+        masking_result = new_engine.query_masking(
+            db_name="archery", sql="db.test_collection.find()", resultset=query_result
+        )
+        self.assertIsInstance(masking_result, ResultSet)
+
 
 class TestClickHouse(TestCase):
     def setUp(self):