diff options
Diffstat (limited to 'tests/unit')
| -rw-r--r-- | tests/unit/models_images_test.py | 17 |
1 files changed, 15 insertions, 2 deletions
diff --git a/tests/unit/models_images_test.py b/tests/unit/models_images_test.py index 9ecb7e4..dacd72b 100644 --- a/tests/unit/models_images_test.py +++ b/tests/unit/models_images_test.py @@ -41,9 +41,22 @@ class ImageCollectionTest(unittest.TestCase): def test_pull(self): client = make_fake_client() - image = client.images.pull('test_image') + image = client.images.pull('test_image:latest') + client.api.pull.assert_called_with('test_image', tag='latest') + client.api.inspect_image.assert_called_with('test_image:latest') + assert isinstance(image, Image) + assert image.id == FAKE_IMAGE_ID + + def test_pull_multiple(self): + client = make_fake_client() + images = client.images.pull('test_image') client.api.pull.assert_called_with('test_image', tag=None) - client.api.inspect_image.assert_called_with('test_image') + client.api.images.assert_called_with( + all=False, name='test_image', filters=None + ) + client.api.inspect_image.assert_called_with(FAKE_IMAGE_ID) + assert len(images) == 1 + image = images[0] assert isinstance(image, Image) assert image.id == FAKE_IMAGE_ID |