class RepositorySelectionTest(unittest.TestCase): DISCOVERED_URLS = ['http://localhost/a/b/', 'http://localhost/a/c/'] def setUp(self): self.create_action = Discovery() self.original_system_exit = katello.client.core.repo.system_exit katello.client.core.repo.system_exit = Mock() def tearDown(self): katello.client.core.repo.system_exit = self.original_system_exit def test_q_forces_exit(self): raw_input_stub = RawInputStub(['q']) self.create_action.select_repositories(self.DISCOVERED_URLS, False, raw_input_stub.raw_input) katello.client.core.repo.system_exit.assert_called_once def test_a_y_adds_all_discovered_repos(self): raw_input_stub = RawInputStub(['a', 'y']) selected_repos = self.create_action.select_repositories(self.DISCOVERED_URLS, False, raw_input_stub.raw_input) self.assertEqual(selected_repos, self.DISCOVERED_URLS) def test_1_y_adds_first_discovered_repo(self): raw_input_stub = RawInputStub(['1', 'y']) selected_repos = self.create_action.select_repositories(self.DISCOVERED_URLS, False, raw_input_stub.raw_input) self.assertEqual(selected_repos, [self.DISCOVERED_URLS[0]]) def test_assumeyes_adds_all_discovered_repos(self): selected_repos = self.create_action.select_repositories(self.DISCOVERED_URLS, True) self.assertEqual(selected_repos, self.DISCOVERED_URLS)
def setUp(self): self.set_action(Discovery()) self.set_module(katello.client.core.repo) self.mock(self.module, 'run_spinner_in_bg', [self.RESULT]) self.mock(self.module, 'system_exit') self.mock(self.action.api, 'repo_discovery', self.DISCOVERY_TASK)
def setUp(self): self.set_action(Discovery()) self.set_module(katello.client.core.repo) self.mock(self.module, 'run_spinner_in_bg', [self.RESULT]) self.mock(self.module, 'system_exit') self.mock(self.action.provider_api, 'provider', {'id':self.PROVIDER_ID, 'discovered_repos':self.URLS}) self.mock(self.action.provider_api, 'repo_discovery', self.DISCOVERY_TASK)
class RepositoryNameTest(unittest.TestCase): NAME = 'REPO' URL = 'http://localhost/a/b/' def setUp(self): self.create_action = Discovery() self.parsedUrl = urlparse.urlparse(self.URL) def test_replaces_slashes_with_underscores(self): self.assertEqual(self.create_action.repository_name(self.NAME, self.parsedUrl.path), "REPO_a_b_")
class CreateRepositoryTest(unittest.TestCase): ORGANIZATION = 'ACME_Corporation' PRODUCT_ID = '123' NAME = 'REPO' URL = 'http://localhost/a/b/' URL2 = 'http://localhost/a/c/' def setUp(self): self.create_action = Discovery() self.create_action.api.create = Mock() def test_create_repo_in_pulp(self): self.create_action.create_repositories(self.ORGANIZATION, self.PRODUCT_ID, self.NAME, [self.URL]) parsedUrl = urlparse.urlparse(self.URL) self.create_action.api.create.assert_called_once_with(self.ORGANIZATION, self.PRODUCT_ID, self.create_action.repository_name(self.NAME, parsedUrl.path), self.URL, None, None) def test_creates_repos_in_pulp_for_all_urls(self): self.create_action.create_repositories(self.ORGANIZATION, self.PRODUCT_ID, self.NAME, [self.URL, self.URL2]) self.create_action.api.create.assert_called_twice
class RequiredCLIOptionsTests(CLIOptionTestCase): action = Discovery() disallowed_options = [ ('--name=repo1', '--url=http://localhost', '--product=product1'), ('--org=ACME', '--name=repo1', '--url=http://localhost'), ('--org=ACME', '--url=http://localhost', '--product=product1'), ('--org=ACME', '--name=repo1', '--product=product1') ] allowed_options = [ ('--org=ACME', '--name=repo1', '--url=http://localhost', '--product=product1') ]
class RequiredCLIOptionsTests(CLIOptionTestCase): action = Discovery() disallowed_options = [ ('--name=repo1', '--url=http://localhost', '--product=product1'), ('--org=ACME', '--name=repo1', '--url=http://localhost'), ('--org=ACME', '--url=http://localhost', '--product=product1'), ('--org=ACME', '--name=repo1', '--product=product1') ] allowed_options = [ ('--org=ACME', '--name=repo1', '--url=http://localhost', '--product=product1', '--provider=foo'), ('--org=ACME', '--name=repo1', '--url=https://localhost', '--product=product1', '--provider=foo'), ('--org=ACME', '--name=repo1', '--url=ftp://localhost', '--product=product1', '--provider=foo'), ('--org=ACME', '--name=repo1', '--url=file:///a/b/c/', '--product=product1', '--provider=foo'), ('--org=ACME', '--name=repo1', '--url=file:///a/b/c/', '--product=product1', '--provider=foo', '--unprotected'), ]
def setUp(self): self.create_action = Discovery() self.original_system_exit = katello.client.core.repo.system_exit katello.client.core.repo.system_exit = Mock()
def setUp(self): self.create_action = Discovery() self.create_action.api.create = Mock()
def setUp(self): self.create_action = Discovery() self.parsedUrl = urlparse.urlparse(self.URL)