def test_connect_success(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init()
handle = connection.connect(FAKE_SERVERS)
self.assertEquals(handle, FAKE_ZK_HANDLE)
self.assertEquals(mock_init.call_count, 1)
self.assertEquals(mock_init.call_args_list[0][0][0], FAKE_SERVER_STRING)
def test_connect_success(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init()
handle = connection.connect(FAKE_SERVERS)
self.assertEquals(handle, FAKE_ZK_HANDLE)
self.assertEquals(mock_init.call_count, 1)
self.assertEquals(mock_init.call_args_list[0][0][0],
FAKE_SERVER_STRING)
def test_connect_with_bad_server_arg(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init()
with self.assertRaises(FakeBadArgumentsException):
connection.connect(None)
with self.assertRaises(FakeBadArgumentsException):
connection.connect([])
with self.assertRaises(FakeBadArgumentsException):
connection.connect("Blah blah")
def test_connect_with_bad_server_arg(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init()
with self.assertRaises(FakeBadArgumentsException):
connection.connect(None)
with self.assertRaises(FakeBadArgumentsException):
connection.connect([])
with self.assertRaises(FakeBadArgumentsException):
connection.connect("Blah blah")
def test_connect_handles_connection_timeout(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init(\
connect_success=False)
with self.assertRaises(FakeZookeeperException):
connection.connect(FAKE_SERVERS, timeout=1)
def test_connect_handles_exceptions(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init(\
throw_exception=Exception)
with self.assertRaises(FakeZookeeperException):
connection.connect(FAKE_SERVERS)
def test_connect_handles_connection_timeout(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init(connect_success=False)
with self.assertRaises(FakeZookeeperException):
connection.connect(FAKE_SERVERS, timeout=1)
def test_connect_handles_exceptions(self):
with mock.patch("zookeeper.init") as mock_init:
mock_init.side_effect = mock_zookeeper_init(throw_exception=Exception)
with self.assertRaises(FakeZookeeperException):
connection.connect(FAKE_SERVERS)
