|
| 1 | +from unittest.mock import MagicMock |
| 2 | + |
| 3 | +from dimo.dimo import DIMO |
| 4 | + |
| 5 | + |
| 6 | +def test_get_full_path_no_params(): |
| 7 | + client = DIMO(env="Dev") |
| 8 | + result = client._get_full_path("Valuations", "/v2/vehicles/1234/valuations") |
| 9 | + assert result == "https://valuations-api.dev.dimo.zone/v2/vehicles/1234/valuations" |
| 10 | + |
| 11 | + |
| 12 | +def test_get_full_path_with_params(): |
| 13 | + client = DIMO(env="Dev") |
| 14 | + result = client._get_full_path( |
| 15 | + "Telemetry", |
| 16 | + "/items/:item_id", |
| 17 | + {"item_id": 123, "detail_id": "abc"}, |
| 18 | + ) |
| 19 | + assert result == "https://telemetry-api.dev.dimo.zone/query/items/123" |
| 20 | + |
| 21 | + |
| 22 | +def test_get_auth_headers(): |
| 23 | + client = DIMO(env="Dev") |
| 24 | + headers = client._get_auth_headers("token123") |
| 25 | + assert headers == { |
| 26 | + "Authorization": "Bearer token123", |
| 27 | + "Content-Type": "application/json", |
| 28 | + } |
| 29 | + |
| 30 | + |
| 31 | +def test_query_calls_request_with_correct_payload(monkeypatch): |
| 32 | + client = DIMO(env="Dev") |
| 33 | + # Create a fake request method on the client |
| 34 | + fake_request = MagicMock(return_value={"data": {"result": True}}) |
| 35 | + monkeypatch.setattr(client, "request", fake_request) |
| 36 | + |
| 37 | + query_str = "query { test }" |
| 38 | + variables = {"key": "value"} |
| 39 | + result = client.query("Trips", query_str, variables=variables, token="mocked_token") |
| 40 | + |
| 41 | + # Verify the fake request was invoked once |
| 42 | + fake_request.assert_called_once() |
| 43 | + # Inspect call arguments |
| 44 | + args, kwargs = fake_request.call_args |
| 45 | + assert args[0] == "POST" |
| 46 | + assert args[1] == "Trips" |
| 47 | + assert args[2] == "" |
| 48 | + |
| 49 | + # Assert correct headers |
| 50 | + headers = kwargs["headers"] |
| 51 | + assert headers["Authorization"] == "Bearer mocked_token" |
| 52 | + assert headers["Content-Type"] == "application/json" |
| 53 | + assert headers["User-Agent"] == "dimo-python-sdk" |
| 54 | + |
| 55 | + # Check payload data |
| 56 | + data = kwargs["data"] |
| 57 | + assert data["query"] == query_str |
| 58 | + assert data["variables"] == variables |
| 59 | + |
| 60 | + assert result == {"data": {"result": True}} |
0 commit comments