summaryrefslogtreecommitdiff
path: root/zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java
diff options
context:
space:
mode:
Diffstat (limited to 'zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java')
-rw-r--r--zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java80
1 files changed, 34 insertions, 46 deletions
diff --git a/zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java b/zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java
index 8c18d78aa..2eb2c6caa 100644
--- a/zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.java
+++ b/zookeeper-server/src/test/java/org/apache/zookeeper/test/ZooKeeperQuotaTest.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
@@ -18,55 +18,51 @@
package org.apache.zookeeper.test;
+import static org.junit.Assert.assertEquals;
+import static org.junit.Assert.assertNotNull;
+import static org.junit.Assert.assertTrue;
import java.io.IOException;
-
import org.apache.zookeeper.CreateMode;
import org.apache.zookeeper.KeeperException;
import org.apache.zookeeper.Quotas;
import org.apache.zookeeper.StatsTrack;
-import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.ZooDefs.Ids;
+import org.apache.zookeeper.ZooKeeper;
import org.apache.zookeeper.cli.MalformedPathException;
import org.apache.zookeeper.cli.SetQuotaCommand;
import org.apache.zookeeper.data.Stat;
import org.apache.zookeeper.server.ZooKeeperServer;
-import org.junit.Assert;
import org.junit.Test;
public class ZooKeeperQuotaTest extends ClientBase {
@Test
- public void testQuota() throws IOException,
- InterruptedException, KeeperException, Exception {
+ public void testQuota() throws Exception {
final ZooKeeper zk = createClient();
final String path = "/a/b/v";
// making sure setdata works on /
zk.setData("/", "some".getBytes(), -1);
- zk.create("/a", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create("/a", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
- zk.create("/a/b", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create("/a/b", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
- zk.create("/a/b/v", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create("/a/b/v", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
- zk.create("/a/b/v/d", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create("/a/b/v/d", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
SetQuotaCommand.createQuota(zk, path, 5L, 10);
// see if its set
String absolutePath = Quotas.quotaZookeeper + path + "/" + Quotas.limitNode;
byte[] data = zk.getData(absolutePath, false, new Stat());
StatsTrack st = new StatsTrack(new String(data));
- Assert.assertTrue("bytes are set", st.getBytes() == 5L);
- Assert.assertTrue("num count is set", st.getCount() == 10);
+ assertTrue("bytes are set", st.getBytes() == 5L);
+ assertTrue("num count is set", st.getCount() == 10);
String statPath = Quotas.quotaZookeeper + path + "/" + Quotas.statNode;
byte[] qdata = zk.getData(statPath, false, new Stat());
StatsTrack qst = new StatsTrack(new String(qdata));
- Assert.assertTrue("bytes are set", qst.getBytes() == 8L);
- Assert.assertTrue("count is set", qst.getCount() == 2);
+ assertTrue("bytes are set", qst.getBytes() == 8L);
+ assertTrue("count is set", qst.getCount() == 2);
//force server to restart and load from snapshot, not txn log
stopServer();
@@ -74,19 +70,17 @@ public class ZooKeeperQuotaTest extends ClientBase {
stopServer();
startServer();
ZooKeeperServer server = serverFactory.getZooKeeperServer();
- Assert.assertNotNull("Quota is still set",
- server.getZKDatabase().getDataTree().getMaxPrefixWithQuota(path) != null);
+ assertNotNull("Quota is still set", server.getZKDatabase().getDataTree().getMaxPrefixWithQuota(path)
+ != null);
}
@Test
- public void testSetQuota() throws IOException,
- InterruptedException, KeeperException, MalformedPathException {
+ public void testSetQuota() throws IOException, InterruptedException, KeeperException, MalformedPathException {
final ZooKeeper zk = createClient();
String path = "/c1";
String nodeData = "foo";
- zk.create(path, nodeData.getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create(path, nodeData.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
int count = 10;
long bytes = 5L;
@@ -96,44 +90,37 @@ public class ZooKeeperQuotaTest extends ClientBase {
String absoluteLimitPath = Quotas.quotaZookeeper + path + "/" + Quotas.limitNode;
byte[] data = zk.getData(absoluteLimitPath, false, null);
StatsTrack st = new StatsTrack(new String(data));
- Assert.assertEquals(bytes, st.getBytes());
- Assert.assertEquals(count, st.getCount());
+ assertEquals(bytes, st.getBytes());
+ assertEquals(count, st.getCount());
//check the stats
String absoluteStatPath = Quotas.quotaZookeeper + path + "/" + Quotas.statNode;
data = zk.getData(absoluteStatPath, false, null);
st = new StatsTrack(new String(data));
- Assert.assertEquals(nodeData.length(), st.getBytes());
- Assert.assertEquals(1, st.getCount());
+ assertEquals(nodeData.length(), st.getBytes());
+ assertEquals(1, st.getCount());
//create another node
String path2 = "/c1/c2";
String nodeData2 = "bar";
- zk.create(path2, nodeData2.getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create(path2, nodeData2.getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
absoluteStatPath = Quotas.quotaZookeeper + path + "/" + Quotas.statNode;
data = zk.getData(absoluteStatPath, false, null);
st = new StatsTrack(new String(data));
//check the stats
- Assert.assertEquals(nodeData.length() + nodeData2.length(), st.getBytes());
- Assert.assertEquals(2, st.getCount());
+ assertEquals(nodeData.length() + nodeData2.length(), st.getBytes());
+ assertEquals(2, st.getCount());
}
@Test
- public void testSetQuotaWhenSetQuotaOnParentOrChildPath() throws IOException,
- InterruptedException, KeeperException, MalformedPathException {
+ public void testSetQuotaWhenSetQuotaOnParentOrChildPath() throws IOException, InterruptedException, KeeperException, MalformedPathException {
final ZooKeeper zk = createClient();
- zk.create("/c1", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
- zk.create("/c1/c2", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
- zk.create("/c1/c2/c3", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
- zk.create("/c1/c2/c3/c4", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
- zk.create("/c1/c2/c3/c4/c5", "some".getBytes(), Ids.OPEN_ACL_UNSAFE,
- CreateMode.PERSISTENT);
+ zk.create("/c1", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
+ zk.create("/c1/c2", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
+ zk.create("/c1/c2/c3", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
+ zk.create("/c1/c2/c3/c4", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
+ zk.create("/c1/c2/c3/c4/c5", "some".getBytes(), Ids.OPEN_ACL_UNSAFE, CreateMode.PERSISTENT);
//set the quota on the path:/c1/c2/c3
SetQuotaCommand.createQuota(zk, "/c1/c2/c3", 5L, 10);
@@ -141,13 +128,14 @@ public class ZooKeeperQuotaTest extends ClientBase {
try {
SetQuotaCommand.createQuota(zk, "/c1", 5L, 10);
} catch (IllegalArgumentException e) {
- Assert.assertEquals("/c1 has a child /c1/c2/c3 which has a quota", e.getMessage());
+ assertEquals("/c1 has a child /c1/c2/c3 which has a quota", e.getMessage());
}
try {
SetQuotaCommand.createQuota(zk, "/c1/c2/c3/c4/c5", 5L, 10);
} catch (IllegalArgumentException e) {
- Assert.assertEquals("/c1/c2/c3/c4/c5 has a parent /c1/c2/c3 which has a quota", e.getMessage());
+ assertEquals("/c1/c2/c3/c4/c5 has a parent /c1/c2/c3 which has a quota", e.getMessage());
}
}
+
}
View Lorry Controller Status