def test_get_repositories(self): # Setup repos = [{ 'id': 'matching', 'notes': { REPO_NOTE_TYPE_KEY: constants.REPO_NOTE_OSTREE, }, 'importers': [{ 'config': {} }], 'distributors': [{ 'id': constants.CLI_WEB_DISTRIBUTOR_ID }] }, { 'id': 'non-rpm-repo', 'notes': {} }] self.context.server.repo.repositories.return_value.response_body = repos # Test command = cudl.ListOSTreeRepositoriesCommand(self.context) repos = command.get_repositories({}) # Verify self.assertEqual(1, len(repos)) self.assertEqual(repos[0]['id'], 'matching')
def test_get_all_repos(self): self.context.server.repo.repositories.return_value.response_body = 'foo' command = cudl.ListOSTreeRepositoriesCommand(self.context) result = command._all_repos({'bar': 'baz'}) self.context.server.repo.repositories.assert_called_once_with( {'bar': 'baz'}) self.assertEquals('foo', result)
def test_get_repositories_no_details(self): # Setup repos = [{ 'id': 'foo', 'display_name': 'bar', 'notes': { REPO_NOTE_TYPE_KEY: constants.REPO_NOTE_OSTREE, } }] self.context.server.repo.repositories.return_value.response_body = repos # Test command = cudl.ListOSTreeRepositoriesCommand(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])
def test_get_all_repos_caches_results(self): command = cudl.ListOSTreeRepositoriesCommand(self.context) command.all_repos_cache = 'foo' result = command._all_repos({'bar': 'baz'}) self.assertFalse(self.context.server.repo.repositories.called) self.assertEquals('foo', result)