fix: expand tests

Author: justinmerrell

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Change Log
 
+## Release 1.6.1 (TBD)
+
+### Added
+
+- User-Agent for better analytics tracking.
+
 ## Release 1.6.0 (1/29/24)
 
 ### Fixed

tests/test_serverless/test_worker.py

@@ -2,13 +2,15 @@
 # pylint: disable=protected-access
 
 import os
+import platform
 import argparse
 from unittest.mock import patch, mock_open, Mock, MagicMock
 
 from unittest import IsolatedAsyncioTestCase
 import nest_asyncio
 
 import runpod
+from runpod._version import __version__ as runpod_version
 from runpod.serverless.modules.rp_logger import RunPodLogger
 from runpod.serverless import _signal_handler
 
@@ -34,10 +36,14 @@ def test_get_auth_header(self):
         '''
         Test _get_auth_header
         '''
+        os_info = f"{platform.system()} {platform.release()}; {platform.machine()}"
         with patch("runpod.serverless.worker.os") as mock_os:
             mock_os.environ.get.return_value = "test"
             assert runpod.serverless.worker._get_auth_header(
-            ) == {'Authorization': 'test'}
+            ) == {
+                'Authorization': 'test',
+                'User-Agent': f'RunPod-Python-SDK/{runpod_version} ({os_info}) Language/Python {platform.python_version()}'  # pylint: disable=line-too-long
+            }
 
     def test_is_local(self):
         '''