test_handler.py

# Copyright (c) Microsoft Corporation.
# Licensed under the MIT License.

"""CLI handler test."""

import unittest

from knack.util import CLIError

import superbench.cli._handler as cli_handler


class CLIHandlerTestCase(unittest.TestCase):
    """A class for CLI handler test cases."""
    def test_split_docker_domain(self):
        """Test split_docker_domain function which splits Docker image name to domain and remainder part.

        Test cases are ported from
        https://github.com/distribution/distribution/blob/v2.7.1/reference/normalize_test.go#L468-L528.
        """
        test_cases = [
            {
                'input': 'test.com/foo',
                'domain': 'test.com',
                'name': 'foo',
            },
            {
                'input': 'test_com/foo',
                'domain': 'docker.io',
                'name': 'test_com/foo',
            },
            {
                'input': 'docker/migrator',
                'domain': 'docker.io',
                'name': 'docker/migrator',
            },
            {
                'input': 'test.com:8080/foo',
                'domain': 'test.com:8080',
                'name': 'foo',
            },
            {
                'input': 'test-com:8080/foo',
                'domain': 'test-com:8080',
                'name': 'foo',
            },
            {
                'input': 'foo',
                'domain': 'docker.io',
                'name': 'library/foo',
            },
            {
                'input': 'xn--n3h.com/foo',
                'domain': 'xn--n3h.com',
                'name': 'foo',
            },
            {
                'input': 'xn--n3h.com:18080/foo',
                'domain': 'xn--n3h.com:18080',
                'name': 'foo',
            },
            {
                'input': 'docker.io/foo',
                'domain': 'docker.io',
                'name': 'library/foo',
            },
            {
                'input': 'docker.io/library/foo',
                'domain': 'docker.io',
                'name': 'library/foo',
            },
            {
                'input': 'docker.io/library/foo/bar',
                'domain': 'docker.io',
                'name': 'library/foo/bar',
            },
        ]
        for test_case in test_cases:
            with self.subTest(msg='Testing with case', test_case=test_case):
                domain, name = cli_handler.split_docker_domain(test_case['input'])
                self.assertEqual(domain, test_case['domain'])
                self.assertEqual(name, test_case['name'])

    def test_process_runner_arguments_container_name(self):
        """Test process_runner_arguments saves custom container name."""
        docker_config, _, _, _ = cli_handler.process_runner_arguments(
            host_list='localhost',
            container_name='sb-custom',
        )
        self.assertEqual(docker_config.container_name, 'sb-custom')

    def test_process_runner_arguments_empty_container_name(self):
        """Test process_runner_arguments rejects empty container name."""
        with self.assertRaises(CLIError):
            cli_handler.process_runner_arguments(
                host_list='localhost',
                container_name='   ',
            )