From ef009e114a3bea46cfdfb78e9baf1d3e1f7110f6 Mon Sep 17 00:00:00 2001 From: Maxime Petazzoni Date: Tue, 24 Dec 2013 10:07:25 -0800 Subject: [PATCH] Style fixes for flake8 Signed-off-by: Maxime Petazzoni --- docker/client.py | 5 ++--- tests/integration_test.py | 2 ++ 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/docker/client.py b/docker/client.py index 05a61d44..8c13ea2a 100644 --- a/docker/client.py +++ b/docker/client.py @@ -276,10 +276,9 @@ class Client(requests.Session): if not length: break data = recvall(socket, length) - if data: - yield data - else: + if not data: break + yield data def attach(self, container): socket = self.attach_socket(container) diff --git a/tests/integration_test.py b/tests/integration_test.py index 1fdde0e3..8d92d1ed 100644 --- a/tests/integration_test.py +++ b/tests/integration_test.py @@ -274,6 +274,7 @@ class TestLogs(BaseTestCase): logs = self.client.logs(id) self.assertEqual(logs, snippet + '\n') + class TestLogsStreaming(BaseTestCase): def runTest(self): snippet = 'Flowering Nights (Sakuya Iyazoi)' @@ -292,6 +293,7 @@ class TestLogsStreaming(BaseTestCase): self.assertEqual(logs, snippet + '\n') + class TestLogsWithDictInsteadOfId(BaseTestCase): def runTest(self): snippet = 'Flowering Nights (Sakuya Iyazoi)'