Cleanup coordinator test fixtures

Author: dpkp

test/test_coordinator.py

@@ -26,11 +26,15 @@
 
 @pytest.fixture
 def client(conn):
-    return KafkaClient(api_version=(0, 9))
+    cli = KafkaClient(api_version=(0, 9))
+    yield cli
+    cli.close()
 
 @pytest.fixture
 def coordinator(client):
-    return ConsumerCoordinator(client, SubscriptionState(), Metrics())
+    c = ConsumerCoordinator(client, SubscriptionState(), Metrics())
+    yield c
+    c.close()
 
 
 def test_init(client, coordinator):
@@ -400,6 +404,8 @@ def patched_coord(mocker, coordinator):
                         return_value=1)
     mocker.patch.object(coordinator._client, 'ready', return_value=True)
     mocker.patch.object(coordinator._client, 'send')
+    mocker.patch.object(coordinator, '_start_heartbeat_thread')
+    mocker.patch.object(coordinator, '_close_heartbeat_thread')
     mocker.spy(coordinator, '_failed_request')
     mocker.spy(coordinator, '_handle_offset_commit_response')
     mocker.spy(coordinator, '_handle_offset_fetch_response')