Bug 3302: fix for GroupTable

[groupbasedpolicy.git] / renderers / ofoverlay / src / test / java / org / opendaylight / groupbasedpolicy / renderer / ofoverlay / flow / FlowTableTest.java

diff --git a/renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/flow/FlowTableTest.java b/renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/flow/FlowTableTest.java
index c884bc1bf9de220bccdc14e276cf838723499edf..e155c0c9058669893a0ef34a85126a54fd57fe2b 100755 (executable)
--- a/renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/flow/FlowTableTest.java
+++ b/renderers/ofoverlay/src/test/java/org/opendaylight/groupbasedpolicy/renderer/ofoverlay/flow/FlowTableTest.java
@@ -10,7 +10,7 @@ package org.opendaylight.groupbasedpolicy.renderer.ofoverlay.flow;
 \r
 import java.util.Map;\r
 \r
-import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.PolicyManager.FlowMap;\r
+import org.opendaylight.groupbasedpolicy.renderer.ofoverlay.OfWriter;\r
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.Table;\r
 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;\r
 import org.opendaylight.yang.gen.v1.urn.opendaylight.flow.inventory.rev130819.tables.table.Flow;\r
@@ -24,17 +24,17 @@ public class FlowTableTest extends OfTableTest {
                                          table.getTableId());\r
     }\r
 \r
-    protected FlowMap dosync(Map<String, Flow> flows) throws Exception {\r
-        FlowMap flowMap = policyManager.new FlowMap();\r
+    protected OfWriter dosync(Map<String, Flow> flows) throws Exception {\r
+        OfWriter ofWriter = new OfWriter();\r
         if (flows != null) {\r
             for (String key : flows.keySet()) {\r
                 Flow flow = flows.get(key);\r
                 if (flow != null) {\r
-                    flowMap.writeFlow(nodeId, flow.getTableId(), flow);\r
+                    ofWriter.writeFlow(nodeId, flow.getTableId(), flow);\r
                 }\r
             }\r
         }\r
-        table.sync(nodeId, policyResolver.getCurrentPolicy(), flowMap);\r
-        return flowMap;\r
+        table.sync(nodeId, policyResolver.getCurrentPolicy(), ofWriter);\r
+        return ofWriter;\r
     }\r
 }\r