/*
* Copyright 2009 Amit Levy, Jeff Prouty, Rylan Hawkins
*
* Licensed 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 edu.washington.cs.cse490h.donut.service;
import static org.easymock.EasyMock.aryEq;
import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.eq;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.junit.Assert.assertArrayEquals;
import static org.junit.Assert.assertEquals;
import java.util.Set;
import java.util.TreeSet;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import edu.washington.cs.cse490h.donut.Constants;
import edu.washington.cs.cse490h.donut.business.DataPair;
import edu.washington.cs.cse490h.donut.business.EntryKey;
import edu.washington.cs.cse490h.donut.business.KeyId;
import edu.washington.cs.cse490h.donut.business.Node;
import edu.washington.cs.cse490h.donut.business.TNode;
import edu.washington.cs.cse490h.donut.service.application.DonutHashTableService;
public class NodeLocatorTest {
private static final EntryKey ENTRY_KEY = new EntryKey(new KeyId(1), "key");
LocatorClientFactory clientFactoryMock;
KeyLocator.Iface nextLocatorMock;
DonutHashTableService service;
@Before
public void setUp() throws Exception {
clientFactoryMock = createMock(LocatorClientFactory.class);
nextLocatorMock = createMock(KeyLocator.Iface.class);
service = createMock(DonutHashTableService.class);
}
@After
public void tearDown() throws Exception {
verify(clientFactoryMock);
verify(nextLocatorMock);
verify(service);
}
@Test
public void testFindSuccessor_ImmediateSuccessor() throws Exception {
Node node1 = new Node("node1", 8080, new KeyId(100));
TNode node2 = new TNode("node2", 8080, new KeyId(900));
node1.setSuccessor(node2);
NodeLocator nodeLocator = new NodeLocator(node1, null, null);
replay(clientFactoryMock, nextLocatorMock, service);
assertEquals(node2, nodeLocator.findSuccessor(new KeyId(456)));
}
@Test
public void testFindSuccessor_NotImmediateSuccessor() throws Exception {
KeyId entryId = new KeyId(1024);
Node node1 = new Node("node1", 8080, new KeyId(100));
TNode node2 = new TNode("node2", 8080, new KeyId(900));
TNode resultNode = new TNode("resultNode", 8080, null);
node1.setSuccessor(node2);
NodeLocator nodeLocator = new NodeLocator(node1, null, clientFactoryMock);
// ClientFactory Expectations:
expect(clientFactoryMock.get(node2)).andReturn(nextLocatorMock);
clientFactoryMock.release(node2);
// NextLocatorMock Expectations:
expect(nextLocatorMock.findSuccessor(entryId)).andReturn(resultNode);
replay(clientFactoryMock, nextLocatorMock, service);
assertEquals(resultNode, nodeLocator.findSuccessor(entryId));
}
@Test(expected = DataNotFoundException.class)
public void testGet_Dne() throws Exception {
NodeLocator nodeLocator = new NodeLocator(null, service, null);
expect(service.get(ENTRY_KEY)).andReturn(null);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.get(ENTRY_KEY);
}
@Test
public void testGet_Exists() throws Exception {
NodeLocator nodeLocator = new NodeLocator(null, service, null);
String value = "Hello World";
expect(service.get(ENTRY_KEY)).andReturn(new DataPair(value.getBytes(), 0));
replay(clientFactoryMock, nextLocatorMock, service);
assertArrayEquals(value.getBytes(), nodeLocator.get(ENTRY_KEY));
}
@Test
public void testReplicatePut_NoReplica() throws Exception {
NodeLocator nodeLocator = new NodeLocator(null, service, null);
String value = "Hello World";
service.put(eq(ENTRY_KEY), aryEq(value.getBytes()), eq(0));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.replicatePut(ENTRY_KEY, value.getBytes(), 0);
}
@Test
public void testReplicatePut_WithReplica() throws Exception {
Node node = new Node(null);
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
String value = "Hello World";
expect(clientFactoryMock.get(node.getSuccessor())).andReturn(nextLocatorMock);
clientFactoryMock.release(node.getSuccessor());
nextLocatorMock.replicatePut(eq(ENTRY_KEY), aryEq(value.getBytes()), eq(1));
service.put(eq(ENTRY_KEY), aryEq(value.getBytes()), eq(2));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.replicatePut(ENTRY_KEY, value.getBytes(), 2);
}
@Test
public void testReplicateRemove_NoReplica() throws Exception {
NodeLocator nodeLocator = new NodeLocator(null, service, null);
service.remove(ENTRY_KEY);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.replicateRemove(ENTRY_KEY, 0);
}
@Test
public void testReplicateRemove_WithReplica() throws Exception {
Node node = new Node(null);
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
expect(clientFactoryMock.get(node.getSuccessor())).andReturn(nextLocatorMock);
clientFactoryMock.release(node.getSuccessor());
nextLocatorMock.replicateRemove(ENTRY_KEY, 1);
service.remove(ENTRY_KEY);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.replicateRemove(ENTRY_KEY, 2);
}
@Test
public void testPut() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
node.setPredecessor(new TNode("predecessor", 1234, new KeyId(0)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
expect(clientFactoryMock.get(node.getSuccessor())).andReturn(nextLocatorMock);
clientFactoryMock.release(node.getSuccessor());
nextLocatorMock.replicatePut(eq(ENTRY_KEY), aryEq("data".getBytes()),
eq(Constants.SUCCESSOR_LIST_SIZE - 1));
service.put(eq(ENTRY_KEY), aryEq("data".getBytes()), eq(Constants.SUCCESSOR_LIST_SIZE));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.put(ENTRY_KEY, "data".getBytes());
}
@Test(expected = NotResponsibleForId.class)
public void testPut_NotResponsible() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
node.setPredecessor(new TNode("predecessor", 1234, new KeyId(0)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.put(new EntryKey(new KeyId(-1), "key"), "data".getBytes());
}
@Test
public void testRemove() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
node.setPredecessor(new TNode("predecessor", 1234, new KeyId(0)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
expect(clientFactoryMock.get(node.getSuccessor())).andReturn(nextLocatorMock);
clientFactoryMock.release(node.getSuccessor());
nextLocatorMock.replicateRemove(eq(ENTRY_KEY), eq(Constants.SUCCESSOR_LIST_SIZE - 1));
service.remove(eq(ENTRY_KEY));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.remove(ENTRY_KEY);
}
@Test(expected = NotResponsibleForId.class)
public void testRemove_NotResponsible() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setSuccessor(new TNode("successor", 1234, new KeyId(123)));
node.setPredecessor(new TNode("predecessor", 1234, new KeyId(0)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.remove(new EntryKey(new KeyId(-1), "key"));
}
@Test
public void getPredecessor() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setPredecessor(new TNode("predecessor", 1234, new KeyId(0)));
NodeLocator nodeLocator = new NodeLocator(node, null, null);
replay(clientFactoryMock, nextLocatorMock, service);
assertEquals(node.getPredecessor(), nodeLocator.getPredecessor());
}
@Test(expected = NodeNotFoundException.class)
public void getPredecessor_Null() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setPredecessor(null);
NodeLocator nodeLocator = new NodeLocator(node, null, null);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.getPredecessor();
}
@Test
public void testNotify_NotPredecessor() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
node.setPredecessor(new TNode("realPred", 8080, new KeyId(999)));
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.notify(new TNode("mynode", 8080, new KeyId(1)));
}
@Test
public void testNotify_IsPredecessor() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
TNode oldPredecessor = new TNode("realPred", 8080, new KeyId(1));
TNode newPredecessor = new TNode("mynode", 8080, new KeyId(999));
node.setPredecessor(oldPredecessor);
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
EntryKey key = new EntryKey(new KeyId(1234), "hello");
Set<EntryKey> keys = new TreeSet<EntryKey>();
keys.add(key);
expect(clientFactoryMock.get(newPredecessor)).andReturn(nextLocatorMock);
clientFactoryMock.release(newPredecessor);
expect(nextLocatorMock.getDataRange(node.getNodeId(), newPredecessor.getNodeId()))
.andReturn(keys);
expect(nextLocatorMock.get(key)).andReturn("world".getBytes());
service.put(eq(key), aryEq("world".getBytes()), eq(Constants.SUCCESSOR_LIST_SIZE));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.notify(newPredecessor);
}
@Test
public void testNotify_WasNullPredecessor() throws Exception {
Node node = new Node(null, 8080, new KeyId(1000));
TNode successor = new TNode("successor", 8080, new KeyId(1500));
TNode newPredecessor = new TNode("mynode", 8080, new KeyId(999));
node.setPredecessor(null);
node.setSuccessor(successor);
NodeLocator nodeLocator = new NodeLocator(node, service, clientFactoryMock);
EntryKey key0 = new EntryKey(new KeyId(4321), "test");
Set<EntryKey> keys0 = new TreeSet<EntryKey>();
keys0.add(key0);
EntryKey key1 = new EntryKey(new KeyId(1234), "hello");
Set<EntryKey> keys1 = new TreeSet<EntryKey>();
keys1.add(key1);
expect(clientFactoryMock.get(successor)).andReturn(nextLocatorMock);
clientFactoryMock.release(successor);
expect(nextLocatorMock.getDataRange(newPredecessor.getNodeId(), node.getNodeId()))
.andReturn(keys0);
expect(nextLocatorMock.get(key0)).andReturn("testing".getBytes());
service.put(eq(key0), aryEq("testing".getBytes()), eq(Constants.SUCCESSOR_LIST_SIZE));
expect(clientFactoryMock.get(newPredecessor)).andReturn(nextLocatorMock);
clientFactoryMock.release(newPredecessor);
expect(nextLocatorMock.getDataRange(node.getNodeId(), newPredecessor.getNodeId()))
.andReturn(keys1);
expect(nextLocatorMock.get(key1)).andReturn("world".getBytes());
service.put(eq(key1), aryEq("world".getBytes()), eq(Constants.SUCCESSOR_LIST_SIZE));
replay(clientFactoryMock, nextLocatorMock, service);
nodeLocator.notify(newPredecessor);
}
}