Skip to content

Commit 06bb664

Browse files
committed
test: simplify transaction_indicator mock setup in tests
Use direct attribute assignment on MagicMock instead of type() to avoid polluting the global MagicMock class state.
1 parent f9e4323 commit 06bb664

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

tests/test_main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -596,7 +596,7 @@ def test_get_prompt_with_transaction_status(transaction_indicator, expected):
596596
cli.pgexecute.port = 5432
597597
cli.pgexecute.pid = 12345
598598
cli.pgexecute.superuser = False
599-
type(cli.pgexecute).transaction_indicator = mock.PropertyMock(return_value=transaction_indicator)
599+
cli.pgexecute.transaction_indicator = transaction_indicator
600600

601601
result = cli.get_prompt("\\T\\u")
602602
assert result == expected
@@ -612,7 +612,7 @@ def test_get_prompt_transaction_status_in_full_prompt():
612612
cli.pgexecute.port = 5432
613613
cli.pgexecute.pid = 12345
614614
cli.pgexecute.superuser = False
615-
type(cli.pgexecute).transaction_indicator = mock.PropertyMock(return_value="*")
615+
cli.pgexecute.transaction_indicator = "*"
616616

617617
result = cli.get_prompt("\\T\\u@\\h:\\d> ")
618618
assert result == "*user@db.example.com:mydb> "

0 commit comments

Comments
 (0)