This commit is contained in:
2025-03-15 00:08:39 +03:00
parent 09c40978a1
commit baaf0e25f3
11 changed files with 77 additions and 9 deletions
View File
+30
View File
@@ -0,0 +1,30 @@
import _io
from unittest.mock import patch, MagicMock
import unittest
import io
from argenta.app import App
from argenta.command import Command
from argenta.router import Router
class TestSystem(unittest.TestCase):
@patch("builtins.input", side_effect=["test", "q"])
@patch("sys.stdout", new_callable=io.StringIO)
def test_run_shell_output(self, mock_stdout: _io.StringIO, magick_mock: MagicMock):
router = Router()
@router.command(Command('test'))
def test():
print('test command')
app = App()
app.include_router(router)
app.start_polling()
output = mock_stdout.getvalue()
self.assertIn('test command', output)