summaryrefslogtreecommitdiff
path: root/zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java
diff options
context:
space:
mode:
Diffstat (limited to 'zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java')
-rw-r--r--zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java72
1 files changed, 48 insertions, 24 deletions
diff --git a/zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java b/zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java
index 34c8c0cba..a02127aae 100644
--- a/zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java
+++ b/zookeeper-server/src/test/java/org/apache/zookeeper/TestableZooKeeper.java
@@ -1,4 +1,4 @@
-/**
+/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
@@ -23,7 +23,6 @@ import java.net.SocketAddress;
import java.util.List;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;
-
import org.apache.jute.Record;
import org.apache.zookeeper.admin.ZooKeeperAdmin;
import org.apache.zookeeper.client.HostProvider;
@@ -32,17 +31,29 @@ import org.apache.zookeeper.proto.RequestHeader;
public class TestableZooKeeper extends ZooKeeperAdmin {
- public TestableZooKeeper(String host, int sessionTimeout,
- Watcher watcher) throws IOException {
+ public TestableZooKeeper(String host, int sessionTimeout, Watcher watcher) throws IOException {
super(host, sessionTimeout, watcher);
}
class TestableClientCnxn extends ClientCnxn {
- TestableClientCnxn(String chrootPath, HostProvider hostProvider, int sessionTimeout, ZooKeeper zooKeeper,
- ClientWatchManager watcher, ClientCnxnSocket clientCnxnSocket, boolean canBeReadOnly)
- throws IOException {
- super(chrootPath, hostProvider, sessionTimeout, zooKeeper, watcher,
- clientCnxnSocket, 0, new byte[16], canBeReadOnly);
+
+ TestableClientCnxn(
+ String chrootPath,
+ HostProvider hostProvider,
+ int sessionTimeout,
+ ZooKeeper zooKeeper,
+ ClientWatchManager watcher,
+ ClientCnxnSocket clientCnxnSocket,
+ boolean canBeReadOnly) throws IOException {
+ super(chrootPath,
+ hostProvider,
+ sessionTimeout,
+ zooKeeper,
+ watcher,
+ clientCnxnSocket,
+ 0,
+ new byte[16],
+ canBeReadOnly);
}
void setXid(int newXid) {
@@ -52,36 +63,45 @@ public class TestableZooKeeper extends ZooKeeperAdmin {
int checkXid() {
return xid;
}
+
}
- protected ClientCnxn createConnection(String chrootPath,
- HostProvider hostProvider, int sessionTimeout, ZooKeeper zooKeeper,
- ClientWatchManager watcher, ClientCnxnSocket clientCnxnSocket,
- boolean canBeReadOnly) throws IOException {
- return new TestableClientCnxn(chrootPath, hostProvider, sessionTimeout, this,
- watcher, clientCnxnSocket, canBeReadOnly);
+ protected ClientCnxn createConnection(
+ String chrootPath,
+ HostProvider hostProvider,
+ int sessionTimeout,
+ ZooKeeper zooKeeper,
+ ClientWatchManager watcher,
+ ClientCnxnSocket clientCnxnSocket,
+ boolean canBeReadOnly) throws IOException {
+ return new TestableClientCnxn(
+ chrootPath,
+ hostProvider,
+ sessionTimeout,
+ this,
+ watcher,
+ clientCnxnSocket,
+ canBeReadOnly);
}
public void setXid(int xid) {
- ((TestableClientCnxn)cnxn).setXid(xid);
+ ((TestableClientCnxn) cnxn).setXid(xid);
}
public int checkXid() {
- return ((TestableClientCnxn)cnxn).checkXid();
+ return ((TestableClientCnxn) cnxn).checkXid();
}
-
+
@Override
public List<String> getChildWatches() {
return super.getChildWatches();
}
-
@Override
public List<String> getDataWatches() {
return super.getDataWatches();
}
-
@Override
public List<String> getExistWatches() {
return super.getExistWatches();
@@ -92,7 +112,7 @@ public class TestableZooKeeper extends ZooKeeperAdmin {
* later attempt to reconnect.
*/
public void testableConnloss() throws IOException {
- synchronized(cnxn) {
+ synchronized (cnxn) {
cnxn.sendThread.testableCloseSocket();
}
}
@@ -107,7 +127,7 @@ public class TestableZooKeeper extends ZooKeeperAdmin {
final CountDownLatch initiatedPause = new CountDownLatch(1);
new Thread() {
public void run() {
- synchronized(cnxn) {
+ synchronized (cnxn) {
try {
try {
cnxn.sendThread.testableCloseSocket();
@@ -146,8 +166,11 @@ public class TestableZooKeeper extends ZooKeeperAdmin {
return cnxn.getLastZxid();
}
- public ReplyHeader submitRequest(RequestHeader h, Record request,
- Record response, WatchRegistration watchRegistration) throws InterruptedException {
+ public ReplyHeader submitRequest(
+ RequestHeader h,
+ Record request,
+ Record response,
+ WatchRegistration watchRegistration) throws InterruptedException {
return cnxn.submitRequest(h, request, response, watchRegistration);
}
@@ -159,4 +182,5 @@ public class TestableZooKeeper extends ZooKeeperAdmin {
public void disconnect() {
cnxn.disconnect();
}
+
}