feat: added time endpoints

This commit is contained in:
ITQ
2025-02-15 19:50:49 +03:00
parent 9f0a3f2a6e
commit c58704852e
4 changed files with 167 additions and 0 deletions
@@ -0,0 +1 @@
# noqa: A005
@@ -0,0 +1,21 @@
from django.core.cache import cache
from ninja import Schema
from pydantic import field_validator
from pydantic.types import NonNegativeInt
class CurrentDate(Schema):
current_date: NonNegativeInt
@field_validator("current_date", mode="after")
@classmethod
def check_bigger_than_setted_date(cls, value: int) -> int:
current_date = cache.get("current_date", default=0)
if value < current_date:
err = (
"current_date can't be less than setted "
f"date ({current_date})."
)
raise ValueError(err)
return value
@@ -0,0 +1,120 @@
from http import HTTPStatus as status
from django.test import TestCase, override_settings
from django.core.cache import cache
import json
class AdvanceTimeTests(TestCase):
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def setUp(self):
cache.clear()
cache.set("current_date", 10)
self.url = "/time/advance"
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_success(self):
self.assertEqual(cache.get("current_date"), 10)
response = self.client.post(
self.url,
data=json.dumps({"current_date": 15}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.OK)
self.assertEqual(response.json()["current_date"], 15)
self.assertEqual(cache.get("current_date"), 15)
# unittest & django pobeda so i can't use override_settings and parametrized at the same time, sorry
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_failure_invalid_value1(self):
response = self.client.post(
self.url,
data=json.dumps({"current_date": list()}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.BAD_REQUEST)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_failure_invalid_value2(self):
response = self.client.post(
self.url,
data=json.dumps({"current_date": -1241}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.BAD_REQUEST)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_failure_invalid_value3(self):
response = self.client.post(
self.url,
data=json.dumps({"current_date": "lol"}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.BAD_REQUEST)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_failure_invalid_value4(self):
response = self.client.post(
self.url,
data=json.dumps({"current_date": dict()}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.BAD_REQUEST)
@override_settings(
CACHES={
"default": {
"BACKEND": "django.core.cache.backends.locmem.LocMemCache",
}
}
)
def test_advance_time_failure_less_than_actual(self):
response = self.client.post(
self.url,
data=json.dumps({"current_date": 5}),
content_type="application/json",
)
self.assertEqual(response.status_code, status.BAD_REQUEST)
@@ -0,0 +1,25 @@
from http import HTTPStatus as status
from django.core.cache import cache
from django.http import HttpRequest
from ninja import Router
from api.v1 import schemas as global_schemas
from api.v1.time import schemas
router = Router(tags=["time"])
@router.post(
"/advance",
response={
status.OK: schemas.CurrentDate,
status.BAD_REQUEST: global_schemas.BadRequestError,
},
)
def advance_time(
request: HttpRequest, new_date: schemas.CurrentDate
) -> tuple[status, schemas.CurrentDate]:
cache.set("current_date", new_date.current_date)
return status.OK, schemas.CurrentDate(current_date=new_date.current_date)