Update buildargs test to avoid race condition838-update

Signed-off-by: Joffrey F <joffrey@docker.com>

1 files changed, 5 insertions, 2 deletions

diff --git a/tests/integration/build_test.py b/tests/integration/build_test.py
index 1f5ed9e..93280ab 100644
--- a/tests/integration/build_test.py
+++ b/tests/integration/build_test.py
@@ -105,9 +105,12 @@ class BuildTest(api_test.BaseTestCase):
             'USER $test'
         ]).encode('ascii'))
 
-        self.client.build(fileobj=script, tag='buildargs',
-                          buildargs={'test': 'OK'}, stream=True)
+        stream = self.client.build(
+            fileobj=script, tag='buildargs', buildargs={'test': 'OK'}
+        )
         self.tmp_imgs.append('buildargs')
+        for chunk in stream:
+            pass
 
         info = self.client.inspect_image('buildargs')
         self.assertEqual(info['Config']['User'], 'OK')