fix: pylint errors

Author: matthew-on-git

tests/test_api.py

@@ -1,28 +1,22 @@
 """Unit tests for Bitbucket API logic in manage_bitbucket_env.py."""
+from manage_bitbucket_env import get_environment_uuid, get_variables
 import unittest
 from unittest.mock import patch, Mock
 import sys
 import os
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from manage_bitbucket_env import get_environment_uuid, get_variables
 
 class TestAPI(unittest.TestCase):
     """Test API-related functions."""
-    logger = None
-    auth = None
-    workspace = None
-    repo_slug = None
-    deployment_name = None
-    env_uuid = None
 
     def setUp(self):
         """Set up test fixtures."""
-        self.logger = Mock()
-        self.auth = Mock()
-        self.workspace = 'ws'
-        self.repo_slug = 'repo'
-        self.deployment_name = 'env'
-        self.env_uuid = 'uuid-123'
+        self.logger: Mock = Mock()
+        self.auth: Mock = Mock()
+        self.workspace: str = 'ws'
+        self.repo_slug: str = 'repo'
+        self.deployment_name: str = 'env'
+        self.env_uuid: str = 'uuid-123'
 
     @patch('manage_bitbucket_env.requests.get')
     def test_get_environment_uuid_found(self, mock_get):

tests/test_arg_parser.py

@@ -1,10 +1,10 @@
 """Unit tests for argument parsing in manage_bitbucket_env.py."""
+from manage_bitbucket_env import arg_parser
 import unittest
 from unittest.mock import patch
 import sys
 import os
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from manage_bitbucket_env import arg_parser
 
 class TestArgParser(unittest.TestCase):
     """Test argument parsing functionality."""

tests/test_export_import.py

@@ -1,33 +1,27 @@
 """Unit tests for export/import logic in manage_bitbucket_env.py."""
 # pylint: disable=duplicate-code
+from manage_bitbucket_env import (
+    export_variables, export_all_variables, export_secure_keys, import_variables
+)
 import unittest
 from unittest.mock import patch, Mock
 import sys
 import os
 import tempfile
 import json
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from manage_bitbucket_env import (
-    export_variables, export_all_variables, export_secure_keys, import_variables
-)
 
 class TestExportImport(unittest.TestCase):
     """Test export and import functionality."""
-    logger = None
-    auth = None
-    workspace = None
-    repo_slug = None
-    deployment_name = None
-    env_uuid = None
 
     def setUp(self):
         """Set up test fixtures."""
-        self.logger = Mock()
-        self.auth = Mock()
-        self.workspace = 'ws'
-        self.repo_slug = 'repo'
-        self.deployment_name = 'env'
-        self.env_uuid = 'uuid-123'
+        self.logger: Mock = Mock()
+        self.auth: Mock = Mock()
+        self.workspace: str = 'ws'
+        self.repo_slug: str = 'repo'
+        self.deployment_name: str = 'env'
+        self.env_uuid: str = 'uuid-123'
 
     @patch('manage_bitbucket_env.get_environment_uuid')
     @patch('manage_bitbucket_env.get_variables')

tests/test_main.py

@@ -1,18 +1,18 @@
 """Unit tests for main function integration in manage_bitbucket_env.py."""
+# pylint: disable=unused-argument
+from manage_bitbucket_env import main
 import unittest
 from unittest.mock import patch, Mock
 import sys
 import os
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from manage_bitbucket_env import main
 
 class TestMainFunction(unittest.TestCase):
     """Test main function integration."""
-    mock_logger = None
 
     def setUp(self):
         """Set up test fixtures."""
-        self.mock_logger = Mock()
+        self.mock_logger: Mock = Mock()
 
     @patch('manage_bitbucket_env.arg_parser')
     @patch('manage_bitbucket_env.load_dotenv')
@@ -29,10 +29,11 @@ def test_main_export_variables(self, mock_export, mock_auth, mock_env_get,
         mock_args.repo_slug = 'test-repo'
         mock_args.deployment_name = 'test-deployment'
         mock_arg_parser.return_value = mock_args
-        mock_env_get.side_effect = lambda x: {
+        env_dict = {
             'BITBUCKET_USERNAME': 'test_user',
             'BITBUCKET_APP_PASSWORD': 'test_password'
-        }.get(x)
+        }
+        mock_env_get.side_effect = env_dict.get
         mock_auth_instance = Mock()
         mock_auth.return_value = mock_auth_instance
         main(self.mock_logger)

tests/test_update_vars.py

@@ -1,27 +1,22 @@
 """Unit tests for update_vars logic in manage_bitbucket_env.py."""
 # pylint: disable=duplicate-code
+from manage_bitbucket_env import update_vars
 import unittest
 from unittest.mock import patch, Mock
 import sys
 import os
 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)))
-from manage_bitbucket_env import update_vars
 
 class TestUpdateVars(unittest.TestCase):
     """Test variable update functionality."""
-    logger = None
-    auth = None
-    workspace = None
-    repo_slug = None
-    env_uuid = None
 
     def setUp(self):
         """Set up test fixtures."""
-        self.logger = Mock()
-        self.auth = Mock()
-        self.workspace = 'ws'
-        self.repo_slug = 'repo'
-        self.env_uuid = 'uuid-123'
+        self.logger: Mock = Mock()
+        self.auth: Mock = Mock()
+        self.workspace: str = 'ws'
+        self.repo_slug: str = 'repo'
+        self.env_uuid: str = 'uuid-123'
 
     @patch('manage_bitbucket_env.requests.put')
     def test_update_existing(self, mock_put):