def test_get_repositories(self):
# Setup
repos = [
{
'id': 'matching',
'notes': {pulp_constants.REPO_NOTE_TYPE_KEY: constants.REPO_NOTE_DOCKER, },
'importers': [
{'config': {}}
],
'distributors': [
{'id': constants.CLI_EXPORT_DISTRIBUTOR_ID},
{'id': constants.CLI_WEB_DISTRIBUTOR_ID}
]
},
{'id': 'non-rpm-repo',
'notes': {}}
]
self.context.server.repo.repositories.return_value.response_body = repos
# Test
command = ListDockerRepositoriesCommand(self.context)
repos = command.get_repositories({})
# Verify
self.assertEqual(1, len(repos))
self.assertEqual(repos[0]['id'], 'matching')
# Make sure two distributors exist
self.assertEqual(len(repos[0]['distributors']), 2)
def test_get_repositories_no_details(self):
# Setup
repos = [
{
'id': 'foo',
'display_name': 'bar',
'notes': {pulp_constants.REPO_NOTE_TYPE_KEY: constants.REPO_NOTE_DOCKER, }
}
]
self.context.server.repo.repositories.return_value.response_body = repos
# Test
command = ListDockerRepositoriesCommand(self.context)
repos = command.get_repositories({})
# Verify
self.assertEqual(1, len(repos))
self.assertEqual(repos[0]['id'], 'foo')
self.assertTrue('importers' not in repos[0])
self.assertTrue('distributors' not in repos[0])
