update test codes
diff --git a/src/main/java/net/onrc/onos/util/GraphDBOperation.java b/src/main/java/net/onrc/onos/util/GraphDBOperation.java
index 1699bde..58cdc22 100644
--- a/src/main/java/net/onrc/onos/util/GraphDBOperation.java
+++ b/src/main/java/net/onrc/onos/util/GraphDBOperation.java
@@ -26,7 +26,7 @@
}
@Override
- public ISwitchObject newSwitch(String dpid) {
+ public ISwitchObject newSwitch(String dpid) {
FramedGraph<TitanGraph> fg = conn.getFramedGraph();
ISwitchObject obj = fg.addVertex(null,ISwitchObject.class);
if (obj != null) {
diff --git a/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java b/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
index 04ef815..01f244c 100644
--- a/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
+++ b/src/test/java/net/onrc/onos/util/GraphDBOperationTest.java
@@ -86,8 +86,7 @@
switches = op.getAllSwitches();
assertFalse(switches.iterator().hasNext());
- ISwitchObject sw = op.newSwitch();
- sw.setDPID("123");
+ ISwitchObject sw = op.newSwitch("123");
sw.setState(SwitchState.ACTIVE.toString());
conn.endTx(Transaction.COMMIT);
@@ -110,8 +109,7 @@
switches = op.getAllSwitches();
assertFalse(switches.iterator().hasNext());
- ISwitchObject sw = op.newSwitch();
- sw.setDPID("123");
+ ISwitchObject sw = op.newSwitch("123");
sw.setState(SwitchState.ACTIVE.toString());
conn.endTx(Transaction.COMMIT);