def test_put_repo_group_distributor_no_config_specified(self, mock_factory, mock_resp):
"""
Modifiy a distributor without specifying the distributor_config.
"""
mock_request = mock.MagicMock()
mock_request.body = json.dumps({'no_dist_config': 'mock_conf'})
repo_group_distributor_resource = RepoGroupDistributorResourceView()
try:
repo_group_distributor_resource.put(mock_request, 'group_id', 'dist_id')
except pulp_exceptions.MissingValue, response:
pass
def test_put_repo_group_distributor(self, mock_factory, mock_resp, mock_rev):
"""
Modifiy a distributor with required argument.
"""
mock_request = mock.MagicMock()
mock_rev.return_value = '/mock/path/'
mock_request.body = json.dumps({'distributor_config': 'mock_conf'})
mock_dist_manager = mock_factory.repo_group_distributor_manager.return_value
mock_dist_manager.update_distributor_config.return_value = {'updated': 'distributor'}
repo_group_distributor_resource = RepoGroupDistributorResourceView()
response = repo_group_distributor_resource.put(mock_request, 'group_id', 'dist_id')
expected_content = {'updated': 'distributor', '_href': '/mock/path/'}
mock_resp.assert_called_once_with(expected_content)
mock_dist_manager.update_distributor_config.assert_called_once_with(
'group_id', 'dist_id', 'mock_conf')
self.assertTrue(response is mock_resp.return_value)