|
2 | 2 |
|
3 | 3 | import pytest |
4 | 4 | from celery import Celery |
5 | | -from pytest_lazyfixture import lazy_fixture |
6 | 5 |
|
7 | | -from pytest_celery import CELERY_WORKER |
8 | | -from pytest_celery import CELERY_WORKER_CLUSTER |
| 6 | +from pytest_celery import CeleryTestCluster |
| 7 | +from pytest_celery import CeleryTestNode |
9 | 8 | from pytest_celery import CeleryTestWorker |
10 | 9 | from pytest_celery import CeleryWorkerCluster |
11 | 10 | from pytest_celery import CeleryWorkerContainer |
| 11 | +from tests.integration.api.test_base import BaseCluster |
| 12 | +from tests.integration.api.test_base import BaseNodes |
12 | 13 |
|
13 | 14 |
|
14 | | -@pytest.mark.parametrize("worker", [lazy_fixture(CELERY_WORKER)]) |
15 | | -class test_celey_test_worker: |
16 | | - def test_app(self, worker: CeleryTestWorker, celery_setup_app: Celery): |
17 | | - assert worker.app is celery_setup_app |
| 15 | +class test_celey_test_worker(BaseNodes): |
| 16 | + @pytest.fixture |
| 17 | + def node(self, celery_worker: CeleryTestWorker) -> CeleryTestNode: |
| 18 | + return celery_worker |
18 | 19 |
|
19 | | - def test_version(self, worker: CeleryTestWorker): |
20 | | - assert worker.version == CeleryWorkerContainer.version() |
| 20 | + def test_app(self, celery_worker: CeleryTestWorker, celery_setup_app: Celery): |
| 21 | + assert celery_worker.app is celery_setup_app |
21 | 22 |
|
22 | | - def test_hostname(self, worker: CeleryTestWorker): |
23 | | - hostname = worker.hostname() |
| 23 | + def test_version(self, celery_worker: CeleryTestWorker): |
| 24 | + assert celery_worker.version == CeleryWorkerContainer.version() |
| 25 | + |
| 26 | + def test_hostname(self, celery_worker: CeleryTestWorker): |
| 27 | + hostname = celery_worker.hostname() |
24 | 28 | assert "@" in hostname |
25 | | - assert worker.worker_name in hostname.split("@")[0] |
26 | | - assert worker.container.id[:12] in hostname.split("@")[1] |
| 29 | + assert celery_worker.worker_name in hostname.split("@")[0] |
| 30 | + assert celery_worker.container.id[:12] in hostname.split("@")[1] |
| 31 | + |
| 32 | + def test_wait_for_log(self, celery_worker: CeleryTestWorker): |
| 33 | + log = f"{celery_worker.hostname()} v{celery_worker.version}" |
| 34 | + celery_worker.wait_for_log(log, "test_celey_test_worker.test_wait_for_log") |
27 | 35 |
|
28 | | - def test_wait_for_log(self, worker: CeleryTestWorker): |
29 | | - log = f"{worker.hostname()} v{worker.version}" |
30 | | - worker.wait_for_log(log, "test_celey_test_worker.test_wait_for_log") |
| 36 | + def test_assert_log_exists(self, celery_worker: CeleryTestWorker): |
| 37 | + log = f"{celery_worker.hostname()} v{celery_worker.version}" |
| 38 | + celery_worker.assert_log_exists(log, "test_celey_test_worker.test_assert_log_exists") |
31 | 39 |
|
32 | | - def test_assert_log_exists(self, worker: CeleryTestWorker): |
33 | | - log = f"{worker.hostname()} v{worker.version}" |
34 | | - worker.assert_log_exists(log, "test_celey_test_worker.test_assert_log_exists") |
35 | 40 |
|
| 41 | +class test_celery_worker_cluster(BaseCluster): |
| 42 | + @pytest.fixture |
| 43 | + def cluster(self, celery_worker_cluster: CeleryWorkerCluster) -> CeleryTestCluster: |
| 44 | + return celery_worker_cluster |
36 | 45 |
|
37 | | -@pytest.mark.parametrize("cluster", [lazy_fixture(CELERY_WORKER_CLUSTER)]) |
38 | | -class test_celery_worker_cluster: |
39 | | - def test_app(self, cluster: CeleryWorkerCluster, celery_setup_app: Celery): |
| 46 | + def test_app(self, celery_worker_cluster: CeleryWorkerCluster, celery_setup_app: Celery): |
40 | 47 | worker: CeleryTestWorker |
41 | | - for worker in cluster: |
| 48 | + for worker in celery_worker_cluster: |
42 | 49 | assert worker.app is celery_setup_app |
43 | 50 |
|
44 | | - def test_config(self, cluster: CeleryWorkerCluster): |
| 51 | + def test_config(self, celery_worker_cluster: CeleryWorkerCluster): |
45 | 52 | with pytest.raises(NotImplementedError): |
46 | | - cluster.config() |
| 53 | + celery_worker_cluster.config() |
47 | 54 |
|
48 | | - def test_versions(self, cluster: CeleryWorkerCluster): |
49 | | - assert cluster.versions == {CeleryWorkerContainer.version()} |
| 55 | + def test_versions(self, celery_worker_cluster: CeleryWorkerCluster): |
| 56 | + assert celery_worker_cluster.versions == {CeleryWorkerContainer.version()} |
0 commit comments