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):
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

View File

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