def process_main(self, unit=None):
wd = self.get_working_dir()
total = len(self.parent.publish_msi.units +
self.parent.publish_msm.units)
checksum_type = 'sha256'
with PrimaryXMLFileContext(wd, total, checksum_type) as primary:
units = itertools.chain(self.parent.publish_msi.units,
self.parent.publish_msm.units)
for unit in units:
primary.add_unit_metadata(unit)
with FilelistsXMLFileContext(wd, checksum_type) as filelists:
for unit in units:
filelists.add_unit_metadata(unit)
with OtherXMLFileContext(wd, checksum_type) as other:
for unit in units:
other.add_unit_metadata(unit)
with RepomdXMLFileContext(wd, checksum_type) as repomd:
repomd.add_metadata_file_metadata('primary',
primary.metadata_file_path,
primary.checksum)
repomd.add_metadata_file_metadata('filelists',
filelists.metadata_file_path,
filelists.checksum)
repomd.add_metadata_file_metadata('other',
other.metadata_file_path,
other.checksum)
def setUp(self):
self.working_dir = tempfile.mkdtemp()
self.context = PrimaryXMLFileContext(self.working_dir,
3,
checksum_type='sha256')
self.unit = mock.Mock()
self.unit.render_primary.return_value = 'somexml'
def setUp(self):
self.working_dir = tempfile.mkdtemp()
self.context = PrimaryXMLFileContext(self.working_dir, 3)
