def test_roundtrip(self):
ref_package = VodPackage(reference_xml)
file_out = BytesIO()
file_out.write(ref_package.write_xml())
file_out.seek(0)
new_package = VodPackage(file_out)
self.assertEqual(file_out.getvalue(), new_package.write_xml())
def test_roundtrip(self):
ref_package = VodPackage(reference_xml)
file_out = BytesIO()
file_out.write(ref_package.write_xml())
file_out.seek(0)
new_package = VodPackage(file_out)
self.assertEqual(file_out.getvalue(), new_package.write_xml())
def test_overwrite_xml(self, mock_open):
mock_open.return_value = MagicMock()
file_handle = mock_open.return_value.__enter__.return_value
vod_package = VodPackage(reference_xml)
vod_package.overwrite_xml()
file_handle.write.assert_called_once_with(vod_package.write_xml())
def test_rewrite(self, mock_check_file):
vod_package = VodPackage(reference_xml)
vod_package.write_xml(rewrite=True)
mock_check_file.assert_called_once_with(vod_package)
def test_missing_movie(self):
vod_package = VodPackage(reference_xml)
del vod_package.D_ams["movie"]
with self.assertRaises(MissingElement):
vod_package.write_xml()
def test_overwrite_xml(self, mock_open):
mock_open.return_value = MagicMock()
file_handle = mock_open.return_value.__enter__.return_value
vod_package = VodPackage(reference_xml)
vod_package.overwrite_xml()
file_handle.write.assert_called_once_with(vod_package.write_xml())
def test_rewrite(self, mock_check_file):
vod_package = VodPackage(reference_xml)
vod_package.write_xml(rewrite=True)
mock_check_file.assert_called_once_with(vod_package)
def test_missing_movie(self):
vod_package = VodPackage(reference_xml)
del vod_package.D_ams["movie"]
with self.assertRaises(MissingElement):
vod_package.write_xml()
