joshua
/
websockify
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Avoid assertIn for Python 2.6 compatibility 

It was added in 2.7, but we want to support older Python a while
longer.
This commit is contained in:
Pierre Ossman 2017-02-09 10:38:53 +01:00
parent df317822d6
commit 600f4e154c
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_websocket.py
View File

@ -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()