From 42aa153ca152496b0e6d6036e38e2b0e5ca84c2a Mon Sep 17 00:00:00 2001
From: Evgeniy Blinov <zheni-b@yandex.ru>
Date: Tue, 27 Aug 2024 13:28:14 +0300
Subject: [PATCH] new test

---
 .../tokens/abstract/test_coroutine_wrapper.py | 28 +++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 tests/units/tokens/abstract/test_coroutine_wrapper.py

diff --git a/tests/units/tokens/abstract/test_coroutine_wrapper.py b/tests/units/tokens/abstract/test_coroutine_wrapper.py
new file mode 100644
index 0000000..957b089
--- /dev/null
+++ b/tests/units/tokens/abstract/test_coroutine_wrapper.py
@@ -0,0 +1,28 @@
+import io
+import sys
+from contextlib import redirect_stdout
+
+import pytest
+
+from cantok import SimpleToken, TimeoutToken, ConditionToken, CounterToken
+
+
+@pytest.mark.parametrize(
+    ['create_value', 'expected_string'],
+    [
+        (lambda: SimpleToken(cancelled=True).wait(), ''),
+        (lambda: TimeoutToken(0.0001).wait(), ''),
+        (lambda: ConditionToken(lambda: True).wait(), ''),
+        (lambda: CounterToken(0).wait(), ''),
+        (lambda: 1, '1\n'),
+        (lambda: 'kek', f'{repr('kek')}\n'),
+    ],
+)
+def test_displayhook_printing_coroutine_wrappers_and_other_objects(create_value, expected_string):
+    buffer = io.StringIO()
+    with redirect_stdout(buffer):
+        sys.displayhook(create_value())
+
+    output = buffer.getvalue()
+
+    assert output == expected_string