Python Discovery.select_repositories示例

示例#1

文件： repo_create_test.py 项目： jsomara/katello

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)

示例#2

文件： repo_create_test.py 项目： beav/katello

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)