def test_handle_for_comps(self, mock_file_units):
# Setup
self.sample_comps_filename = os.path.join(DATA_DIR, 'simple_repo_comps', XML_FILENAME)
unit_key = {}
metadata = {}
config = PluginCallConfiguration({}, {})
mock_repo = mock.MagicMock()
mock_repo.id = 'some_id'
mock_conduit = mock.MagicMock()
# Test
upload._handle_group_category(mock_repo, models.PackageCategory.TYPE, unit_key, metadata,
self.sample_comps_filename, mock_conduit, config)
# Verify
self.assertEqual(mock_file_units.call_count, 3)
group_args = (self.sample_comps_filename, group.process_group_element, group.GROUP_TAG,
mock_conduit, mock_repo.id)
category_args = (self.sample_comps_filename, group.process_category_element,
group.CATEGORY_TAG, mock_conduit, mock_repo.id)
environment_args = (self.sample_comps_filename, group.process_environment_element,
group.ENVIRONMENT_TAG, mock_conduit, mock_repo.id)
self.assertEqual(mock_file_units.call_args_list[0][0], group_args)
self.assertEqual(mock_file_units.call_args_list[1][0], category_args)
self.assertEqual(mock_file_units.call_args_list[2][0], environment_args)
def test_handle_for_category(self):
# Setup
unit_key = {"id": "test-category", "repo_id": "test-repo"}
metadata = {}
config = PluginCallConfiguration({}, {})
mock_conduit = mock.MagicMock()
inited_unit = Unit(models.PackageCategory.TYPE, unit_key, metadata, None)
mock_conduit.init_unit.return_value = inited_unit
# Test
upload._handle_group_category(models.PackageCategory.TYPE, unit_key, metadata, None, mock_conduit, config)
# Verify
mock_conduit.init_unit.assert_called_once_with(models.PackageCategory.TYPE, unit_key, metadata, None)
mock_conduit.save_unit.assert_called_once()
saved_unit = mock_conduit.save_unit.call_args[0][0]
self.assertEqual(inited_unit, saved_unit)
def test_handle_for_category(self):
# Setup
unit_key = {'id': 'test-category', 'repo_id': 'test-repo'}
metadata = {}
config = PluginCallConfiguration({}, {})
mock_conduit = mock.MagicMock()
inited_unit = Unit(models.PackageCategory.TYPE, unit_key, metadata, None)
mock_conduit.init_unit.return_value = inited_unit
# Test
upload._handle_group_category(models.PackageCategory.TYPE, unit_key, metadata, None,
mock_conduit, config)
# Verify
mock_conduit.init_unit.assert_called_once_with(models.PackageCategory.TYPE, unit_key,
metadata, None)
mock_conduit.save_unit.assert_called_once()
saved_unit = mock_conduit.save_unit.call_args[0][0]
self.assertEqual(inited_unit, saved_unit)
