From f66e3a6f6cfcc5bc9587c86195542eb822625dd5 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=8E=8B=E9=A3=9E?= <feiazifeiazi@hotmail.com>
Date: Wed, 25 Dec 2024 10:55:07 +0800
Subject: [PATCH] test_query_masking

---
 sql/engines/tests.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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):