File: test_prompt.py

package info (click to toggle)
duckdb 1.5.1-2
  • links: PTS, VCS
  • area: main
  • in suites:
  • size: 299,196 kB
  • sloc: cpp: 865,414; ansic: 57,292; python: 18,871; sql: 12,663; lisp: 11,751; yacc: 7,412; lex: 1,682; sh: 747; makefile: 558
file content (50 lines) | stat: -rw-r--r-- 1,058 bytes parent folder | download | duplicates (3)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
# fmt: off

import pytest
import subprocess
import sys
from typing import List
from conftest import ShellTest


def test_prompt_unterminated_bracket(shell):
    test = (
        ShellTest(shell)
        .statement('.prompt {')
    )
    result = test.run()
    result.check_stderr("unterminated bracket")

def test_prompt_unterminated_escape(shell):
    test = (
        ShellTest(shell)
        .statement('.prompt \\')
    )
    result = test.run()
    result.check_stderr("unterminated")

def test_prompt_invalid_option(shell):
    test = (
        ShellTest(shell)
        .statement('.prompt "{yyy}"')
    )
    result = test.run()
    result.check_stderr("yyy")

def test_prompt_missing_query(shell):
    test = (
        ShellTest(shell)
        .statement('.prompt "{sql}"')
    )
    result = test.run()
    result.check_stderr("sql requires a parameter")

def test_prompt_invalid_color(shell):
    test = (
        ShellTest(shell)
        .statement('.prompt "{color:xxx}"')
    )
    result = test.run()
    result.check_stderr("xxx")

# fmt: on