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))
