def test_open_handle(self):
path = os.path.join(self.metadata_file_dir, 'open_handle.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
self.assertTrue(os.path.exists(path))
context._close_metadata_file_handle()
def test_open_handle(self):
path = os.path.join(self.metadata_file_dir, 'open_handle.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
self.assertTrue(os.path.exists(path))
context._close_metadata_file_handle()
def test_finalize_checksum_type_none(self):
path = os.path.join(self.metadata_file_dir, 'test.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
context.finalize()
self.assertEqual(context.metadata_file_path, path)
def test_finalize_checksum_type_none(self):
path = os.path.join(self.metadata_file_dir, 'test.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
context.finalize()
self.assertEqual(context.metadata_file_path, path)
def test_finalize_for_repomd_file_with_valid_checksum_type(self):
path = os.path.join(self.metadata_file_dir, 'repomd.xml')
checksum_type = 'sha1'
context = MetadataFileContext(path, checksum_type)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
context.finalize()
self.assertEqual(context.metadata_file_path, path)
def test_finalize_for_repomd_file_with_valid_checksum_type(self):
path = os.path.join(self.metadata_file_dir, 'repomd.xml')
checksum_type = 'sha1'
context = MetadataFileContext(path, checksum_type)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
context.finalize()
self.assertEqual(context.metadata_file_path, path)
def test_finalize_with_valid_checksum_type(self):
path = os.path.join(self.metadata_file_dir, "test.xml")
checksum_type = "sha1"
context = MetadataFileContext(path, checksum_type)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
context.finalize()
expected_metadata_file_name = context.checksum + "-" + "test.xml"
expected_metadata_file_path = os.path.join(self.metadata_file_dir, expected_metadata_file_name)
self.assertEqual(expected_metadata_file_path, context.metadata_file_path)
def test_write_xml_header(self):
path = os.path.join(self.metadata_file_dir, 'header.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
self.assertTrue(os.path.exists(path))
with open(path) as h:
content = h.read()
expected_content = '<?xml version="1.0" encoding="UTF-8"?>\n'
self.assertEqual(content, expected_content)
def test_write_xml_header(self):
path = os.path.join(self.metadata_file_dir, 'header.xml')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
context._write_xml_header()
context._close_metadata_file_handle()
self.assertTrue(os.path.exists(path))
with open(path) as h:
content = h.read()
expected_content = '<?xml version="1.0" encoding="UTF-8"?>\n'
self.assertEqual(content, expected_content)
def test_open_handle_gzip(self):
path = os.path.join(self.metadata_file_dir, 'test.xml.gz')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
self.assertTrue(os.path.exists(path))
context._write_xml_header()
context._close_metadata_file_handle()
try:
h = gzip.open(path)
except Exception, e:
self.fail(e.message)
def test_open_handle_gzip(self):
path = os.path.join(self.metadata_file_dir, 'test.xml.gz')
context = MetadataFileContext(path)
context._open_metadata_file_handle()
self.assertTrue(os.path.exists(path))
context._write_xml_header()
context._close_metadata_file_handle()
try:
h = gzip.open(path)
except Exception, e:
self.fail(e.message)
