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)