diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/test_language_generators.py | 11 | ||||
| -rw-r--r-- | test/test_language_websocket.py | 23 |
2 files changed, 25 insertions, 9 deletions
diff --git a/test/test_language_generators.py b/test/test_language_generators.py index 3db1ac91..854b8485 100644 --- a/test/test_language_generators.py +++ b/test/test_language_generators.py @@ -36,3 +36,14 @@ def test_filegenerator(): # remove all references to FileGenerator instance to close the file # handle. del g + + +def test_transform_generator(): + def trans(offset, data): + return "a" * len(data) + g = generators.LiteralGenerator("one") + t = generators.TransformGenerator(g, trans) + assert len(t) == len(g) + assert t[0] == "a" + assert t[:] == "a" * len(g) + assert repr(t) diff --git a/test/test_language_websocket.py b/test/test_language_websocket.py index 8a683d81..3b1d4852 100644 --- a/test/test_language_websocket.py +++ b/test/test_language_websocket.py @@ -8,13 +8,18 @@ def parse_request(s): class TestWebsocketFrame: - def test_spec(self): - e = websockets.WebsocketFrame.expr() - wf = e.parseString("wf:b'foo'") - assert wf - - assert parse_request("wf:b'foo'") - def test_values(self): - r = parse_request("wf:b'foo'") - assert r.values(language.Settings()) + specs = [ + "wf", + "wf:b'foo'" + ] + for i in specs: + wf = parse_request(i) + assert isinstance(wf, websockets.WebsocketFrame) + assert wf + assert wf.values(language.Settings()) + assert wf.resolve(language.Settings()) + + spec = wf.spec() + wf2 = parse_request(spec) + assert wf2.spec() == spec |