ensure proper init and shutdown during tests as these contexts and registries are now sensitive

Author: andreilitvin

src/app/server-cluster/testing/TestServerClusterContext.h

@@ -16,6 +16,7 @@
  */
 #pragma once
 
+#include "app/data-model-provider/Context.h"
 #include <app/data-model-provider/ActionContext.h>
 #include <app/data-model-provider/Provider.h>
 #include <app/server-cluster/ServerClusterContext.h>
@@ -74,6 +75,8 @@ class TestServerClusterContext
     LogOnlyEvents & EventsGenerator() { return mTestEventsGenerator; }
     TestProviderChangeListener & ChangeListener() { return mTestDataModelChangeListener; }
     TestPersistentStorageDelegate & StorageDelegate() { return mTestStorage; }
+    app::DataModel::InteractionModelContext &ImContext() { return mTestContext; }
+
 
 private:
     NullActionContext mNullActionContext;

src/data-model-providers/codegen/tests/TestCodegenModelViaMocks.cpp

@@ -19,6 +19,7 @@
 #include <data-model-providers/codegen/tests/EmberInvokeOverride.h>
 #include <data-model-providers/codegen/tests/EmberReadWriteOverride.h>
 
+#include <app/server-cluster/testing/TestServerClusterContext.h>
 #include <access/AccessControl.h>
 #include <access/SubjectDescriptor.h>
 #include <app-common/zap-generated/attribute-type.h>
@@ -2599,9 +2600,15 @@ static CHIP_ERROR ReadU32Attribute(DataModel::Provider & provider, const Concret
 
 TEST_F(TestCodegenModelViaMocks, ServerClusterInterfacesRegistration)
 {
+    TestServerClusterContext testContext;
+
     UseMockNodeConfig config(gTestNodeConfig);
     CodegenDataModelProviderWithContext model;
 
+
+    model.SetPersistentStorageDelegate(&testContext.StorageDelegate());
+    ASSERT_EQ(model.Startup(testContext.ImContext()), CHIP_NO_ERROR);
+
     const ConcreteClusterPath kTestClusterPath(kMockEndpoint1, MockClusterId(2));
 
     FakeDefaultServerCluster fakeClusterServer(kTestClusterPath);
@@ -2670,13 +2677,21 @@ TEST_F(TestCodegenModelViaMocks, ServerClusterInterfacesRegistration)
         std::optional<ActionReturnStatus> result = model.WriteAttribute(test.GetRequest(), decoder);
         ASSERT_TRUE(result.has_value() && result->GetUnderlyingError() == CHIP_ERROR_INCORRECT_STATE);
     }
+
+    model.Registry().Unregister(kTestClusterPath);
+    model.Shutdown();
 }
 
 TEST_F(TestCodegenModelViaMocks, ServerClusterInterfacesListClusters)
 {
+    TestServerClusterContext testContext;
+
     UseMockNodeConfig config(gTestNodeConfig);
     CodegenDataModelProviderWithContext model;
 
+    model.SetPersistentStorageDelegate(&testContext.StorageDelegate());
+    ASSERT_EQ(model.Startup(testContext.ImContext()), CHIP_NO_ERROR);
+
     // will register a fake cluster server which overrides the cluster data version
     // once registered
     const ConcreteClusterPath kTestClusterPath(kMockEndpoint1, MockClusterId(2));
@@ -2743,4 +2758,7 @@ TEST_F(TestCodegenModelViaMocks, ServerClusterInterfacesListClusters)
         }
     }
     EXPECT_TRUE(updatedClusterFound);
+
+    model.Registry().Unregister(kTestClusterPath);
+    model.Shutdown();
 }