22import logging
33import sys
44from subprocess import CalledProcessError
5+ from typing import List
56
67import pytest
78from pytest import LogCaptureFixture , MonkeyPatch
@@ -91,7 +92,7 @@ def test_ensure_snow_installed_binary(monkeypatch: MonkeyPatch, caplog: LogCaptu
9192 monkeypatch .setattr ("builtins.__import__" , mock_failed_import )
9293
9394 # Mock run to return success
94- def mock_run (cmd : list [str ], ** kwargs ):
95+ def mock_run (cmd : List [str ], ** kwargs ):
9596 assert cmd == ["snow" , "--version" ]
9697 assert kwargs .get ("capture_output" ) is True
9798 assert kwargs .get ("check" ) is True
@@ -116,7 +117,7 @@ def test_ensure_snow_installed_nobinary(monkeypatch: MonkeyPatch, caplog: LogCap
116117 monkeypatch .setattr ("builtins.__import__" , mock_failed_import )
117118
118119 # Mock run to raise FileNotFoundError
119- def mock_run (cmd : list [str ], ** kwargs ):
120+ def mock_run (cmd : List [str ], ** kwargs ):
120121 if cmd == ["snow" , "--version" ]:
121122 raise FileNotFoundError ("No such file or directory: 'snow'" )
122123 return MockRunResult (returncode = 0 )
@@ -142,7 +143,7 @@ def test_ensure_snow_installed_failing_binary(monkeypatch: MonkeyPatch, caplog:
142143 monkeypatch .setattr ("builtins.__import__" , mock_failed_import )
143144
144145 # Mock run to raise CalledProcessError
145- def mock_run (cmd : list [str ], ** kwargs ):
146+ def mock_run (cmd : List [str ], ** kwargs ):
146147 if cmd == ["snow" , "--version" ]:
147148 raise CalledProcessError (returncode = 1 , cmd = cmd , output = "" , stderr = "Command failed with exit code 1" )
148149 return MockRunResult (returncode = 0 )
0 commit comments