Python get_args示例

示例#1

def test_get_args_fail():
    tests = [
        ['updown_service'],
        ['updown_service', 'myservice.name'],
        ['updown_service', 'myservice', 'up'],
        ['updown_service', 'myservice.name', 'wibble'],
    ]

    for argv in tests:
        with mock.patch('sys.argv', argv):
            with pytest.raises(SystemExit) as excinfo:
                updown_service.get_args()
            assert str(excinfo.value) == '2', argv

示例#2

文件： updown_service_test.py 项目： Yelp/nerve-tools

def test_get_args_fail():
    tests = [
        ['updown_service'],
        ['updown_service', 'myservice.name'],
        ['updown_service', 'myservice', 'up'],
        ['updown_service', 'myservice.name', 'wibble'],
    ]

    for argv in tests:
        with mock.patch('sys.argv', argv):
            with pytest.raises(SystemExit) as excinfo:
                updown_service.get_args()
            assert str(excinfo.value) == '2', argv

示例#3

def test_get_args_pass():
    tests = [
        [['updown_service', 'myservice.name', 'up'], 'up', None, 300, False,
         None],
        [['updown_service', 'myservice.name', 'down'], 'down', None, 300,
         False, None],
        [['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42,
         42, False, None],
        [['updown_service', 'myservice.name', 'down', '-x'], 'down', None, 300,
         True, None],
        [['updown_service', 'myservice.name:1234', 'down', '-x'], 'down', None,
         300, True, 1234],
    ]

    for test in tests:
        (argv, expected_state, expected_args_timeout, expected_timeout,
         expected_wait_only, expected_port) = test

        with mock.patch('sys.argv', argv):
            args = updown_service.get_args()
            timeout = updown_service._get_timeout_s(args.service, args.timeout)

        assert args.service == 'myservice.name'
        assert args.state == expected_state
        assert args.timeout == expected_args_timeout
        assert timeout == expected_timeout
        assert args.wait_only == expected_wait_only
        assert args.port == expected_port

示例#4

文件： updown_service_test.py 项目： dev-alex-alex2006hw/nerve-tools

def test_get_args_pass():
    tests = [
        [['updown_service', 'myservice.name', 'up'], 'up', None, 300],
        [['updown_service', 'myservice.name', 'down'], 'down', None, 300],
        [['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42, 42],
    ]

    for test in tests:
        argv, expected_state, expected_args_timeout, expected_timeout = test

        with mock.patch('sys.argv', argv):
            args = updown_service.get_args()
            timeout = updown_service._get_timeout_s(args.service, args.timeout)

        assert args.service == 'myservice.name'
        assert args.state == expected_state
        assert args.timeout == expected_args_timeout
        assert timeout == expected_timeout

示例#5

文件： updown_service_test.py 项目： hashbrowncipher/nerve-tools

def test_get_args_pass():
    tests = [
        [['updown_service', 'myservice.name', 'up'], 'up', None, 300],
        [['updown_service', 'myservice.name', 'down'], 'down', None, 300],
        [['updown_service', 'myservice.name', 'down', '-t', '42'], 'down', 42,
         42],
    ]

    for test in tests:
        argv, expected_state, expected_args_timeout, expected_timeout = test

        with mock.patch('sys.argv', argv):
            args = updown_service.get_args()
            timeout = updown_service._get_timeout_s(args.service, args.timeout)

        assert args.service == 'myservice.name'
        assert args.state == expected_state
        assert args.timeout == expected_args_timeout
        assert timeout == expected_timeout