Fix #130 -- FakeLock context enter returns lock instance (#136)

* context manager returns a lock
* extend returns correct True value instead of None

---------

Co-authored-by: Anchal Gurung <[email protected]>

Author: codingjoe

dramatiq_crontab/utils.py

@@ -5,13 +5,13 @@
 
 class FakeLock:
     def __enter__(self):
-        pass
+        return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         pass
 
     def extend(self, additional_time=None, replace_ttl=False):
-        pass
+        return True
 
 
 if redis_url := get_settings().REDIS_URL:

tests/test_utils.py

@@ -20,3 +20,17 @@ def test_extend_lock__error():
         utils.extend_lock(lock, scheduler)
     assert lock.extend.call_count == 1
     assert scheduler.shutdown.call_count == 1
+
+
+class TestFakeLock:
+    def test_enter(self):
+        fake_lock = utils.FakeLock()
+        assert fake_lock.__enter__() is fake_lock
+
+    def test_exit(self):
+        fake_lock = utils.FakeLock()
+        assert fake_lock.__exit__(None, None, None) is None
+
+    def test_extend(self):
+        fake_lock = utils.FakeLock()
+        assert fake_lock.extend(additional_time=10, replace_ttl=True)