You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@hama.apache.org by tj...@apache.org on 2012/02/08 16:45:16 UTC
svn commit: r1241943 -
/incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java
Author: tjungblut
Date: Wed Feb 8 15:45:16 2012
New Revision: 1241943
URL: http://svn.apache.org/viewvc?rev=1241943&view=rev
Log:
Turning main method to testcase
Modified:
incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java
Modified: incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java
URL: http://svn.apache.org/viewvc/incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java?rev=1241943&r1=1241942&r2=1241943&view=diff
==============================================================================
--- incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java (original)
+++ incubator/hama/trunk/core/src/test/java/org/apache/hama/bsp/message/TestAvroMessageManager.java Wed Feb 8 15:45:16 2012
@@ -9,6 +9,8 @@ import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.util.Random;
+import junit.framework.TestCase;
+
import org.apache.avro.AvroRemoteException;
import org.apache.avro.ipc.NettyServer;
import org.apache.avro.ipc.NettyTransceiver;
@@ -23,21 +25,28 @@ import org.apache.hama.bsp.BooleanMessag
import org.apache.hama.bsp.DoubleMessage;
import org.apache.hama.bsp.IntegerMessage;
-public class TestAvroMessageManager {
+public class TestAvroMessageManager extends TestCase {
private static NettyServer server;
private static Server hadoopServer;
private static long start;
+ public void compareMessengers() throws Exception {
+ BSPMessageBundle randomBundle = getRandomBundle();
+ testAvro(randomBundle);
+ testHadoop(randomBundle);
+ }
+
public static final class MessageSender implements Sender {
@Override
public Void transfer(AvroBSPMessageBundle messagebundle)
throws AvroRemoteException {
try {
- BSPMessageBundle msg = deserializeMessage(messagebundle.data);
+ BSPMessageBundle msg = deserializeMessage(messagebundle.getData());
System.out.println("Received message in "
- + (System.currentTimeMillis() - start) + "ms");
+ + (System.currentTimeMillis() - start) + "ms. Size: "
+ + msg.getMessages().size());
} catch (IOException e) {
e.printStackTrace();
}
@@ -86,20 +95,14 @@ public class TestAvroMessageManager {
return bundle;
}
- public static final void main(String[] args) throws IOException {
- BSPMessageBundle randomBundle = getRandomBundle();
- testAvro(randomBundle);
- testHadoop(randomBundle);
- }
-
private static final void testAvro(BSPMessageBundle bundle)
throws IOException, AvroRemoteException {
server = new NettyServer(new SpecificResponder(Sender.class,
new MessageSender()), new InetSocketAddress(13530));
- NettyTransceiver client = new NettyTransceiver(new InetSocketAddress(server
- .getPort()));
+ NettyTransceiver client = new NettyTransceiver(new InetSocketAddress(
+ server.getPort()));
Sender proxy = (Sender) SpecificRequestor.getClient(Sender.class, client);
AvroBSPMessageBundle msg = new AvroBSPMessageBundle();