Skip to content

Commit 4cf055f

Browse files
committed
test: refactor: replace patch.dict with monkeypatch for environment variable testing
1 parent b03edcc commit 4cf055f

File tree

1 file changed

+12
-13
lines changed

1 file changed

+12
-13
lines changed

tests/test_brew_file.py

Lines changed: 12 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
import logging
55
import os
66
from pathlib import Path
7-
from unittest.mock import patch
87

98
import pytest
109

@@ -74,23 +73,23 @@ def test_dryrun_banner(bf: BrewFile, caplog: pytest.LogCaptureFixture) -> None:
7473
]
7574

7675

77-
def test_parse_env_opts(bf: BrewFile) -> None:
78-
with patch.dict('os.environ', {'TEST_OPT': '--opt2=3 --opt3 opt4=4'}):
79-
opts = bf.parse_env_opts('TEST_OPT', {'--opt1': '1', '--opt2': '2'})
80-
assert opts == {
81-
'--opt1': '1',
82-
'--opt2': '3',
83-
'--opt3': '',
84-
'opt4': '4',
85-
}
76+
def test_parse_env_opts(bf: BrewFile, monkeypatch: pytest.MonkeyPatch) -> None:
77+
monkeypatch.setenv('TEST_OPT', '--opt2=3 --opt3 opt4=4')
78+
opts = bf.parse_env_opts('TEST_OPT', {'--opt1': '1', '--opt2': '2'})
79+
assert opts == {
80+
'--opt1': '1',
81+
'--opt2': '3',
82+
'--opt3': '',
83+
'opt4': '4',
84+
}
8685

8786

88-
def test_set_verbose(bf: BrewFile) -> None:
87+
def test_set_verbose(bf: BrewFile, monkeypatch: pytest.MonkeyPatch) -> None:
8988
bf.set_verbose()
9089
assert bf.opt['verbose'] == 'info'
9190
assert bf.log.getEffectiveLevel() == logging.INFO
92-
with patch.dict('os.environ', {'HOMEBREW_BREWFILE_VERBOSE': 'error'}):
93-
bf.set_verbose()
91+
monkeypatch.setenv('HOMEBREW_BREWFILE_VERBOSE', 'error')
92+
bf.set_verbose()
9493
assert bf.opt['verbose'] == 'error'
9594
assert bf.log.getEffectiveLevel() == logging.ERROR
9695
bf.set_verbose('0')

0 commit comments

Comments
 (0)