File: test_application_name.py

package info (click to toggle)
pgcli 4.3.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky
  • size: 1,476 kB
  • sloc: python: 10,390; sh: 67; makefile: 8
file content (17 lines) | stat: -rw-r--r-- 510 bytes parent folder | download | duplicates (2) 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
 
from unittest.mock import patch

from click.testing import CliRunner

from pgcli.main import cli
from pgcli.pgexecute import PGExecute


def test_application_name_in_env():
    runner = CliRunner()
    app_name = "wonderful_app"
    with patch.object(PGExecute, "__init__") as mock_pgxecute:
        runner.invoke(
            cli, ["127.0.0.1:5432/hello", "user"], env={"PGAPPNAME": app_name}
        )
        kwargs = mock_pgxecute.call_args.kwargs
        assert kwargs.get("application_name") == app_name