python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2025-12-12 10:11:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Small refactoring of telegram api server

Browse source
This commit is contained in:
Alex Root Junior 2019-11-17 22:20:29 +02:00
parent 0e35b6f55e
commit a46cd6d001
6 changed files with 43 additions and 26 deletions
Download patch file Download diff file Expand all files Collapse all files

11
tests/test_api/test_client/test_session/test_api_server.py
View file

@ -1,11 +0,0 @@
from aiogram.api.client.session.base import PRODUCTION
class TestAPIServer:
def test_method_url(self):
method_url = PRODUCTION.api_url(token="TOKEN", method="apiMethod")
assert method_url == "https://api.telegram.org/botTOKEN/apiMethod"
def test_file_url(self):
file_url = PRODUCTION.file_url(token="TOKEN", path="path")
assert file_url == "https://api.telegram.org/file/botTOKEN/path"

16
tests/test_api/test_client/test_session/test_base_session.py
View file

@ -2,11 +2,11 @@ import datetime
from unittest.mock import patch
import pytest
from asynctest import CoroutineMock
from aiogram.api.client.session.base import BaseSession
from aiogram.api.client.telegram import PRODUCTION, TelegramAPIServer
from aiogram.api.methods import GetMe, Response
from aiogram.utils.mixins import DataMixin
from asynctest import CoroutineMock
class TestBaseSession(DataMixin):
@ -17,6 +17,18 @@ class TestBaseSession(DataMixin):
def teardown(self):
BaseSession.__abstractmethods__ = self["__abstractmethods__"]
def test_init_api(self):
session = BaseSession()
assert session.api == PRODUCTION
def test_init_custom_api(self):
api = TelegramAPIServer(
base="http://example.com/{token}/{method}",
file="http://example.com/{token}/file/{path{",
)
session = BaseSession(api=api)
assert session.api == api
def test_sync_close(self):
session = BaseSession()