devhub-python/tests/test_client.py at 8eb87f6fef8ca6c3a86b2698f9043b24cca4b773 · Devotel/devhub-python · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
from unittest.mock import Mock, patch

import pytest
import requests

from devo_global_comms_python import DevoClient
from devo_global_comms_python.exceptions import DevoAPIException, DevoAuthenticationException


class TestDevoClient:
    """Test cases for the DevoClient class."""

    def test_client_initialization_with_api_key(self, api_key):
        """Test client initialization with API key."""
        client = DevoClient(api_key=api_key)

        assert client.base_url == DevoClient.DEFAULT_BASE_URL
        assert client.timeout == DevoClient.DEFAULT_TIMEOUT
        assert client.auth.api_key == api_key

    def test_client_initialization_with_custom_params(self, api_key):
        """Test client initialization with custom parameters."""
        timeout = 60.0
        max_retries = 5

        client = DevoClient(api_key=api_key, timeout=timeout, max_retries=max_retries)

        assert client.base_url == DevoClient.DEFAULT_BASE_URL
        assert client.timeout == timeout

    def test_client_has_all_resources(self, api_key):
        """Test that client has all expected resources."""
        client = DevoClient(api_key=api_key)

        assert hasattr(client, "sms")
        assert hasattr(client, "email")
        assert hasattr(client, "whatsapp")
        assert hasattr(client, "rcs")
        assert hasattr(client, "contacts")
        assert hasattr(client, "messages")

    @patch("requests.Session.request")
    def test_successful_request(self, mock_request, api_key):
        """Test successful API request."""
        # Setup mock response
        mock_response = Mock()
        mock_response.ok = True
        mock_response.json.return_value = {"success": True}
        mock_request.return_value = mock_response

        client = DevoClient(api_key=api_key)
        response = client.get("test/endpoint")

        assert response == mock_response
        mock_request.assert_called_once()

    @patch("requests.Session.request")
    def test_api_error_handling(self, mock_request, api_key):
        """Test API error handling."""
        # Setup mock error response
        mock_response = Mock()
        mock_response.ok = False
        mock_response.status_code = 400
        mock_response.json.return_value = {
            "message": "Bad Request",
            "code": "INVALID_PARAMS",
        }
        mock_request.return_value = mock_response

        client = DevoClient(api_key=api_key)

        with pytest.raises(DevoAPIException) as exc_info:
            client.get("test/endpoint")

        assert exc_info.value.status_code == 400
        assert "Bad Request" in str(exc_info.value)

    @patch("requests.Session.request")
    def test_authentication_error_handling(self, mock_request, api_key):
        """Test authentication error handling."""
        # Setup mock auth error response
        mock_response = Mock()
        mock_response.ok = False
        mock_response.status_code = 401
        mock_response.json.return_value = {"message": "Unauthorized"}
        mock_request.return_value = mock_response

        client = DevoClient(api_key=api_key)

        with pytest.raises(DevoAuthenticationException):
            client.get("test/endpoint")

    @patch("requests.Session.request")
    def test_timeout_error_handling(self, mock_request, api_key):
        """Test timeout error handling."""
        mock_request.side_effect = requests.exceptions.Timeout()

        client = DevoClient(api_key=api_key)

        with pytest.raises(Exception) as exc_info:
            client.get("test/endpoint")

        assert "timed out" in str(exc_info.value).lower()

    def test_http_methods(self, api_key):
        """Test HTTP method shortcuts."""
        client = DevoClient(api_key=api_key)

        with patch.object(client, "request") as mock_request:
            client.get("path")
            mock_request.assert_called_with("GET", "path")

            client.post("path", json={"data": "test"})
            mock_request.assert_called_with("POST", "path", json={"data": "test"})

            client.put("path")
            mock_request.assert_called_with("PUT", "path")

            client.delete("path")
            mock_request.assert_called_with("DELETE", "path")

            client.patch("path")
            mock_request.assert_called_with("PATCH", "path")