feat(override-plan): Ability to fetch a single subscription (#196)

getlago · Oct 5, 2023 · c9d0efe · c9d0efe
1 parent 1fbd78a
commit c9d0efe
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/lago_python_client/subscriptions/clients.py b/lago_python_client/subscriptions/clients.py
@@ -1,14 +1,15 @@
 from typing import ClassVar, Type
 
 from ..base_client import BaseClient
-from ..mixins import CreateCommandMixin, DestroyCommandMixin, FindAllCommandMixin, UpdateCommandMixin
+from ..mixins import CreateCommandMixin, DestroyCommandMixin, FindAllCommandMixin, FindCommandMixin, UpdateCommandMixin
 from ..models.subscription import SubscriptionResponse
 
 
 class SubscriptionClient(
     CreateCommandMixin[SubscriptionResponse],
     DestroyCommandMixin[SubscriptionResponse],
     FindAllCommandMixin[SubscriptionResponse],
+    FindCommandMixin[SubscriptionResponse],
     UpdateCommandMixin[SubscriptionResponse],
     BaseClient,
 ):

diff --git a/tests/test_subscription_client.py b/tests/test_subscription_client.py
@@ -114,6 +114,25 @@ def test_invalid_destroy_subscription_request(httpx_mock: HTTPXMock):
     with pytest.raises(LagoApiError):
         client.subscriptions.destroy(identifier)
 
+def test_valid_find_subscription_request(httpx_mock: HTTPXMock):
+    client = Client(api_key='886fe239-927d-4072-ab72-6dd345e8dd0d')
+    external_id = '5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba'
+
+    httpx_mock.add_response(method='GET', url='https://api.getlago.com/api/v1/subscriptions/' + external_id, content=mock_response())
+    response = client.subscriptions.find(external_id)
+
+    assert response.lago_id == 'b7ab2926-1de8-4428-9bcd-779314ac129b'
+    assert response.external_customer_id == '5eb02857-a71e-4ea2-bcf9-57d3a41bc6ba'
+
+
+def test_invalid_find_subscription_request(httpx_mock: HTTPXMock):
+    client = Client(api_key='invalid')
+    external_id = 'invalid'
+
+    httpx_mock.add_response(method='GET', url='https://api.getlago.com/api/v1/subscriptions/' + external_id, status_code=404, content=b'')
+
+    with pytest.raises(LagoApiError):
+        client.subscriptions.find(external_id)
 
 def test_valid_find_all_subscription_request_with_options(httpx_mock: HTTPXMock):
     client = Client(api_key='886fe239-927d-4072-ab72-6dd345e8dd0d')