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()