Bugfix: Disabled clusters raised exception on len(celery_setup) (#197)

Author: Nusnus

src/pytest_celery/api/setup.py

@@ -56,7 +56,14 @@ def __init__(
 
     def __len__(self) -> int:
         """The total number of nodes in the setup."""
-        return len(self._worker_cluster) + len(self._broker_cluster) + len(self._backend_cluster)
+        nodes_count = 0
+        if self.broker_cluster:
+            nodes_count += len(self.broker_cluster)
+        if self.backend_cluster:
+            nodes_count += len(self.backend_cluster)
+        if self.worker_cluster:
+            nodes_count += len(self.worker_cluster)
+        return nodes_count
 
     @property
     def app(self) -> Celery:

tests/unit/api/test_setup.py

@@ -80,6 +80,9 @@ def test_disabling_backend_cluster(self, celery_setup: CeleryTestSetup):
             assert celery_setup.backend_cluster is None
             assert celery_setup.backend is None
 
+        def test_setup_length(self, celery_setup: CeleryTestSetup):
+            assert len(celery_setup)
+
     class test_disabling_broker_cluster:
         @pytest.fixture
         def celery_broker_cluster(self) -> CeleryBrokerCluster:
@@ -89,6 +92,9 @@ def test_disabling_broker_cluster(self, celery_setup: CeleryTestSetup):
             assert celery_setup.broker_cluster is None
             assert celery_setup.broker is None
 
+        def test_setup_length(self, celery_setup: CeleryTestSetup):
+            assert len(celery_setup)
+
     class test_disabling_worker_cluster:
         @pytest.fixture
         def celery_worker_cluster(self) -> CeleryWorkerCluster:
@@ -97,3 +103,6 @@ def celery_worker_cluster(self) -> CeleryWorkerCluster:
         def test_disabling_worker_cluster(self, celery_setup: CeleryTestSetup):
             assert celery_setup.worker_cluster is None
             assert celery_setup.worker is None
+
+        def test_setup_length(self, celery_setup: CeleryTestSetup):
+            assert len(celery_setup)