From eeb994a5849f9c48e48f78a7f5b68b0ac047f881 Mon Sep 17 00:00:00 2001
From: Alexey Rybalchenko <alexryba@gmail.com>
Date: Wed, 23 Aug 2017 11:12:29 +0200
Subject: [PATCH] Tests for MQ examples

---
 fairmq/FairMQDevice.cxx     | 4 ++--
 fairmq/FairMQStateMachine.h | 2 ++
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/fairmq/FairMQDevice.cxx b/fairmq/FairMQDevice.cxx
index e8e9074a..08f61ade 100644
--- a/fairmq/FairMQDevice.cxx
+++ b/fairmq/FairMQDevice.cxx
@@ -559,8 +559,6 @@ void FairMQDevice::RunWrapper()
 
             Run();
         }
-
-        PostRun();
     }
     catch (const out_of_range& oor)
     {
@@ -574,6 +572,8 @@ void FairMQDevice::RunWrapper()
         ChangeState(internal_READY);
     }
 
+    PostRun();
+
     rateLogger.join();
 }
 
diff --git a/fairmq/FairMQStateMachine.h b/fairmq/FairMQStateMachine.h
index 2d5e07ca..22f5eb31 100644
--- a/fairmq/FairMQStateMachine.h
+++ b/fairmq/FairMQStateMachine.h
@@ -291,6 +291,8 @@ struct FairMQFSM_ : public msmf::state_machine_def<FairMQFSM_>
             {
                 fsm.fStateChangeCallback(READY);
             }
+
+            fsm.Unblock();
         }
     };