svn commit: r1089643 - /cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java

[jbellis]

Author: jbellis
Date: Wed Apr  6 21:48:44 2011
New Revision: 1089643

URL: http://svn.apache.org/viewvc?rev=1089643&view=rev
Log:
add SystemTableTest

Added:
    cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java

Added: cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java
URL: 
http://svn.apache.org/viewvc/cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java?rev=1089643&view=auto
==============================================================================
--- cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java 
(added)
+++ cassandra/trunk/test/unit/org/apache/cassandra/db/SystemTableTest.java Wed 
Apr  6 21:48:44 2011
@@ -0,0 +1,34 @@
+package org.apache.cassandra.db;
+
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+
+import com.google.common.base.Charsets;
+import org.junit.Test;
+
+import org.apache.cassandra.dht.BytesToken;
+import org.apache.cassandra.utils.ByteBufferUtil;
+
+public class SystemTableTest
+{
+    @Test
+    public void testLocalToken()
+    {
+        SystemTable.updateToken(new BytesToken(ByteBufferUtil.bytes("token")));
+        assert new String(((BytesToken) SystemTable.getSavedToken()).token, 
Charsets.UTF_8).equals("token");
+
+        SystemTable.updateToken(new 
BytesToken(ByteBufferUtil.bytes("token2")));
+        assert new String(((BytesToken) SystemTable.getSavedToken()).token, 
Charsets.UTF_8).equals("token2");
+    }
+
+    @Test
+    public void testNonLocalToken() throws UnknownHostException
+    {
+        BytesToken token = new BytesToken(ByteBufferUtil.bytes("token3"));
+        InetAddress address = InetAddress.getByName("127.0.0.2");
+        SystemTable.updateToken(address, token);
+        assert SystemTable.loadTokens().get(token).equals(address);
+        SystemTable.removeToken(token);
+        assert !SystemTable.loadTokens().containsKey(token);
+    }
+}