def test_generate_rpm_data(self):
unit_key, unit_metadata = generate_rpm_data(self.filename)
self.assertEqual(unit_key['name'], 'bear')
self.assertEqual(unit_key['release'], '1')
self.assertEqual(unit_key['version'], '4.1')
self.assertEqual(unit_key['arch'], 'noarch')
self.assertEqual(unit_metadata['filename'], 'bear-4.1-1.noarch.rpm')
self.assertEqual(unit_metadata['license'], 'GPLv2')
self.assertEqual(unit_metadata['vendor'], None)
def get_content_data(cls, content_type, filepath):
unit_key = metadata = None
if content_type == "yum":
try:
unit_key, metadata = generate_rpm_data(filepath)
except InvalidRPMError:
print _("Invalid rpm '%s'. Please check the file and try again.") % filepath
elif content_type == "puppet":
try:
unit_key, metadata = generate_puppet_data(filepath)
except ExtractionException:
print _("Invalid puppet module '%s'. Please check the file and try again.") % filepath
else:
print _("Content type '%s' not valid. Must be puppet or yum.") % content_type
return unit_key, metadata
def get_content_data(cls, content_type, filepath):
unit_key = metadata = None
if content_type == "yum":
try:
unit_key, metadata = generate_rpm_data(filepath)
except InvalidRPMError:
print _("Invalid rpm '%s'. Please check the file and try again.") % filepath
elif content_type == "puppet":
try:
unit_key, metadata = generate_puppet_data(filepath)
except ExtractionException:
print _(
"Invalid puppet module '%s'. Please make sure the file is valid and is named "
+ "author-name-version.tar.gz (eg: puppetlabs-ntp-2.0.1.tar.gz)."
) % filepath
return unit_key, metadata
