mirror of
https://github.com/aiogram/aiogram.git
synced 2026-04-08 16:37:47 +00:00
42 lines
1.5 KiB
Python
42 lines
1.5 KiB
Python
import aresponses
|
|
|
|
from aiogram import Bot
|
|
|
|
TOKEN = "123456789:AABBCCDDEEFFaabbccddeeff-1234567890"
|
|
|
|
|
|
class FakeTelegram(aresponses.ResponsesMockServer):
|
|
def __init__(self, message_dict, bot=None, **kwargs):
|
|
super().__init__(**kwargs)
|
|
self._body, self._headers = self.parse_data(message_dict)
|
|
|
|
if isinstance(bot, Bot):
|
|
Bot.set_current(bot)
|
|
|
|
async def __aenter__(self):
|
|
await super().__aenter__()
|
|
_response = self.Response(text=self._body, headers=self._headers, status=200, reason="OK")
|
|
self.add(self.ANY, response=_response)
|
|
|
|
async def __aexit__(self, exc_type, exc_val, exc_tb):
|
|
if hasattr(self, "monkeypatch"):
|
|
self.monkeypatch.undo()
|
|
await super().__aexit__(exc_type, exc_val, exc_tb)
|
|
|
|
@staticmethod
|
|
def parse_data(message_dict):
|
|
import json
|
|
|
|
_body = '{"ok":true,"result":' + json.dumps(message_dict) + "}"
|
|
_headers = {
|
|
"Server": "nginx/1.12.2",
|
|
"Date": "Tue, 03 Apr 2018 16:59:54 GMT",
|
|
"Content-Type": "application/json",
|
|
"Content-Length": str(len(_body)),
|
|
"Connection": "keep-alive",
|
|
"Access-Control-Allow-Origin": "*",
|
|
"Access-Control-Allow-Methods": "GET, POST, OPTIONS",
|
|
"Access-Control-Expose-Headers": "Content-Length,Content-Type,Date,Server,Connection",
|
|
"Strict-Transport-Security": "max-age=31536000; includeSubdomains",
|
|
}
|
|
return _body, _headers
|