diff --git a/sql/engines/tests.py b/sql/engines/tests.py index 8cb5c63eb3..b381c79f4e 100644 --- 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):