Update pytest fixtures to new yield syntax (#919)

This commit is contained in:
Jeff Widman
2017-01-11 17:18:28 -08:00
committed by Dana Powers
parent 9a08efbdd2
commit cb06a6b125
2 changed files with 6 additions and 10 deletions

View File

@@ -18,10 +18,8 @@ def version():
def zookeeper(version, request): def zookeeper(version, request):
assert version assert version
zk = ZookeeperFixture.instance() zk = ZookeeperFixture.instance()
def fin(): yield zk
zk.close() zk.close()
request.addfinalizer(fin)
return zk
@pytest.fixture(scope="module") @pytest.fixture(scope="module")
@@ -29,10 +27,8 @@ def kafka_broker(version, zookeeper, request):
assert version assert version
k = KafkaFixture.instance(0, zookeeper.host, zookeeper.port, k = KafkaFixture.instance(0, zookeeper.host, zookeeper.port,
partitions=4) partitions=4)
def fin(): yield k
k.close() k.close()
request.addfinalizer(fin)
return k
@pytest.fixture @pytest.fixture

View File

@@ -32,8 +32,8 @@ def reporter():
@pytest.fixture @pytest.fixture
def metrics(request, config, reporter): def metrics(request, config, reporter):
metrics = Metrics(config, [reporter], enable_expiration=True) metrics = Metrics(config, [reporter], enable_expiration=True)
request.addfinalizer(lambda: metrics.close()) yield metrics
return metrics metrics.close()
def test_MetricName(): def test_MetricName():