diff options
| author | Pierre Ossman <ossman@cendio.se> | 2017-02-09 10:38:53 +0100 |
|---|---|---|
| committer | Pierre Ossman <ossman@cendio.se> | 2017-02-09 10:38:53 +0100 |
| commit | 600f4e154c19ee473b4e7aad559c931fd0b08c73 (patch) | |
| tree | fa5a671c12f1222d49537b5cc552ce7c8577150f /tests | |
| parent | df317822d6af7dc954913e5a9947d36b33bfbf33 (diff) | |
| download | websockify-600f4e154c19ee473b4e7aad559c931fd0b08c73.tar.gz | |
Avoid assertIn for Python 2.6 compatibility
It was added in 2.7, but we want to support older Python a while
longer.
Diffstat (limited to 'tests')
| -rw-r--r-- | tests/test_websocket.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/test_websocket.py b/tests/test_websocket.py index 49cc620..9c7870a 100644 --- a/tests/test_websocket.py +++ b/tests/test_websocket.py @@ -34,9 +34,9 @@ class AcceptTestCase(unittest.TestCase): 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q=='}) self.assertEqual(sock.data[:13], b'HTTP/1.1 101 ') - self.assertIn(b'\r\nUpgrade: websocket\r\n', sock.data) - self.assertIn(b'\r\nConnection: Upgrade\r\n', sock.data) - self.assertIn(b'\r\nSec-WebSocket-Accept: pczpYSQsvE1vBpTQYjFQPcuoj6M=\r\n', sock.data) + self.assertTrue(b'\r\nUpgrade: websocket\r\n' in sock.data) + self.assertTrue(b'\r\nConnection: Upgrade\r\n' in sock.data) + self.assertTrue(b'\r\nSec-WebSocket-Accept: pczpYSQsvE1vBpTQYjFQPcuoj6M=\r\n' in sock.data) def test_bad_version(self): ws = websocket.WebSocket() @@ -83,7 +83,7 @@ class AcceptTestCase(unittest.TestCase): 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q==', 'Sec-WebSocket-Protocol': 'foobar gazonk'}) self.assertEqual(sock.data[:13], b'HTTP/1.1 101 ') - self.assertIn(b'\r\nSec-WebSocket-Protocol: gazonk\r\n', sock.data) + self.assertTrue(b'\r\nSec-WebSocket-Protocol: gazonk\r\n' in sock.data) def test_no_protocol(self): ws = websocket.WebSocket() @@ -92,7 +92,7 @@ class AcceptTestCase(unittest.TestCase): 'Sec-WebSocket-Version': '13', 'Sec-WebSocket-Key': 'DKURYVK9cRFul1vOZVA56Q=='}) self.assertEqual(sock.data[:13], b'HTTP/1.1 101 ') - self.assertNotIn(b'\r\nSec-WebSocket-Protocol:', sock.data) + self.assertFalse(b'\r\nSec-WebSocket-Protocol:' in sock.data) def test_missing_protocol(self): ws = websocket.WebSocket() |