private V3LcapMessage makeTestVoteMessage(Collection voteBlocks) throws IOException {
mPollMgr.setStateDir("key", tempDir);
V3LcapMessage msg =
new V3LcapMessage(
"ArchivalID_2",
"key",
"Plug42",
m_testBytes,
m_testBytes,
V3LcapMessage.MSG_VOTE,
987654321,
m_testID,
tempDir,
theDaemon);
// Set msg vote blocks.
for (Iterator ix = voteBlocks.iterator(); ix.hasNext(); ) {
msg.addVoteBlock((VoteBlock) ix.next());
}
msg.setHashAlgorithm(LcapMessage.getDefaultHashAlgorithm());
msg.setArchivalId(m_archivalID);
msg.setPluginVersion("PlugVer42");
return msg;
}
public void testPollDuration() throws Exception {
TimeBase.setSimulated(TimeBase.nowMs());
V3LcapMessage src = this.makePollMessage(6 * Constants.WEEK);
InputStream srcStream = src.getInputStream();
V3LcapMessage copy = new V3LcapMessage(srcStream, tempDir, theDaemon);
assertEqualMessages(src, copy);
assertEquals(6 * Constants.WEEK, copy.getDuration());
}
public void testNullPollNak() throws Exception {
V3LcapMessage src = this.makePollAckMessage(null);
InputStream srcStream = src.getInputStream();
V3LcapMessage copy = new V3LcapMessage(srcStream, tempDir, theDaemon);
assertEqualMessages(src, copy);
assertNull(src.getNak());
assertNull(copy.getNak());
}
public void testUnknownPollNak() throws Exception {
MyV3LcapMessage src = makePollAckMessage(V3LcapMessage.PollNak.NAK_NO_TIME);
src.setTestNak("KNACKERED");
InputStream srcStream = src.getInputStream();
V3LcapMessage copy = new V3LcapMessage(srcStream, tempDir, theDaemon);
assertEqualMessages(src, copy);
assertNotNull(src.getNak());
assertNotNull(copy.getNak());
assertEquals(V3LcapMessage.PollNak.NAK_NO_TIME, src.getNak());
assertEquals(V3LcapMessage.PollNak.NAK_UNKNOWN, copy.getNak());
}
public void testDiskBasedStreamEncodingTest() throws Exception {
// Make a list of vote blocks large enough to trigger on-disk
// vote message creation.
List testVoteBlocks = V3TestUtils.makeVoteBlockList(21);
V3LcapMessage testMsg = makeTestVoteMessage(testVoteBlocks);
assertTrue(testMsg.m_voteBlocks instanceof DiskVoteBlocks);
// Encode the test message.
InputStream is = testMsg.getInputStream();
V3LcapMessage decodedMsg = new V3LcapMessage(is, tempDir, theDaemon);
// Ensure that the decoded message matches the test message.
assertEqualMessages(testMsg, decodedMsg);
}
public void testNoOpEncoding() throws Exception {
V3LcapMessage noopMsg =
V3LcapMessage.makeNoOpMsg(m_testID, m_testBytes, m_testBytes, theDaemon);
InputStream fromMsg = noopMsg.getInputStream();
V3LcapMessage msg = new V3LcapMessage(fromMsg, tempDir, theDaemon);
// now test to see if we got back what we started with
assertTrue(m_testID == msg.getOriginatorId());
assertEquals(V3LcapMessage.MSG_NO_OP, msg.getOpcode());
assertEquals(m_testBytes, msg.getPollerNonce());
assertEquals(m_testBytes, msg.getVoterNonce());
assertEquals(null, msg.getVoterNonce2());
}
public void testTestMessageToString() throws IOException {
String expectedResult =
"[V3LcapMessage: from "
+ m_testID.toString()
+ ", Vote AUID: TestAU_1.0 "
+ "Key:key "
+ "PN:AQIDBAUGBwgJAAECAwQFBgcICQA= "
+ "VN:AQIDBAUGBwgJAAECAwQFBgcICQA= "
+ "B:10 ver 3 rev 5]";
assertEquals(expectedResult, m_testMsg.toString());
}
public void testNonNullPollNak1() throws Exception {
V3LcapMessage src = this.makePollAckMessage(V3LcapMessage.PollNak.NAK_GROUP_MISMATCH);
InputStream srcStream = src.getInputStream();
V3LcapMessage copy = new V3LcapMessage(srcStream, tempDir, theDaemon);
assertEqualMessages(src, copy);
assertNotNull(src.getNak());
assertNotNull(copy.getNak());
assertEquals(V3LcapMessage.PollNak.NAK_GROUP_MISMATCH, src.getNak());
assertEquals(V3LcapMessage.PollNak.NAK_GROUP_MISMATCH, copy.getNak());
}
public void testDiskRepairMessage() throws Exception {
int len = 100 * 1024;
byte[] repairData = ByteArray.makeRandomBytes(len);
V3LcapMessage src = makeRepairMessage(repairData);
assertEquals(len, src.getRepairDataLength());
assertEquals(V3LcapMessage.EST_ENCODED_HEADER_LENGTH + len, src.getEstimatedEncodedLength());
InputStream srcStream = src.getInputStream();
V3LcapMessage copy = new V3LcapMessage(srcStream, tempDir, theDaemon);
assertEqualMessages(src, copy);
assertEquals(len, copy.getRepairDataLength());
assertEquals(V3LcapMessage.EST_ENCODED_HEADER_LENGTH + len, src.getEstimatedEncodedLength());
InputStream in = copy.getRepairDataInputStream();
assertTrue(in + "", in instanceof FileInputStream);
ByteArrayOutputStream out = new ByteArrayOutputStream();
StreamUtil.copy(in, out);
byte[] repairCopy = out.toByteArray();
assertEquals(repairData, repairCopy);
// ensure that repeated delete doesn't cause error
copy.delete();
copy.delete();
}
private V3LcapMessage makeRepairMessage(byte[] repairData) {
V3LcapMessage msg =
new V3LcapMessage(
"ArchivalID_2",
"key",
"Plug42",
m_testBytes,
m_testBytes,
V3LcapMessage.MSG_REPAIR_REP,
987654321,
m_testID,
tempDir,
theDaemon);
msg.setHashAlgorithm(LcapMessage.getDefaultHashAlgorithm());
msg.setTargetUrl(m_url);
msg.setArchivalId(m_archivalID);
msg.setPluginVersion("PlugVer42");
msg.setRepairDataLength(repairData.length);
msg.setRepairProps(m_repairProps);
msg.setInputStream(new ByteArrayInputStream(repairData));
return msg;
}
private void assertEqualMessages(V3LcapMessage a, V3LcapMessage b) throws Exception {
assertTrue(a.getOriginatorId() == b.getOriginatorId());
assertEquals(a.getOpcode(), b.getOpcode());
assertEquals(a.getTargetUrl(), b.getTargetUrl());
assertEquals(a.getArchivalId(), b.getArchivalId());
assertEquals(a.getProtocolVersion(), b.getProtocolVersion());
assertEquals(a.getPollerNonce(), b.getPollerNonce());
assertEquals(a.getVoterNonce(), b.getVoterNonce());
assertEquals(a.getVoterNonce2(), b.getVoterNonce2());
assertEquals(a.getPluginVersion(), b.getPluginVersion());
assertEquals(a.getHashAlgorithm(), b.getHashAlgorithm());
assertEquals(a.isVoteComplete(), b.isVoteComplete());
assertEquals(a.getRepairDataLength(), b.getRepairDataLength());
assertEquals(a.getLastVoteBlockURL(), b.getLastVoteBlockURL());
assertIsomorphic(a.getNominees(), b.getNominees());
List aBlocks = new ArrayList();
List bBlocks = new ArrayList();
for (VoteBlocksIterator iter = a.getVoteBlockIterator(); iter.hasNext(); ) {
aBlocks.add(iter.next());
}
for (VoteBlocksIterator iter = b.getVoteBlockIterator(); iter.hasNext(); ) {
bBlocks.add(iter.next());
}
assertTrue(aBlocks.equals(bBlocks));
// TODO: Figure out how to test time.
}
public void testRequestMessageCreation() throws Exception {
V3LcapMessage reqMsg =
new V3LcapMessage(
"ArchivalID_2",
"key",
"Plug42",
m_testBytes,
m_testBytes,
V3LcapMessage.MSG_REPAIR_REQ,
987654321,
m_testID,
tempDir,
theDaemon);
reqMsg.setTargetUrl("http://foo.com/");
for (Iterator ix = m_testVoteBlocks.iterator(); ix.hasNext(); ) {
reqMsg.addVoteBlock((VoteBlock) ix.next());
}
assertEquals(3, reqMsg.getProtocolVersion());
assertEquals("Plug42", reqMsg.getPluginVersion());
assertTrue(m_testID == reqMsg.getOriginatorId());
assertEquals(V3LcapMessage.MSG_REPAIR_REQ, reqMsg.getOpcode());
assertEquals("ArchivalID_2", reqMsg.getArchivalId());
assertEquals("http://foo.com/", reqMsg.getTargetUrl());
assertEquals(m_testBytes, reqMsg.getPollerNonce());
assertEquals(m_testBytes, reqMsg.getVoterNonce());
assertEquals(null, reqMsg.getVoterNonce2());
List aBlocks = new ArrayList();
List bBlocks = new ArrayList();
for (VoteBlocksIterator iter = m_testMsg.getVoteBlockIterator(); iter.hasNext(); ) {
aBlocks.add(iter.next());
}
for (VoteBlocksIterator iter = reqMsg.getVoteBlockIterator(); iter.hasNext(); ) {
bBlocks.add(iter.next());
}
assertEquals(aBlocks, bBlocks);
// Actual size of test vote blocks is unpredictable
assertTrue(reqMsg.getEstimatedEncodedLength() > V3LcapMessage.EST_ENCODED_HEADER_LENGTH);
}
public void testRandomNoOpMessageCreation() throws Exception {
V3LcapMessage noopMsg1 = V3LcapMessage.makeNoOpMsg(m_testID, theDaemon);
V3LcapMessage noopMsg2 = V3LcapMessage.makeNoOpMsg(m_testID, theDaemon);
// now check the fields we expect to be valid
assertEquals(V3LcapMessage.MSG_NO_OP, noopMsg1.getOpcode());
assertEquals(V3LcapMessage.MSG_NO_OP, noopMsg2.getOpcode());
assertTrue(noopMsg1.getOriginatorId() == m_testID);
assertTrue(noopMsg1.getOriginatorId() == noopMsg2.getOriginatorId());
assertFalse(noopMsg1.getPollerNonce() == noopMsg2.getPollerNonce());
assertFalse(noopMsg1.getVoterNonce() == noopMsg2.getVoterNonce());
assertEquals(null, noopMsg1.getVoterNonce2());
assertEquals(null, noopMsg2.getVoterNonce2());
assertEquals(null, noopMsg1.getVoteBlocks());
assertEquals(null, noopMsg2.getVoteBlocks());
}
public void getGroup() throws Exception {
m_testMsg.setGroups(null);
assertNull(m_testMsg.getGroups());
assertNull(m_testMsg.getGroupList());
m_testMsg.setGroups("foo");
assertEquals("foo", m_testMsg.getGroups());
assertEquals(ListUtil.list("foo"), m_testMsg.getGroupList());
m_testMsg.setGroups("foo;bar");
assertEquals("foo;bar", m_testMsg.getGroups());
assertEquals(ListUtil.list("foo", "bar"), m_testMsg.getGroupList());
m_testMsg.setGroups("foo;bar;baz");
assertEquals("foo;bar;baz", m_testMsg.getGroups());
assertEquals(ListUtil.list("foo", "bar", "baz"), m_testMsg.getGroupList());
}
public void testNoOpMessageToString() throws IOException {
V3LcapMessage noopMsg =
V3LcapMessage.makeNoOpMsg(m_testID, m_testBytes, m_testBytes, theDaemon);
String expectedResult = "[V3LcapMessage: from " + m_testID.toString() + ", NoOp]";
assertEquals(expectedResult, noopMsg.toString());
}
public void storeProps() throws IOException {
super.storeProps();
if (testNak != null) {
m_props.setProperty("nak", testNak);
}
}
public void testNoOpMessageCreation() throws Exception {
// Without voterNonce2
V3LcapMessage noopMsg =
V3LcapMessage.makeNoOpMsg(m_testID, m_testBytes, m_testBytes, theDaemon);
// With voterNonce2
V3LcapMessage noopMsg2 =
V3LcapMessage.makeNoOpMsg(m_testID, m_testBytes, m_testBytes, m_testBytes, theDaemon);
// now check the fields we expect to be valid
assertEquals(V3LcapMessage.MSG_NO_OP, noopMsg.getOpcode());
assertTrue(m_testID == noopMsg.getOriginatorId());
assertEquals(m_testBytes, noopMsg.getPollerNonce());
assertEquals(m_testBytes, noopMsg.getVoterNonce());
assertEquals(null, noopMsg.getVoterNonce2());
assertEquals(null, noopMsg.getVoteBlocks());
assertEquals(V3LcapMessage.EST_ENCODED_HEADER_LENGTH, noopMsg.getEstimatedEncodedLength());
// Same for msg with voterNonce2
assertEquals(V3LcapMessage.MSG_NO_OP, noopMsg2.getOpcode());
assertTrue(m_testID == noopMsg2.getOriginatorId());
assertEquals(m_testBytes, noopMsg2.getPollerNonce());
assertEquals(m_testBytes, noopMsg2.getVoterNonce());
assertEquals(m_testBytes, noopMsg2.getVoterNonce2());
assertEquals(null, noopMsg2.getVoteBlocks());
assertEquals(V3LcapMessage.EST_ENCODED_HEADER_LENGTH, noopMsg2.getEstimatedEncodedLength());
}