first commit

test_batch3.py

@@ -0,0 +1,47 @@
+import threading
+import time
+from license_manager.models import HeartbeatRequest, HeartbeatResponse
+from license_manager.heartbeat import HeartbeatManager
+
+# Mock service for testing
+class MockHeartbeatService:
+    def __init__(self):
+        self.call_count = 0
+        
+    def send(self, req: HeartbeatRequest) -> HeartbeatResponse:
+        self.call_count += 1
+        # alternate failing and success
+        if self.call_count % 2 == 0:
+            raise ValueError("mock network error")
+        return HeartbeatResponse(status="ok", heartbeat_interval=1)
+
+def test_heartbeat_manager():
+    service = MockHeartbeatService()
+    req = HeartbeatRequest(license_key="test", hardware_fingerprint="fp")
+    
+    cb_calls = []
+    err_calls = []
+    
+    def on_update(resp):
+        cb_calls.append(resp)
+        
+    def on_error(err):
+        err_calls.append(err)
+        
+    mgr = HeartbeatManager(service, req, interval_seconds=1, on_license_updated=on_update, on_error=on_error)
+    mgr.start()
+    
+    # let it run for 2.5 seconds. Should make ~2 calls.
+    time.sleep(2.5)
+    mgr.pause()
+    time.sleep(1.5) # Shouldn't make calls while paused
+    
+    mgr.stop()
+    
+    assert service.call_count > 0, "Should have made calls"
+    assert len(err_calls) > 0, "Should have received errors based on mock"
+    
+    print("Heartbeat manager verified.")
+
+if __name__ == "__main__":
+    test_heartbeat_manager()