Fix sanity tests

Author: abikouo

plugins/plugin_utils/ssm_file_transfer.py

@@ -17,6 +17,7 @@
 from typing import Dict
 from typing import List
 from typing import NoReturn
+from typing import Optional
 from typing import Union
 
 from ansible.errors import AnsibleConnectionFailure
@@ -27,7 +28,7 @@
 
 
 class ConnectionPluginDisplay:
-    def __init__(self, host: str | None) -> None:
+    def __init__(self, host: Optional[str]) -> None:
         self._host_args = {}
         if host:
             self._host_args = {"host": host}
@@ -111,10 +112,10 @@ def __init__(
         instance_id: str,
         executable: str,
         ssm_timeout: int,
-        region_name: str | None,
+        region_name: Optional[str],
         profile_name: str,
-        document_name: str | None = None,
-        parameters: Dict[str, List[str]] | None = None,
+        document_name: Optional[str] = None,
+        parameters: Optional[Dict[str, List[str]]] = None,
     ) -> None:
         self._client = ssm_client
         self._session = None
@@ -178,15 +179,15 @@ class PortForwardingFileTransferManager:
 
     def __init__(
         self,
-        host: str | None,
+        host: Optional[str],
         ssm_client: Any,
         instance_id: str,
         executable: str,
         ssm_timeout: int,
-        region_name: str | None,
+        region_name: Optional[str],
         profile_name: str,
-        host_port: int | None = None,
-        local_port: int | None = None,
+        host_port: Optional[int],
+        local_port: Optional[int],
     ) -> None:
         self._client = ssm_client
         self._session = None
@@ -244,7 +245,7 @@ def _socket_connect(self, session: Any, port: int, host: str = "localhost", max_
                     raise
                 time.sleep(0.05)
 
-    def _socket_read(self, port: int, out_path: str | None = None) -> None:
+    def _socket_read(self, port: int, out_path: Optional[str] = None) -> None:
         self._display.vvvv(f"Read content from socket on port '{port}'...")
         with socket.socket(socket.AF_INET, socket.SOCK_STREAM) as session:
             session.settimeout(1)

tests/unit/plugins/connection/aws_ssm/test_aws_ssm.py

@@ -43,6 +43,8 @@ def mock_get_option(key):
 
         # Mock the _initialize_ssm_client and _initialize_s3_client methods
         conn._initialize_ssm_client = MagicMock()
+        conn._use_bucket = MagicMock()
+        conn._use_bucket.return_value = True
 
         conn._init_clients()
 
@@ -196,6 +198,8 @@ def test_plugins_connection_aws_ssm_put_file(self, mock_ospe):
         conn._connect = MagicMock()
         conn._file_transport_command = MagicMock()
         conn._file_transport_command.return_value = (0, "stdout", "stderr")
+        conn._use_bucket = MagicMock()
+        conn._use_bucket.return_value = True
         conn.put_file("/in/file", "/out/file")
 
     def test_plugins_connection_aws_ssm_fetch_file(self):
@@ -205,6 +209,8 @@ def test_plugins_connection_aws_ssm_fetch_file(self):
         conn._connect = MagicMock()
         conn._file_transport_command = MagicMock()
         conn._file_transport_command.return_value = (0, "stdout", "stderr")
+        conn._use_bucket = MagicMock()
+        conn._use_bucket.return_value = True
         conn.fetch_file("/in/file", "/out/file")
 
     @patch("subprocess.check_output")