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_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)