Update unit tests for skipping OPTIONS message

This commit is contained in:
Tyler Hobbs
2014-02-12 12:00:23 -06:00
parent 7b029c98c9
commit eb402f6aad
2 changed files with 3 additions and 3 deletions

View File

@@ -21,7 +21,7 @@ from cassandra.marshal import uint8_pack, uint32_pack
from cassandra.io.asyncorereactor import AsyncoreConnection
class LibevConnectionTest(unittest.TestCase):
class AsyncoreConnectionTest(unittest.TestCase):
@classmethod
def setUpClass(cls):
@@ -35,7 +35,7 @@ class LibevConnectionTest(unittest.TestCase):
cls.socket_patcher.stop()
def make_connection(self):
c = AsyncoreConnection('1.2.3.4')
c = AsyncoreConnection('1.2.3.4', cql_version='3.0.1')
c.socket = Mock()
c.socket.send.side_effect = lambda x: len(x)
return c

View File

@@ -35,7 +35,7 @@ class LibevConnectionTest(unittest.TestCase):
raise unittest.SkipTest('libev does not appear to be installed correctly')
def make_connection(self):
c = LibevConnection('1.2.3.4')
c = LibevConnection('1.2.3.4', cql_version='3.0.1')
c._socket = Mock()
c._socket.send.side_effect = lambda x: len(x)
return c