diff options
| author | Aldo Cortesi <aldo@nullcube.com> | 2015-05-02 17:19:48 +1200 |
|---|---|---|
| committer | Aldo Cortesi <aldo@nullcube.com> | 2015-05-02 17:19:48 +1200 |
| commit | 2091d1638734022a9ff879148ed432556c99db0f (patch) | |
| tree | 0d7f08904fc8a2e4c102259089873c129ff8fb16 /test/test_language_websocket.py | |
| parent | a6dbb829366fef47387739fa615bee904930a8a9 (diff) | |
| download | mitmproxy-2091d1638734022a9ff879148ed432556c99db0f.tar.gz mitmproxy-2091d1638734022a9ff879148ed432556c99db0f.tar.bz2 mitmproxy-2091d1638734022a9ff879148ed432556c99db0f.zip | |
test coverage: generators and websockets to 100%
Diffstat (limited to 'test/test_language_websocket.py')
| -rw-r--r-- | test/test_language_websocket.py | 23 |
1 files changed, 14 insertions, 9 deletions
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 |