diff options
Diffstat (limited to 'test/conftest.py')
| -rw-r--r-- | test/conftest.py | 12 |
1 files changed, 4 insertions, 8 deletions
diff --git a/test/conftest.py b/test/conftest.py index 79ad074..e85b977 100644 --- a/test/conftest.py +++ b/test/conftest.py @@ -18,10 +18,8 @@ def version(): def zookeeper(version, request): assert version zk = ZookeeperFixture.instance() - def fin(): - zk.close() - request.addfinalizer(fin) - return zk + yield zk + zk.close() @pytest.fixture(scope="module") @@ -29,10 +27,8 @@ def kafka_broker(version, zookeeper, request): assert version k = KafkaFixture.instance(0, zookeeper.host, zookeeper.port, partitions=4) - def fin(): - k.close() - request.addfinalizer(fin) - return k + yield k + k.close() @pytest.fixture |