summaryrefslogtreecommitdiffstats
path: root/vio
diff options
context:
space:
mode:
Diffstat (limited to 'vio')
-rw-r--r--vio/vio/tests/test_fake_image_view.py14
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)