def test_units(self): result = content.parse_units(['foo/bar', 'abc/xyz']) self.assertEqual(len(result), 2) for unit in result: self.assertEqual(unit.get('type_id'), constants.TYPE_PUPPET_MODULE) unit_key = unit.get('unit_key', {}) self.assertTrue(unit_key.get('author') in ['foo', 'abc']) self.assertTrue(unit_key.get('name') in ['bar', 'xyz']) self.assertTrue('version' not in unit_key)
def test_single_unit_with_version(self): result = content.parse_units(['foo/bar/1.2.3']) self.assertEqual(len(result), 1) unit = result[0] self.assertEqual(unit.get('type_id'), constants.TYPE_PUPPET_MODULE) unit_key = unit.get('unit_key', {}) self.assertEqual(unit_key.get('author'), 'foo') self.assertEqual(unit_key.get('name'), 'bar') self.assertEqual(unit_key.get('version'), '1.2.3')
def test_empty_list(self): result = content.parse_units([]) self.assertEqual(result, [])