more tests

tests/app/clients/test_aws_pinpoint.py

@@ -0,0 +1,40 @@
+from unittest.mock import MagicMock
+
+from aiohttp import ClientError
+from flask import current_app
+
+from app.clients.pinpoint.aws_pinpoint import AwsPinpointClient
+
+
+def test_validate_phone_number_success():
+    mock_response = {
+        "NumberValidateResponse": {
+            "PhoneType": "MOBILE",
+            "CleansedPhoneNumberE164": "+1234567890",
+        }
+    }
+    client_instance = AwsPinpointClient()
+    client_instance._client = MagicMock()
+    client_instance._client.phone_number_validate.return_value = mock_response
+
+    result = client_instance.validate_phone_number("US", "+1234567890")
+    assert result == "Foo"
+    client_instance._client.phone_number_validate.assert_called_once_with(
+        {"IsoCountryCode": "US", "PhoneNumber": "+1234567890"}
+    )
+    current_app.logger.info.assert_called_once()
+
+
+def test_validate_phone_number_client_error():
+    client_instance = AwsPinpointClient()
+    client_instance._client = MagicMock()
+    client_instance._client.phone_number_validate.side_effect = ClientError(
+        {"Error": {"Code": "BadRequest1", "MEssage": "Invalid phone"}},
+        "phone number validate",
+    )
+
+    result = client_instance.validate_phone_number("US", "bad-number")
+    assert result == "Foo"
+    current_app.logger.exception.assert_called_once_with(
+        "#notify-debug-validate-phone-number Could not validate with pinpoint"
+    )