Change Intent classes to self contained
Change-Id: I0def86a280646983efb98dbe5937b93b4c9a4885
diff --git a/src/test/java/net/onrc/onos/intent/ShortestPathIntentTest.java b/src/test/java/net/onrc/onos/intent/ShortestPathIntentTest.java
index a0da341..e325be8 100644
--- a/src/test/java/net/onrc/onos/intent/ShortestPathIntentTest.java
+++ b/src/test/java/net/onrc/onos/intent/ShortestPathIntentTest.java
@@ -2,10 +2,15 @@
import static org.junit.Assert.*;
import net.onrc.onos.ofcontroller.networkgraph.NetworkGraph;
+
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
+import com.esotericsoftware.kryo.Kryo;
+import com.esotericsoftware.kryo.io.Input;
+import com.esotericsoftware.kryo.io.Output;
+
public class ShortestPathIntentTest {
NetworkGraph g;
@@ -22,21 +27,26 @@
@Test
public void test() {
+ Kryo kryo = new Kryo();
+ Output output = new Output(1024);
+
ShortestPathIntent intent1 =
- new ShortestPathIntent(g, "1", 1L, 20L, 1L, 4L, 20L, 4L);
+ new ShortestPathIntent("1", 2L, 3L, 4L, 5L, 6L, 7L);
- byte b[] = intent1.toBytes();
+ kryo.writeObject(output, intent1);
+ output.close();
+ Input input = new Input(output.toBytes());
ShortestPathIntent intent2 =
- ShortestPathIntent.fromBytes(g, b);
+ kryo.readObject(input, ShortestPathIntent.class);
+ input.close();
assertEquals("1", intent2.getId());
- assertEquals(Long.valueOf(1), intent2.getSourcePort().getSwitch().getDpid());
- assertEquals(Long.valueOf(20), intent2.getSourcePort().getNumber());
- assertEquals(1L, intent2.getSourceMac().toLong());
- assertEquals(Long.valueOf(4), intent2.getDestinationPort().getSwitch().getDpid());
- assertEquals(Long.valueOf(20), intent2.getDestinationPort().getNumber());
- assertEquals(4L, intent2.getDestinationMac().toLong());
+ assertEquals(2L, intent2.getSrcSwitchDpid());
+ assertEquals(3L, intent2.getSrcPortNumber());
+ assertEquals(4L, intent2.getSrcMac());
+ assertEquals(5L, intent2.getDstSwitchDpid());
+ assertEquals(6L, intent2.getDstPortNumber());
+ assertEquals(7L, intent2.getDstMac());
}
-
}