3 files changed, 41 insertions(+), 1 deletions(-)

M module-sys/SystemManager/GovernorSentinelOperations.cpp
M module-sys/SystemManager/tests/unittest_CpuSentinelsGovernor.cpp
M pure_changelog.md

@@ 17,7 17,9 @@ namespace sys
             auto sentinelWeakPointer = (*sentinel)->GetSentinel();
             // remove expired ptrs
             if (sentinelWeakPointer.expired()) {
-                sentinels.erase(sentinel.base());
+                // shift iterator to point on the correct successor,
+                // without std::next it removes (sentinel - 1) element
+                sentinels.erase(std::next(sentinel).base());
                 continue;
             }
             // call foo on sentinel

@@ 119,6 119,43 @@ TEST_CASE("GovernorSentinelsVector - single sentinel add and remove")
     REQUIRE(container.empty());
 }
 
+TEST_CASE("GovernorSentinelsVector - a few removed in the middle")
+{
+    using namespace sys;
+    size_t count          = 0;
+    constexpr size_t size = 10;
+    GovernorSentinelsVector container;
+    std::vector<std::shared_ptr<CpuSentinel>> sentinels;
+    std::vector<std::string> sentinelName;
+
+    governed_callback foo = [&](const GovernorSentinel &s) { return false; };
+
+    for (size_t i = 0; i < size; ++i) {
+        sentinelName.push_back("test_" + std::to_string(i));
+        sentinels.push_back(std::make_shared<CpuSentinel>(sentinelName[i], nullptr));
+        container.push_back(std::make_unique<GovernorSentinel>(sentinels[i]));
+    }
+
+    // Remove sentinel in the middle
+    sentinels.erase(sentinels.begin() + 2);
+    sentinelName.erase(sentinelName.begin() + 2);
+
+    // Remove sentinel in the middle
+    sentinels.erase(sentinels.begin() + 6);
+    sentinelName.erase(sentinelName.begin() + 6);
+
+    for_each_governed_sentinel(container, foo);
+
+    auto countSentinels = sentinels.size();
+    for (size_t i = 0; i < countSentinels; ++i) {
+        auto cont = container[i]->GetSentinel();
+        if (not cont.expired()) {
+            count++;
+        }
+    }
+    REQUIRE(count == countSentinels);
+}
+
 TEST_CASE("GovernorSentinelsVector - multiple sentinels add and remove")
 {
     using namespace sys;

@@ 7,6 7,7 @@
 * Separated system volume from Bluetooth device volume for A2DP
 
 ### Fixed
+* Fixed removing wrong sentinels
 * Fixed music player behaviour when connecting/disconnecting audio devices
 * Fixed dropping the call during the DND mode
 * Fixed wrong tethering popup order