Skip to content

Commit ad9ccdd

Browse files
committed
Update test_worker.py
1 parent ef1a231 commit ad9ccdd

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

tests/test_serverless/test_worker.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -39,10 +39,11 @@ def test_get_auth_header(self):
3939
os_info = f"{platform.system()} {platform.release()}; {platform.machine()}"
4040
with patch("runpod.serverless.worker.os") as mock_os:
4141
mock_os.environ.get.return_value = "test"
42+
mock_os.getenv.return_value = "test"
4243
assert runpod.serverless.worker._get_auth_header(
4344
) == {
4445
'Authorization': 'test',
45-
'User-Agent': f'RunPod-Python-SDK/{runpod_version} ({os_info}) Language/Python {platform.python_version()}' # pylint: disable=line-too-long
46+
'User-Agent': f'RunPod-Python-SDK/{runpod_version} ({os_info}) Language/Python {platform.python_version()} Integration/test' # pylint: disable=line-too-long
4647
}
4748

4849
def test_is_local(self):

0 commit comments

Comments
 (0)