You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@cassandra.apache.org by jb...@apache.org on 2010/06/14 21:36:40 UTC
svn commit: r954605 -
/cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java
Author: jbellis
Date: Mon Jun 14 19:36:40 2010
New Revision: 954605
URL: http://svn.apache.org/viewvc?rev=954605&view=rev
Log:
convert RackAwareStrategyTest newlines to \n
Modified:
cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java
Modified: cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java
URL: http://svn.apache.org/viewvc/cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java?rev=954605&r1=954604&r2=954605&view=diff
==============================================================================
--- cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java (original)
+++ cassandra/trunk/test/unit/org/apache/cassandra/locator/RackAwareStrategyTest.java Mon Jun 14 19:36:40 2010
@@ -1,172 +1,172 @@
-/*
-* 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
-* regarding copyright ownership. The ASF licenses this file
-* to you under the Apache License, Version 2.0 (the
-* "License"); you may not use this file except in compliance
-* with the License. You may obtain a copy of the License at
-*
-* http://www.apache.org/licenses/LICENSE-2.0
-*
-* Unless required by applicable law or agreed to in writing,
-* software distributed under the License is distributed on an
-* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
-* KIND, either express or implied. See the License for the
-* specific language governing permissions and limitations
-* under the License.
-*/
-package org.apache.cassandra.locator;
-
-import static org.junit.Assert.assertEquals;
-
-import java.net.InetAddress;
-import java.net.UnknownHostException;
-import java.util.ArrayList;
-import java.util.HashMap;
-import java.util.List;
-import java.util.Map;
-
-import org.apache.cassandra.config.DatabaseDescriptor;
-import org.apache.cassandra.dht.BigIntegerToken;
-import org.apache.cassandra.dht.Token;
-
-import org.junit.Before;
-import org.junit.Test;
-
-public class RackAwareStrategyTest
-{
- private List<Token> endpointTokens;
- private List<Token> keyTokens;
- private TokenMetadata tmd;
- private Map<String, ArrayList<InetAddress>> expectedResults;
-
- @Before
- public void init()
- {
- endpointTokens = new ArrayList<Token>();
- keyTokens = new ArrayList<Token>();
- tmd = new TokenMetadata();
- expectedResults = new HashMap<String, ArrayList<InetAddress>>();
- }
-
- /**
- * 4 same rack endpoints
- *
- * @throws UnknownHostException
- */
- @Test
- public void testBigIntegerEndpointsA() throws UnknownHostException
- {
- RackInferringSnitch endpointSnitch = new RackInferringSnitch();
-
- AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
- addEndpoint("0", "5", "254.0.0.1");
- addEndpoint("10", "15", "254.0.0.2");
- addEndpoint("20", "25", "254.0.0.3");
- addEndpoint("30", "35", "254.0.0.4");
-
- expectedResults.put("5", buildResult("254.0.0.2", "254.0.0.3", "254.0.0.4"));
- expectedResults.put("15", buildResult("254.0.0.3", "254.0.0.4", "254.0.0.1"));
- expectedResults.put("25", buildResult("254.0.0.4", "254.0.0.1", "254.0.0.2"));
- expectedResults.put("35", buildResult("254.0.0.1", "254.0.0.2", "254.0.0.3"));
-
- runTestForReplicatedTables(strategy);
- }
-
- /**
- * 3 same rack endpoints
- * 1 external datacenter
- *
- * @throws UnknownHostException
- */
- @Test
- public void testBigIntegerEndpointsB() throws UnknownHostException
- {
- RackInferringSnitch endpointSnitch = new RackInferringSnitch();
-
- AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
- addEndpoint("0", "5", "254.0.0.1");
- addEndpoint("10", "15", "254.0.0.2");
- addEndpoint("20", "25", "254.1.0.3");
- addEndpoint("30", "35", "254.0.0.4");
-
- expectedResults.put("5", buildResult("254.0.0.2", "254.1.0.3", "254.0.0.4"));
- expectedResults.put("15", buildResult("254.1.0.3", "254.0.0.4", "254.0.0.1"));
- expectedResults.put("25", buildResult("254.0.0.4", "254.1.0.3", "254.0.0.1"));
- expectedResults.put("35", buildResult("254.0.0.1", "254.1.0.3", "254.0.0.2"));
-
- runTestForReplicatedTables(strategy);
- }
-
- /**
- * 2 same rack endpoints
- * 1 same datacenter, different rack endpoints
- * 1 external datacenter
- *
- * @throws UnknownHostException
- */
- @Test
- public void testBigIntegerEndpointsC() throws UnknownHostException
- {
- RackInferringSnitch endpointSnitch = new RackInferringSnitch();
-
- AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
- addEndpoint("0", "5", "254.0.0.1");
- addEndpoint("10", "15", "254.0.0.2");
- addEndpoint("20", "25", "254.0.1.3");
- addEndpoint("30", "35", "254.1.0.4");
-
- expectedResults.put("5", buildResult("254.0.0.2", "254.0.1.3", "254.1.0.4"));
- expectedResults.put("15", buildResult("254.0.1.3", "254.1.0.4", "254.0.0.1"));
- expectedResults.put("25", buildResult("254.1.0.4", "254.0.0.1", "254.0.0.2"));
- expectedResults.put("35", buildResult("254.0.0.1", "254.0.1.3", "254.1.0.4"));
-
- runTestForReplicatedTables(strategy);
- }
-
- private void runTestForReplicatedTables(AbstractReplicationStrategy strategy) throws UnknownHostException
- {
- for (String table : DatabaseDescriptor.getNonSystemTables())
- {
- if (DatabaseDescriptor.getReplicationFactor(table) == 3)
- testGetEndpoints(strategy, keyTokens.toArray(new Token[0]), table);
- }
- }
-
- private ArrayList<InetAddress> buildResult(String... addresses) throws UnknownHostException
- {
- ArrayList<InetAddress> result = new ArrayList<InetAddress>();
- for (String address : addresses)
- {
- result.add(InetAddress.getByName(address));
- }
- return result;
- }
-
- private void addEndpoint(String endpointTokenID, String keyTokenID, String endpointAddress) throws UnknownHostException
- {
- BigIntegerToken endpointToken = new BigIntegerToken(endpointTokenID);
- endpointTokens.add(endpointToken);
-
- BigIntegerToken keyToken = new BigIntegerToken(keyTokenID);
- keyTokens.add(keyToken);
-
- InetAddress ep = InetAddress.getByName(endpointAddress);
- tmd.updateNormalToken(endpointToken, ep);
- }
-
- private void testGetEndpoints(AbstractReplicationStrategy strategy, Token[] keyTokens, String table) throws UnknownHostException
- {
- for (Token keyToken : keyTokens)
- {
- List<InetAddress> endpoints = strategy.getNaturalEndpoints(keyToken, tmd, table);
- for (int j = 0; j < endpoints.size(); j++)
- {
- ArrayList<InetAddress> hostsExpected = expectedResults.get(keyToken.toString());
- assertEquals(endpoints.get(j), hostsExpected.get(j));
- }
- }
- }
-
-}
\ No newline at end of file
+/*
+* 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
+* regarding copyright ownership. The ASF licenses this file
+* to you under the Apache License, Version 2.0 (the
+* "License"); you may not use this file except in compliance
+* with the License. You may obtain a copy of the License at
+*
+* http://www.apache.org/licenses/LICENSE-2.0
+*
+* Unless required by applicable law or agreed to in writing,
+* software distributed under the License is distributed on an
+* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+* KIND, either express or implied. See the License for the
+* specific language governing permissions and limitations
+* under the License.
+*/
+package org.apache.cassandra.locator;
+
+import static org.junit.Assert.assertEquals;
+
+import java.net.InetAddress;
+import java.net.UnknownHostException;
+import java.util.ArrayList;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
+
+import org.apache.cassandra.config.DatabaseDescriptor;
+import org.apache.cassandra.dht.BigIntegerToken;
+import org.apache.cassandra.dht.Token;
+
+import org.junit.Before;
+import org.junit.Test;
+
+public class RackAwareStrategyTest
+{
+ private List<Token> endpointTokens;
+ private List<Token> keyTokens;
+ private TokenMetadata tmd;
+ private Map<String, ArrayList<InetAddress>> expectedResults;
+
+ @Before
+ public void init()
+ {
+ endpointTokens = new ArrayList<Token>();
+ keyTokens = new ArrayList<Token>();
+ tmd = new TokenMetadata();
+ expectedResults = new HashMap<String, ArrayList<InetAddress>>();
+ }
+
+ /**
+ * 4 same rack endpoints
+ *
+ * @throws UnknownHostException
+ */
+ @Test
+ public void testBigIntegerEndpointsA() throws UnknownHostException
+ {
+ RackInferringSnitch endpointSnitch = new RackInferringSnitch();
+
+ AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
+ addEndpoint("0", "5", "254.0.0.1");
+ addEndpoint("10", "15", "254.0.0.2");
+ addEndpoint("20", "25", "254.0.0.3");
+ addEndpoint("30", "35", "254.0.0.4");
+
+ expectedResults.put("5", buildResult("254.0.0.2", "254.0.0.3", "254.0.0.4"));
+ expectedResults.put("15", buildResult("254.0.0.3", "254.0.0.4", "254.0.0.1"));
+ expectedResults.put("25", buildResult("254.0.0.4", "254.0.0.1", "254.0.0.2"));
+ expectedResults.put("35", buildResult("254.0.0.1", "254.0.0.2", "254.0.0.3"));
+
+ runTestForReplicatedTables(strategy);
+ }
+
+ /**
+ * 3 same rack endpoints
+ * 1 external datacenter
+ *
+ * @throws UnknownHostException
+ */
+ @Test
+ public void testBigIntegerEndpointsB() throws UnknownHostException
+ {
+ RackInferringSnitch endpointSnitch = new RackInferringSnitch();
+
+ AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
+ addEndpoint("0", "5", "254.0.0.1");
+ addEndpoint("10", "15", "254.0.0.2");
+ addEndpoint("20", "25", "254.1.0.3");
+ addEndpoint("30", "35", "254.0.0.4");
+
+ expectedResults.put("5", buildResult("254.0.0.2", "254.1.0.3", "254.0.0.4"));
+ expectedResults.put("15", buildResult("254.1.0.3", "254.0.0.4", "254.0.0.1"));
+ expectedResults.put("25", buildResult("254.0.0.4", "254.1.0.3", "254.0.0.1"));
+ expectedResults.put("35", buildResult("254.0.0.1", "254.1.0.3", "254.0.0.2"));
+
+ runTestForReplicatedTables(strategy);
+ }
+
+ /**
+ * 2 same rack endpoints
+ * 1 same datacenter, different rack endpoints
+ * 1 external datacenter
+ *
+ * @throws UnknownHostException
+ */
+ @Test
+ public void testBigIntegerEndpointsC() throws UnknownHostException
+ {
+ RackInferringSnitch endpointSnitch = new RackInferringSnitch();
+
+ AbstractReplicationStrategy strategy = new RackAwareStrategy(tmd, endpointSnitch);
+ addEndpoint("0", "5", "254.0.0.1");
+ addEndpoint("10", "15", "254.0.0.2");
+ addEndpoint("20", "25", "254.0.1.3");
+ addEndpoint("30", "35", "254.1.0.4");
+
+ expectedResults.put("5", buildResult("254.0.0.2", "254.0.1.3", "254.1.0.4"));
+ expectedResults.put("15", buildResult("254.0.1.3", "254.1.0.4", "254.0.0.1"));
+ expectedResults.put("25", buildResult("254.1.0.4", "254.0.0.1", "254.0.0.2"));
+ expectedResults.put("35", buildResult("254.0.0.1", "254.0.1.3", "254.1.0.4"));
+
+ runTestForReplicatedTables(strategy);
+ }
+
+ private void runTestForReplicatedTables(AbstractReplicationStrategy strategy) throws UnknownHostException
+ {
+ for (String table : DatabaseDescriptor.getNonSystemTables())
+ {
+ if (DatabaseDescriptor.getReplicationFactor(table) == 3)
+ testGetEndpoints(strategy, keyTokens.toArray(new Token[0]), table);
+ }
+ }
+
+ private ArrayList<InetAddress> buildResult(String... addresses) throws UnknownHostException
+ {
+ ArrayList<InetAddress> result = new ArrayList<InetAddress>();
+ for (String address : addresses)
+ {
+ result.add(InetAddress.getByName(address));
+ }
+ return result;
+ }
+
+ private void addEndpoint(String endpointTokenID, String keyTokenID, String endpointAddress) throws UnknownHostException
+ {
+ BigIntegerToken endpointToken = new BigIntegerToken(endpointTokenID);
+ endpointTokens.add(endpointToken);
+
+ BigIntegerToken keyToken = new BigIntegerToken(keyTokenID);
+ keyTokens.add(keyToken);
+
+ InetAddress ep = InetAddress.getByName(endpointAddress);
+ tmd.updateNormalToken(endpointToken, ep);
+ }
+
+ private void testGetEndpoints(AbstractReplicationStrategy strategy, Token[] keyTokens, String table) throws UnknownHostException
+ {
+ for (Token keyToken : keyTokens)
+ {
+ List<InetAddress> endpoints = strategy.getNaturalEndpoints(keyToken, tmd, table);
+ for (int j = 0; j < endpoints.size(); j++)
+ {
+ ArrayList<InetAddress> hostsExpected = expectedResults.get(keyToken.toString());
+ assertEquals(endpoints.get(j), hostsExpected.get(j));
+ }
+ }
+ }
+
+}