17 lines
523 B
Python
17 lines
523 B
Python
import click
|
|
|
|
from pgcli.packages.prompt_utils import confirm_destructive_query
|
|
|
|
|
|
def test_confirm_destructive_query_notty():
|
|
stdin = click.get_text_stream("stdin")
|
|
if not stdin.isatty():
|
|
sql = "drop database foo;"
|
|
assert confirm_destructive_query(sql, [], None) is None
|
|
|
|
|
|
def test_confirm_destructive_query_with_alias():
|
|
stdin = click.get_text_stream("stdin")
|
|
if not stdin.isatty():
|
|
sql = "drop database foo;"
|
|
assert confirm_destructive_query(sql, ["drop"], "test") is None
|