🧪 test: decorator & utils

Author: mic1on

tests/test_decorator.py

@@ -0,0 +1,78 @@
+from usepy import (
+    useCachedProperty,  # noqa
+    useCatchError,
+    useListify,
+    useRunInThread,
+    useSingleton,
+    useTimeIt,
+)
+
+
+def test_cached_property():
+    class Foo(object):
+        @useCachedProperty
+        def bar(self):
+            return 1
+
+    foo = Foo()
+    assert foo.bar == 1
+    assert foo.bar == 1
+
+
+def test_catch_error():
+    @useCatchError()
+    def foo():
+        raise Exception('foo')
+
+    @useCatchError(return_val=1)
+    def foo2():
+        raise Exception('foo2')
+
+    assert foo() is None
+    assert foo2() == 1
+
+
+def test_listify():
+    @useListify()
+    def foo():
+        yield 1
+
+    @useListify()
+    def foo2():
+        yield 1
+        yield 2
+
+    @useListify(dict)
+    def foo3():
+        yield 'a', 1
+        yield 'b', 2
+
+    assert foo() == [1]
+    assert foo2() == [1, 2]
+    assert foo3() == {'a': 1, 'b': 2}
+
+
+def test_run_in_thread():
+    @useRunInThread
+    def foo():
+        return 1
+
+    assert foo() is None
+
+
+def test_singleton():
+    @useSingleton
+    class Foo(object):
+        pass
+
+    foo = Foo()
+    foo2 = Foo()
+    assert foo is foo2
+
+
+def test_timeit():
+    @useTimeIt
+    def foo():
+        return 1
+
+    assert foo() == 1

tests/test_utils/test_is.py

@@ -0,0 +1,20 @@
+from usepy import useIs
+
+
+def test_is_token():
+    assert useIs.is_token("token")
+    assert not useIs.is_token("to")
+    assert useIs.is_token("cd9bb741c5553b2243c6d39fb5dd3c34")
+
+
+def test_is_regexp():
+    import re
+    assert not useIs.is_regexp("regex")
+    assert not useIs.is_regexp(".*?")
+    assert useIs.is_regexp(re.compile(".*?"))
+
+
+def test_is_string():
+    assert useIs.is_string("string")
+    assert useIs.is_string(b"bytes")
+    assert not useIs.is_string(123)

tests/test_utils/test_to.py

@@ -0,0 +1,38 @@
+from usepy import useTo
+
+
+def test_to_dict():
+    assert useTo.cookie_to_dict(
+        "code=1000; name=miclon; UID=359FEA9A6F1C7E97BFE909A1A700F5DE"
+    ) == {
+               "code": "1000",
+               "name": "miclon",
+               "UID": "359FEA9A6F1C7E97BFE909A1A700F5DE",
+           }
+
+    assert useTo.headers_to_dict(
+        """
+        pragma: no-cache
+referer: https://github.com/
+user-agent: Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0
+        """) == {
+               "pragma": "no-cache",
+               "referer": "https://github.com/",
+               "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/108.0.0.0",
+           }
+    assert useTo.data_to_dict("code=1000&name=miclon&UID=359FEA9A6F1C7E97BFE909A1A700F5DE") == {
+        "code": "1000",
+        "name": "miclon",
+        "UID": "359FEA9A6F1C7E97BFE909A1A700F5DE",
+    }
+
+
+def test_to_string():
+    assert useTo.to_string("test") == "test"
+    assert useTo.to_string(b"test") == "test"
+    assert useTo.to_string(1000) == "1000"
+    assert useTo.to_string(1000.0) == "1000.0"
+    assert useTo.to_string(True) == "True"
+    assert useTo.to_string(None) == "None"
+    assert useTo.to_string([1, 2, 3]) == "[1, 2, 3]"
+    assert useTo.to_string({"code": 1000}) == "{'code': 1000}"