Make the test stable

Author: style95

tests/src/test/scala/org/apache/openwhisk/core/containerpool/v2/test/FunctionPullingContainerProxyTests.scala

@@ -31,7 +31,7 @@ import org.apache.openwhisk.common.{GracefulShutdown, Logging, TransactionId}
 import org.apache.openwhisk.core.ack.ActiveAck
 import org.apache.openwhisk.core.connector.{AcknowledegmentMessage, ActivationMessage}
 import org.apache.openwhisk.core.containerpool.logging.LogCollectingException
-import org.apache.openwhisk.core.containerpool.v2._
+import org.apache.openwhisk.core.containerpool.v2.{ContainerCreatedData, _}
 import org.apache.openwhisk.core.containerpool.{
   ContainerRemoved,
   Paused => _,
@@ -1575,6 +1575,9 @@ class FunctionPullingContainerProxyTests
 
     machine ! Initialize(invocationNamespace.asString, fqn, action, schedulerHost, rpcPort, messageTransId)
     probe.expectMsg(Transition(machine, Uninitialized, CreatingClient))
+    awaitAssert {
+      machine.stateData shouldBe a[ContainerCreatedData]
+    }
     client.expectMsg(StartClient)
     client.send(machine, ClientCreationCompleted())