forked from breatheco-de/apiv2
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
6 changed files
with
277 additions
and
47 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
179 changes: 179 additions & 0 deletions
179
breathecode/payments/tests/urls/tests_me_service_slug_consumptionsession.py
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,179 @@ | ||
| import random | ||
| from datetime import datetime, timedelta, timezone | ||
|
|
||
| import pytest | ||
| from django.urls import reverse_lazy | ||
| from rest_framework import status | ||
|
|
||
| from breathecode.tests.mixins.breathecode_mixin.breathecode import Breathecode | ||
| from capyc.rest_framework.pytest import fixtures as rfx | ||
|
|
||
|
|
||
| @pytest.fixture(autouse=True) | ||
| def setup(db, monkeypatch: pytest.MonkeyPatch): | ||
| monkeypatch.setattr( | ||
| "breathecode.payments.tasks.end_the_consumption_session.apply_async", lambda *args, **kwargs: None | ||
| ) | ||
| yield | ||
|
|
||
|
|
||
| def db_item(service, data={}): | ||
| return { | ||
| "consumable_id": 1, | ||
| "duration": ..., | ||
| "eta": ..., | ||
| "how_many": 1.0, | ||
| "id": 1, | ||
| "operation_code": "unsafe-consume-service-set", | ||
| "path": "", | ||
| "related_id": 0, | ||
| "related_slug": "", | ||
| "request": { | ||
| "args": [], | ||
| "headers": { | ||
| "academy": None, | ||
| }, | ||
| "kwargs": { | ||
| "service_slug": service.slug, | ||
| }, | ||
| "user": 1, | ||
| }, | ||
| "status": "PENDING", | ||
| "user_id": 1, | ||
| "was_discounted": False, | ||
| **data, | ||
| } | ||
|
|
||
|
|
||
| def random_duration(): | ||
| hours = random.randint(0, 23) | ||
| minutes = random.randint(0, 59) | ||
| seconds = random.randint(0, 59) | ||
| return timedelta(hours=hours, minutes=minutes, seconds=seconds) | ||
|
|
||
|
|
||
| def test_no_auth(bc: Breathecode, client: rfx.Client): | ||
| url = reverse_lazy("payments:me_service_slug_consumptionsession", kwargs={"service_slug": "my-service"}) | ||
|
|
||
| response = client.put(url) | ||
|
|
||
| json = response.json() | ||
| expected = {"detail": "Authentication credentials were not provided.", "status_code": 401} | ||
|
|
||
| assert json == expected | ||
| assert response.status_code == status.HTTP_401_UNAUTHORIZED | ||
| assert bc.database.list_of("payments.ConsumptionSession") == [] | ||
|
|
||
|
|
||
| def test_no_consumables(bc: Breathecode, client: rfx.Client): | ||
| url = reverse_lazy("payments:me_service_slug_consumptionsession", kwargs={"service_slug": "my-service"}) | ||
|
|
||
| model = bc.database.create(user=1) | ||
| client.force_authenticate(user=model.user) | ||
|
|
||
| response = client.put(url) | ||
|
|
||
| json = response.json() | ||
| expected = {"detail": "insufficient-credits", "status_code": 402} | ||
|
|
||
| assert json == expected | ||
| assert response.status_code == status.HTTP_402_PAYMENT_REQUIRED | ||
| assert bc.database.list_of("payments.ConsumptionSession") == [] | ||
|
|
||
|
|
||
| def test_created(bc: Breathecode, client: rfx.Client, utc_now): | ||
| duration = random_duration() | ||
| model = bc.database.create(user=1, consumable=1, service={"session_duration": duration, "type": "VOID"}) | ||
| url = reverse_lazy("payments:me_service_slug_consumptionsession", kwargs={"service_slug": model.service.slug}) | ||
|
|
||
| client.force_authenticate(user=model.user) | ||
|
|
||
| response = client.put(url) | ||
|
|
||
| json = response.json() | ||
| expected = {"id": 1, "status": "ok"} | ||
|
|
||
| assert json == expected | ||
| assert response.status_code == status.HTTP_201_CREATED | ||
| assert bc.database.list_of("payments.ConsumptionSession") == [ | ||
| db_item( | ||
| model.service, | ||
| data={ | ||
| "related_id": 1, | ||
| "related_slug": model.service.slug, | ||
| "path": "payments.Service", | ||
| "duration": duration, | ||
| "eta": utc_now + duration, | ||
| }, | ||
| ) | ||
| ] | ||
|
|
||
|
|
||
| def test_cached(bc: Breathecode, client: rfx.Client, utc_now, fake): | ||
| slug = fake.slug() | ||
| duration = random_duration() | ||
| model = bc.database.create( | ||
| user=1, | ||
| consumable=1, | ||
| service={ | ||
| "session_duration": duration, | ||
| "slug": slug, | ||
| "type": "VOID", | ||
| }, | ||
| consumption_session={ | ||
| "how_many": 1, | ||
| "eta": utc_now + duration, | ||
| "duration": duration, | ||
| "was_discounted": False, | ||
| "operation_code": "unsafe-consume-service-set", | ||
| "related_id": 1, | ||
| "related_slug": slug, | ||
| "status": "PENDING", | ||
| "path": "payments.Service", | ||
| "request": { | ||
| "args": [], | ||
| "headers": { | ||
| "academy": None, | ||
| }, | ||
| "kwargs": { | ||
| "service_slug": slug, | ||
| }, | ||
| "user": 1, | ||
| }, | ||
| }, | ||
| ) | ||
| url = reverse_lazy("payments:me_service_slug_consumptionsession", kwargs={"service_slug": model.service.slug}) | ||
|
|
||
| client.force_authenticate(user=model.user) | ||
|
|
||
| response = client.put(url) | ||
|
|
||
| json = response.json() | ||
| expected = {"id": 2, "status": "ok"} | ||
|
|
||
| assert json == expected | ||
| assert response.status_code == status.HTTP_201_CREATED | ||
| assert bc.database.list_of("payments.ConsumptionSession") == [ | ||
| db_item( | ||
| model.service, | ||
| data={ | ||
| "id": 1, | ||
| "related_id": 1, | ||
| "related_slug": model.service.slug, | ||
| "path": "payments.Service", | ||
| "duration": duration, | ||
| "eta": utc_now + duration, | ||
| }, | ||
| ), | ||
| db_item( | ||
| model.service, | ||
| data={ | ||
| "id": 2, | ||
| "related_id": 1, | ||
| "related_slug": model.service.slug, | ||
| "path": "payments.Service", | ||
| "duration": duration, | ||
| "eta": utc_now + duration, | ||
| }, | ||
| ), | ||
| ] |
Oops, something went wrong.