diff options
Diffstat (limited to 'tests')
-rw-r--r-- | tests/test_main.py | 16 |
1 files changed, 16 insertions, 0 deletions
diff --git a/tests/test_main.py b/tests/test_main.py index 9635fc04..cbf20a6a 100644 --- a/tests/test_main.py +++ b/tests/test_main.py @@ -297,6 +297,22 @@ def test_i_works(tmpdir, executor): @dbtest +def test_echo_works(executor): + cli = PGCli(pgexecute=executor) + statement = r"\echo asdf" + result = run(executor, statement, pgspecial=cli.pgspecial) + assert result == ["asdf"] + + +@dbtest +def test_qecho_works(executor): + cli = PGCli(pgexecute=executor) + statement = r"\qecho asdf" + result = run(executor, statement, pgspecial=cli.pgspecial) + assert result == ["asdf"] + + +@dbtest def test_watch_works(executor): cli = PGCli(pgexecute=executor) |