add unit test

pixee · Oct 17, 2023 · fb55680 · fb55680
1 parent 969d6bb
commit fb55680
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/tests/codemods/test_with_threading_lock.py b/tests/codemods/test_with_threading_lock.py
@@ -170,6 +170,21 @@ def f(l):
     lock = threading.Lock()
     with lock:
         print()
+""",
+            ),
+            (
+                """import threading
+def my_func():
+    lock = "whatever"
+    with threading.Lock():
+        foo()
+""",
+                """import threading
+def my_func():
+    lock = "whatever"
+    lock_1 = threading.Lock()
+    with lock_1:
+        foo()
 """,
             ),
         ],