tests: conditional mock for telemetry client

argilla-io · Aug 29, 2022 · 830d0ee · 830d0ee
1 parent aea7c2a
commit 830d0ee
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/tests/conftest.py b/tests/conftest.py
@@ -20,11 +20,12 @@
 def telemetry_track_data(mocker):
 
     client = telemetry._TelemetryClient.get()
-    # Disable sending data for tests
-    client._client = telemetry._configure_analytics(disable_send=True)
-    spy = mocker.spy(client, "track_data")
+    if client:
+        # Disable sending data for tests
+        client._client = telemetry._configure_analytics(disable_send=True)
+        spy = mocker.spy(client, "track_data")
 
-    return spy
+        return spy
 
 
 @pytest.fixture