preparing QueueKeeper and message translation

[openflowplugin.git] / openflowplugin / src / test / java / org / opendaylight / openflowplugin / openflow / md / core / plan / EventFactory.java

diff --git a/openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/plan/EventFactory.java b/openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/plan/EventFactory.java
index d6705c1ccedf56483fc046ed9e03021dbc082a51..0f738e9512423b15179f350a2686014fc39e0d61 100644 (file)
--- a/openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/plan/EventFactory.java
+++ b/openflowplugin/src/test/java/org/opendaylight/openflowplugin/openflow/md/core/plan/EventFactory.java
@@ -8,6 +8,11 @@
 
 package org.opendaylight.openflowplugin.openflow.md.core.plan;
 
+import java.util.Arrays;
+import java.util.HashSet;
+import java.util.concurrent.Callable;
+
+import org.opendaylight.openflowjava.protocol.api.connection.ConnectionReadyListener;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.openflow.protocol.rev130731.OfHeader;
 import org.opendaylight.yangtools.yang.binding.Notification;
 import org.slf4j.Logger;
@@ -90,6 +95,18 @@ public abstract class EventFactory {
         event.setXid(xid);
         return event;
     }
+    
+    /**
+     * @param events
+     * @return wait for all wrapper
+     */
+    public static SwitchTestWaitForAllEvent createDefaultWaitForAllEvent(
+            SwitchTestWaitForRpcEvent... events) {
+        SwitchTestWaitForAllEventImpl eventBag = new SwitchTestWaitForAllEventImpl();
+        HashSet<SwitchTestWaitForRpcEvent> eventsSet = new HashSet<>(Arrays.asList(events));
+        eventBag.setEventBag(eventsSet);
+        return eventBag;
+    }
 
     /**
      * @param xid
@@ -120,4 +137,21 @@ public abstract class EventFactory {
         return setupHeader(xid, DEFAULT_VERSION, builder);
     }
 
+    /**
+     * @param connectionConductor 
+     * @return scenario callback
+     */
+    public static SwitchTestCallbackEventImpl createConnectionReadyCallback(
+            final ConnectionReadyListener connectionConductor) {
+        SwitchTestCallbackEventImpl connectionReadyCallback = new SwitchTestCallbackEventImpl();
+        connectionReadyCallback.setCallable(new Callable<Void>() {
+            @Override
+            public Void call() throws Exception {
+                connectionConductor.onConnectionReady();
+                return null;
+            }
+        });
+        return connectionReadyCallback;
+    }
+
 }