Python State.do_testの例

コード例 #1

ファイル: test_line_info.py プロジェクト: fossabot/sqlwhat

def test_line_info(sql_cmd, start, pos):
    state = State(sql_cmd,
                  "",
                  "",
                  None,
                  None, {}, {},
                  Reporter(),
                  ast_dispatcher=Dispatcher.from_module(
                      PARSER_MODULES['mssql']))
    try:
        state.do_test("failure message")
    except TF as tf:
        for ii, k in enumerate(pos_names):
            assert tf.payload[k] == pos[ii]

コード例 #2

def test_state_line_info():
    state = State("\nSELECT x FROM y",
                  "SELECT x FROM y",
                  "",
                  None,
                  None, {}, {},
                  Reporter(),
                  ast_dispatcher=Dispatcher.from_module(
                      PARSER_MODULES['mssql']))

    with pytest.raises(TF):
        state.do_test("failure message")

    payload = state.reporter.build_payload()

    pos = [2, 1, 2, 15]
    pos_names = ['line_start', 'column_start', 'line_end', 'column_end']
    for ii, k in enumerate(pos_names):
        assert payload[k] == pos[ii]