diff options
Diffstat (limited to 'glance/tests/unit/common/test_format_inspector.py')
-rw-r--r-- | glance/tests/unit/common/test_format_inspector.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/glance/tests/unit/common/test_format_inspector.py b/glance/tests/unit/common/test_format_inspector.py index 4b29c19b3..d229d094f 100644 --- a/glance/tests/unit/common/test_format_inspector.py +++ b/glance/tests/unit/common/test_format_inspector.py @@ -153,6 +153,24 @@ class TestFormatInspectors(test_utils.BaseTestCase): def test_vdi_invalid(self): self._test_format_with_invalid_data('vdi') + def test_vmdk_invalid_type(self): + fmt = format_inspector.get_inspector('vmdk')() + wrapper = format_inspector.InfoWrapper(open(__file__, 'rb'), fmt) + while True: + chunk = wrapper.read(32) + if not chunk: + break + + wrapper.close() + + fake_rgn = mock.MagicMock() + fake_rgn.complete = True + fake_rgn.data = b'foocreateType="someunknownformat"bar' + + with mock.patch.object(fmt, 'has_region', return_value=True): + with mock.patch.object(fmt, 'region', return_value=fake_rgn): + self.assertEqual(0, fmt.virtual_size) + class TestFormatInspectorInfra(test_utils.BaseTestCase): def _test_capture_region_bs(self, bs): |