diff options
| -rw-r--r-- | vio/vio/tests/test_fake_image_view.py | 14 |
1 files changed, 14 insertions, 0 deletions
diff --git a/vio/vio/tests/test_fake_image_view.py b/vio/vio/tests/test_fake_image_view.py index d215156..7a49014 100644 --- a/vio/vio/tests/test_fake_image_view.py +++ b/vio/vio/tests/test_fake_image_view.py @@ -71,3 +71,17 @@ class TestFakeImageVersion(unittest.TestCase): } resp = self.view.get(mock.Mock()) self.assertEqual(200, resp.status_code) + + +class TestFakeImageDownload(unittest.TestCase): + + def setUp(self): + self.view = views.FakeImageDownload() + + @mock.patch.object(fakeResponse, "image_detail") + def test_get(self, mock_image_version): + mock_image_version.return_value = { + "id": "1234abcd" + } + resp = self.view.get(mock.Mock(), "1234abcd") + self.assertEqual(200, resp.status_code) |