Add tests for BaseSession and TelegramAPIServer

This commit is contained in:
Alex Root Junior 2019-11-16 00:52:18 +02:00
parent c824b298c9
commit a24708d589
12 changed files with 159 additions and 9 deletions

View file

@ -2,7 +2,7 @@ import abc
import asyncio
import datetime
import json
from typing import Any, Callable, Dict, List, Optional, TypeVar, Union
from typing import Any, Callable, Optional, TypeVar, Union
from pydantic.dataclasses import dataclass
@ -76,13 +76,13 @@ class BaseSession(abc.ABC):
return self.json_dumps(self.clean_json(value))
if isinstance(value, datetime.timedelta):
now = datetime.datetime.now()
return int((now + value).timestamp())
return str(round((now + value).timestamp()))
if isinstance(value, datetime.datetime):
return round(value.timestamp())
return str(round(value.timestamp()))
else:
return str(value)
def clean_json(self, value: Union[List, Dict]):
def clean_json(self, value: Any):
if isinstance(value, list):
return [self.clean_json(v) for v in value if v is not None]
elif isinstance(value, dict):