Resolve race condition before UserPrompt in simulated tests. (#25576)

project-chip · Oct 23, 2023 · 2468792 · 2468792
1 parent a3e1dc5
commit 2468792
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/examples/placeholder/linux/include/TestCommand.h b/examples/placeholder/linux/include/TestCommand.h
@@ -111,7 +111,6 @@ class TestCommand : public TestRunner,
         TestCommand * command = reinterpret_cast<TestCommand *>(context);
         command->isRunning    = true;
         command->NextTest();
-        chip::DeviceLayer::PlatformMgr().RemoveEventHandler(OnPlatformEvent, context);
     }
 
     static void OnPlatformEvent(const chip::DeviceLayer::ChipDeviceEvent * event, intptr_t arg)
@@ -121,6 +120,7 @@ class TestCommand : public TestRunner,
         case chip::DeviceLayer::DeviceEventType::kCommissioningComplete:
             ChipLogProgress(chipTool, "Commissioning complete");
             chip::DeviceLayer::PlatformMgr().ScheduleWork(ScheduleNextTest, arg);
+            chip::DeviceLayer::PlatformMgr().RemoveEventHandler(OnPlatformEvent, arg);
             break;
         }
     }
@@ -129,7 +129,7 @@ class TestCommand : public TestRunner,
     {
         if (commandPath == mCommandPath)
         {
-            NextTest();
+            chip::DeviceLayer::PlatformMgr().ScheduleWork(ScheduleNextTest, reinterpret_cast<intptr_t>(this));
             return;
         }
 
@@ -141,7 +141,7 @@ class TestCommand : public TestRunner,
     {
         if (attributePath == mAttributePath)
         {
-            NextTest();
+            chip::DeviceLayer::PlatformMgr().ScheduleWork(ScheduleNextTest, reinterpret_cast<intptr_t>(this));
             return;
         }