mirror of
https://github.com/koloideal/Argenta.git
synced 2026-06-10 10:05:28 +03:00
23 lines
639 B
Python
23 lines
639 B
Python
import io
|
|
import unittest
|
|
from contextlib import redirect_stdout
|
|
|
|
from argenta import Router, Command, Response
|
|
from argenta.command import InputCommand
|
|
|
|
|
|
router = Router(title="Demo")
|
|
|
|
|
|
@router.command(Command("PING", description="Ping command"))
|
|
def ping(response: Response):
|
|
print("PONG")
|
|
|
|
|
|
class TestSimpleHandler(unittest.TestCase):
|
|
def test_ping_prints_pong(self):
|
|
# Имитация запуска хендлера через роутер
|
|
with redirect_stdout(io.StringIO()) as stdout:
|
|
router.finds_appropriate_handler(InputCommand.parse("PING"))
|
|
self.assertIn("PONG", stdout.getvalue())
|