Python Parser.add_command 예제들

예제 #1

파일: test_cli.py 프로젝트: demas/stackify

def test_in_list_command_not_ok(message):
    func = mock.MagicMock()
    c = Command(command_type=CommandType.IN_LIST,
                value=["a", "b", "c"],
                action=func)
    parser = Parser()
    parser.add_command(c)

    parser.process_message(message)

    assert not func.called

예제 #2

파일: test_cli.py 프로젝트: demas/stackify

def test_call_first_command():
    func = mock.MagicMock()
    c1 = Command(command_type=CommandType.EQUAL, value="a", action=func.foo)
    c2 = Command(command_type=CommandType.START_WITH,
                 value="a",
                 action=func.bar)
    parser = Parser()
    parser.add_command(c1)
    parser.add_command(c2)

    parser.process_message("a")

    assert func.foo.called
    assert not func.bar.called