diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_language_base.py | 11 | ||||
| -rw-r--r-- | test/test_language_websocket.py | 8 |
2 files changed, 19 insertions, 0 deletions
diff --git a/test/test_language_base.py b/test/test_language_base.py index 404b302d..192ad727 100644 --- a/test/test_language_base.py +++ b/test/test_language_base.py @@ -305,6 +305,17 @@ def test_integer(): assert v.freeze({}).value == v.value + class BInt(base.Integer): + bounds = (1, 5) + + tutils.raises("must be between", BInt, 0) + tutils.raises("must be between", BInt, 6) + assert BInt(5) + assert BInt(1) + assert BInt(3) + + + class TBoolean(base.Boolean): name = "test" diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py index f9e8310a..0228c350 100644 --- a/test/test_language_websocket.py +++ b/test/test_language_websocket.py @@ -15,6 +15,7 @@ class TestWebsocketFrame: "wf", "wf:dr", "wf:b'foo'", + "wf:l1024:b'foo'", "wf:cbinary", "wf:c1", "wf:mask:knone", @@ -106,3 +107,10 @@ class TestWebsocketFrame: self.fr, "wf:b'foo':mask:knone", ) + + def test_length(self): + assert self.fr("wf:l3:b'foo'").header.payload_length == 3 + frm = self.fr("wf:l2:b'foo'") + assert frm.header.payload_length == 2 + assert frm.payload == "fo" + tutils.raises("expected 1024 bytes", self.fr, "wf:l1024:b'foo'") |