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)