class TestRepositoryDistributorAPI(unittest.TestCase): """ Tests for the RepositoryDistributorAPI. """ def setUp(self): self.api = RepositoryDistributorAPI( mock.MagicMock(spec=PulpConnection)) def test_update(self): # Setup expected_path = self.api.base_path % 'test-repo' + 'test-distributor/' expected_body = { 'distributor_config': { 'key': 'value' }, 'delta': { 'auto_publish': True } } # Test result = self.api.update('test-repo', 'test-distributor', {'key': 'value'}, {'auto_publish': True}) self.api.server.PUT.assert_called_once_with(expected_path, expected_body) self.assertEqual(result, self.api.server.PUT.return_value)
def get_repo_relative_paths(repo_ids, source_path): # Since pulp-admin does NOT provide any mean to get specific repo information, # it's easier to access information via. pulp api calls. # Initiate pulp server calls relative_paths = [] hostname = socket.gethostname() server = PulpConnection(hostname) login_api = ActionsAPI(server) distributor_api = RepositoryDistributorAPI(server) login_api.login("admin", "admin") #TODO: hardcode for now, move as options later for id in repo_ids.split(","): print "Getting information on repo %s" % id distributor_response = distributor_api.distributors(id) if distributor_response != None or distributor_response.response_code == 200: distributors = distributor_response.response_body yum_distributor = [d for d in distributors if d['id'] == 'yum_distributor'][0] relative_path = yum_distributor['config']['relative_url'] relative_paths.append("%s/%s" % (source_path, relative_path)) else: print "Cannot find distributor for repo %s\n" % id print "Unable to determine relative path for repo %s\n" % id return relative_paths
class TestRepositoryDistributorAPI(unittest.TestCase): """ Tests for the RepositoryDistributorAPI. """ def setUp(self): self.api = RepositoryDistributorAPI(mock.MagicMock(spec=PulpConnection)) def test_update(self): # Setup expected_path = self.api.base_path % 'test-repo' + 'test-distributor/' expected_body = {'distributor_config': {'key': 'value'}, 'delta': {'auto_publish': True}} # Test result = self.api.update('test-repo', 'test-distributor', {'key': 'value'}, {'auto_publish': True}) self.api.server.PUT.assert_called_once_with(expected_path, expected_body) self.assertEqual(result, self.api.server.PUT.return_value)
def setUp(self): self.api = RepositoryDistributorAPI( mock.MagicMock(spec=PulpConnection))
def setUp(self): self.api = RepositoryDistributorAPI(mock.MagicMock(spec=PulpConnection))