Add utility function tests

2019-10-27 14:40:08 +01:00
parent e757e51a4e
commit 77d9b25bf0
2 changed files with 323 additions and 0 deletions
--- a/tests/test_core.py
+++ b/tests/test_core.py
@@ -0,0 +1,14 @@
+import pytest
+from fbchat._core import Enum
+
+
+@pytest.mark.filterwarnings("ignore::DeprecationWarning")
+def test_enum_extend_if_invalid():
+    class TestEnum(Enum):
+        A = 1
+        B = 2
+
+    assert TestEnum._extend_if_invalid(1) == TestEnum.A
+    assert TestEnum._extend_if_invalid(3) == TestEnum.UNKNOWN_3
+    assert TestEnum._extend_if_invalid(3) == TestEnum.UNKNOWN_3
+    assert TestEnum(3) == TestEnum.UNKNOWN_3