tests/test_prompt_utils.py


1
2
3
4
5
6
7
8
9
10
11
12
13
14

# -*- coding: utf-8 -*-


import click

from pgcli.packages.prompt_utils import confirm_destructive_query


def test_confirm_destructive_query_notty():
    stdin = click.get_text_stream('stdin')
    assert stdin.isatty() is False

    sql = 'drop database foo;'
    assert confirm_destructive_query(sql) is None