testing

Author: bboynton97

tests/test_telemetry.py

@@ -1,7 +1,10 @@
 import os
 import unittest
-from agentstack.utils import get_telemetry_opt_out
+import uuid
+from unittest.mock import patch, mock_open
 
+from agentstack.telemetry import _get_cli_user_guid
+from agentstack.utils import get_telemetry_opt_out
 
 class TelemetryTest(unittest.TestCase):
     def test_telemetry_opt_out_env_var_set(self):
@@ -10,3 +13,70 @@ def test_telemetry_opt_out_env_var_set(self):
 
     def test_telemetry_opt_out_set_in_test_environment(self):
         assert get_telemetry_opt_out()
+
+    @patch('pathlib.Path.exists')
+    @patch('builtins.open', new_callable=mock_open, read_data='existing-guid')
+    def test_existing_guid_file(self, mock_file, mock_exists):
+        """Test when GUID file exists and can be read successfully"""
+        mock_exists.return_value = True
+
+        result = _get_cli_user_guid()
+
+        self.assertEqual(result, 'existing-guid')
+        mock_exists.assert_called_once_with()
+
+    @patch('pathlib.Path.exists')
+    @patch('pathlib.Path.mkdir')
+    @patch('uuid.uuid4')
+    @patch('builtins.open', new_callable=mock_open)
+    def test_create_new_guid(self, mock_file, mock_uuid, mock_mkdir, mock_exists):
+        """Test creation of new GUID when file doesn't exist"""
+        mock_exists.return_value = False
+        mock_uuid.return_value = uuid.UUID('12345678-1234-5678-1234-567812345678')
+
+        result = _get_cli_user_guid()
+
+        self.assertEqual(result, '12345678-1234-5678-1234-567812345678')
+        mock_exists.assert_called_once_with()
+        mock_mkdir.assert_called_once_with(parents=True, exist_ok=True)
+        handle = mock_file()
+        handle.write.assert_called_once_with('12345678-1234-5678-1234-567812345678')
+
+    @patch('pathlib.Path.exists')
+    @patch('builtins.open')
+    def test_permission_error_on_read(self, mock_file, mock_exists):
+        """Test handling of PermissionError when reading file"""
+        mock_exists.return_value = True
+        mock_file.side_effect = PermissionError()
+
+        result = _get_cli_user_guid()
+
+        self.assertEqual(result, 'unknown')
+        mock_exists.assert_called_once_with()
+
+    @patch('pathlib.Path.exists')
+    @patch('pathlib.Path.mkdir')
+    @patch('builtins.open')
+    def test_permission_error_on_write(self, mock_file, mock_mkdir, mock_exists):
+        """Test handling of PermissionError when writing new file"""
+        mock_exists.return_value = False
+        mock_file.side_effect = PermissionError()
+
+        result = _get_cli_user_guid()
+
+        self.assertEqual(result, 'unknown')
+        mock_exists.assert_called_once_with()
+        mock_mkdir.assert_called_once_with(parents=True, exist_ok=True)
+
+    @patch('pathlib.Path.exists')
+    @patch('pathlib.Path.mkdir')
+    def test_os_error_on_mkdir(self, mock_mkdir, mock_exists):
+        """Test handling of OSError when creating directory"""
+        mock_exists.return_value = False
+        mock_mkdir.side_effect = OSError()
+
+        result = _get_cli_user_guid()
+
+        self.assertEqual(result, 'unknown')
+        mock_exists.assert_called_once_with()
+        mock_mkdir.assert_called_once_with(parents=True, exist_ok=True)

tests/test_update.py

@@ -10,7 +10,6 @@
     _is_ci_environment,
     CI_ENV_VARS,
     should_update,
-    _get_base_dir,
     get_latest_version,
     AGENTSTACK_PACKAGE,
     load_update_data,
@@ -46,32 +45,6 @@ def test_updates_disabled_by_env_var_in_test(self):
         with patch.dict('os.environ', {'AGENTSTACK_UPDATE_DISABLE': 'true'}):
             self.assertFalse(should_update())
 
-    @patch('agentstack.update.user_data_dir')
-    def test_get_base_dir_writable(self, mock_user_data_dir):
-        """
-        Test that _get_base_dir() returns a writable Path when user_data_dir is accessible.
-        """
-        mock_path = '/mock/user/data/dir'
-        mock_user_data_dir.return_value = mock_path
-
-        result = _get_base_dir()
-
-        self.assertIsInstance(result, Path)
-        self.assertTrue(result.is_absolute())
-
-    @patch('agentstack.update.user_data_dir')
-    def test_get_base_dir_not_writable(self, mock_user_data_dir):
-        """
-        Test that _get_base_dir() falls back to a temporary directory when user_data_dir is not writable.
-        """
-        mock_user_data_dir.side_effect = PermissionError
-
-        result = _get_base_dir()
-
-        self.assertIsInstance(result, Path)
-        self.assertTrue(result.is_absolute())
-        self.assertIn(str(result), ['/tmp', os.environ.get('TEMP', '/tmp')])
-
     def test_get_latest_version(self):
         """
         Test that get_latest_version returns a valid Version object from the actual PyPI.

tests/test_utils.py

@@ -1,6 +1,14 @@
+import os
 import unittest
+from pathlib import Path
+from unittest.mock import patch
 
-from agentstack.utils import clean_input, is_snake_case, validator_not_empty
+from agentstack.utils import (
+    clean_input,
+    is_snake_case,
+    validator_not_empty,
+    get_base_dir
+)
 from inquirer import errors as inquirer_errors
 
 
@@ -37,3 +45,28 @@ def test_validator_not_empty(self):
         with self.assertRaises(inquirer_errors.ValidationError):
             validator(None, "ab")
 
+    @patch('agentstack.utils.user_data_dir')
+    def test_get_base_dir_not_writable(self, mock_user_data_dir):
+        """
+        Test that get_base_dir() falls back to a temporary directory when user_data_dir is not writable.
+        """
+        mock_user_data_dir.side_effect = PermissionError
+
+        result = get_base_dir()
+
+        self.assertIsInstance(result, Path)
+        self.assertTrue(result.is_absolute())
+        self.assertIn(str(result), ['/tmp', os.environ.get('TEMP', '/tmp')])
+
+    @patch('agentstack.utils.user_data_dir')
+    def test_get_base_dir_writable(self, mock_user_data_dir):
+        """
+        Test that get_base_dir() returns a writable Path when user_data_dir is accessible.
+        """
+        mock_path = '/mock/user/data/dir'
+        mock_user_data_dir.return_value = mock_path
+
+        result = get_base_dir()
+
+        self.assertIsInstance(result, Path)
+        self.assertTrue(result.is_absolute())