Merge branch 'main' into clinic-syslog

python · serhiy-storchaka · Oct 8, 2022 · Jul 19, 2022 · Jul 20, 2022 · Jul 20, 2022
commit 0e0d9c69a62e06d302d26b7edc92d016d3e931cb
diff --git a/Lib/test/audit-tests.py b/Lib/test/audit-tests.py
@@ -419,6 +419,17 @@ def hook(event, args):
     sys._getframe()
 
 
+def test_wmi_exec_query():
+    import _wmi
+
+    def hook(event, args):
+        if event.startswith("_wmi."):
+            print(event, args[0])
+
+    sys.addaudithook(hook)
+    _wmi.exec_query("SELECT * FROM Win32_OperatingSystem")
+
+
 def test_syslog():
     import syslog
 

diff --git a/Lib/test/test_audit.py b/Lib/test/test_audit.py
@@ -185,10 +185,26 @@ def test_sys_getframe(self):
 
         self.assertEqual(actual, expected)
 
+
+    def test_wmi_exec_query(self):
+        import_helper.import_module("_wmi")
+        returncode, events, stderr = self.run_python("test_wmi_exec_query")
+        if returncode:
+            self.fail(stderr)
+
+        if support.verbose:
+            print(*events, sep='\n')
+        actual = [(ev[0], ev[2]) for ev in events]
+        expected = [("_wmi.exec_query", "SELECT * FROM Win32_OperatingSystem")]
+
+        self.assertEqual(actual, expected)
+
+
     def test_syslog(self):
         syslog = import_helper.import_module("syslog")
 
         returncode, events, stderr = self.run_python("test_syslog")
+
         if returncode:
             self.fail(stderr)
 
@@ -203,5 +219,6 @@ def test_syslog(self):
             events
         )
 
+
 if __name__ == "__main__":
     unittest.main()