[nrf toup] Test Fail-Safe Marker

Test if:
* Marker is stored during AddNOC.
* Marker is not stored during UpdateNOC.
* Marker is cleared During DisarmFailSafe and after timer expiry.

Signed-off-by: Adrian Gielniewski <adrian.gielniewski@nordicsemi.no>

Author: adigie

src/app/tests/TestFailSafeContext.cpp

@@ -34,6 +34,7 @@
 #include <lib/core/StringBuilderAdapters.h>
 #include <lib/support/CHIPMem.h>
 #include <lib/support/CodeUtils.h>
+#include <lib/support/DefaultStorageKeyAllocator.h>
 #include <lib/support/TestPersistentStorageDelegate.h>
 #include <platform/CHIPDeviceLayer.h>
 
@@ -77,12 +78,14 @@ TEST_F(TestFailSafeContext, TestFailSafeContext_ArmFailSafe)
     EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex1);
     EXPECT_TRUE(failSafeContext.IsFailSafeArmed(kTestAccessingFabricIndex1));
     EXPECT_FALSE(failSafeContext.IsFailSafeArmed(kTestAccessingFabricIndex2));
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
 
     failSafeContext.DisarmFailSafe();
     EXPECT_FALSE(failSafeContext.IsFailSafeArmed());
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
 }
 
-TEST_F(TestFailSafeContext, TestFailSafeContext_NocCommandInvoked)
+TEST_F(TestFailSafeContext, TestFailSafeContext_AddNocCommandInvoked)
 {
     chip::TestPersistentStorageDelegate testStorage;
     chip::app::FailSafeContext failSafeContext;
@@ -91,17 +94,67 @@ TEST_F(TestFailSafeContext, TestFailSafeContext_NocCommandInvoked)
 
     EXPECT_EQ(failSafeContext.ArmFailSafe(kTestAccessingFabricIndex1, System::Clock::Seconds16(1)), CHIP_NO_ERROR);
     EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex1);
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+
+    failSafeContext.SetAddNocCommandStarted(kTestAccessingFabricIndex2);
+    EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex2);
+    EXPECT_TRUE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
 
     failSafeContext.SetAddNocCommandInvoked(kTestAccessingFabricIndex2);
     EXPECT_TRUE(failSafeContext.NocCommandHasBeenInvoked());
     EXPECT_TRUE(failSafeContext.AddNocCommandHasBeenInvoked());
     EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex2);
 
+    failSafeContext.DisarmFailSafe();
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+}
+
+TEST_F(TestFailSafeContext, TestFailSafeContext_UpdateNocCommandInvoked)
+{
+    chip::TestPersistentStorageDelegate testStorage;
+    chip::app::FailSafeContext failSafeContext;
+
+    failSafeContext.Init({ &testStorage });
+
+    EXPECT_EQ(failSafeContext.ArmFailSafe(kTestAccessingFabricIndex1, System::Clock::Seconds16(1)), CHIP_NO_ERROR);
+    EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex1);
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+
     failSafeContext.SetUpdateNocCommandInvoked();
     EXPECT_TRUE(failSafeContext.NocCommandHasBeenInvoked());
     EXPECT_TRUE(failSafeContext.UpdateNocCommandHasBeenInvoked());
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
 
     failSafeContext.DisarmFailSafe();
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+}
+
+TEST_F(TestFailSafeContext, TestFailSafeContext_NocCommandInvokedTimerExpiry)
+{
+    chip::TestPersistentStorageDelegate testStorage;
+    chip::app::FailSafeContext failSafeContext;
+
+    failSafeContext.Init({ &testStorage });
+
+    EXPECT_EQ(failSafeContext.ArmFailSafe(kTestAccessingFabricIndex1, System::Clock::Seconds16(1)), CHIP_NO_ERROR);
+    EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex1);
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+
+    failSafeContext.SetAddNocCommandStarted(kTestAccessingFabricIndex2);
+    EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex2);
+    EXPECT_TRUE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
+
+    failSafeContext.SetAddNocCommandInvoked(kTestAccessingFabricIndex2);
+    EXPECT_TRUE(failSafeContext.NocCommandHasBeenInvoked());
+    EXPECT_TRUE(failSafeContext.AddNocCommandHasBeenInvoked());
+    EXPECT_EQ(failSafeContext.GetFabricIndex(), kTestAccessingFabricIndex2);
+
+    failSafeContext.ForceFailSafeTimerExpiry();
+
+    PlatformMgr().ScheduleWork([](intptr_t) -> void { PlatformMgr().StopEventLoopTask(); }, (intptr_t) nullptr);
+    PlatformMgr().RunEventLoop();
+
+    EXPECT_FALSE(testStorage.SyncDoesKeyExist(DefaultStorageKeyAllocator::FailSafeMarkerKey().KeyName()));
 }
 
 } // namespace