cm: updatemanager: implement sending full unit status on cloud connect

Signed-off-by: Oleksandr Grytsov <[email protected]>

Author: al1img

src/core/cm/CMakeLists.txt

@@ -27,10 +27,11 @@ set(INSTALL_HEADERS
     aos_core_cm_smcontroller
     aos_core_cm_storagestate
     aos_core_cm_unitconfig
-    aos_core_cm_updatemanager
 )
 
-set(INSTALL_LIBRARIES aos_core_cm_alerts aos_core_cm_imagemanager aos_core_cm_monitoring aos_core_cm_storagestate)
+set(INSTALL_LIBRARIES aos_core_cm_alerts aos_core_cm_imagemanager aos_core_cm_monitoring aos_core_cm_storagestate
+                      aos_core_cm_updatemanager
+)
 
 if(WITH_TEST)
     list(APPEND INSTALL_HEADERS aos_core_cm_tests_mocks)

src/core/cm/updatemanager/CMakeLists.txt

@@ -14,13 +14,31 @@ set(TARGET_NAME updatemanager)
 # Headers
 # ######################################################################################################################
 
-set(HEADERS itf/sender.hpp itf/updatemanager.hpp)
+set(HEADERS itf/sender.hpp itf/updatemanager.hpp updatemanager.hpp unitstatushandler.hpp)
+
+# ######################################################################################################################
+# Sources
+# ######################################################################################################################
+
+set(SOURCES unitstatushandler.cpp updatemanager.cpp)
 
 # ######################################################################################################################
 # Target
 # ######################################################################################################################
 
-add_module(TARGET_NAME ${TARGET_NAME} HEADERS ${HEADERS})
+add_module(
+    TARGET_NAME
+    ${TARGET_NAME}
+    LOG_MODULE
+    STACK_USAGE
+    ${AOS_STACK_USAGE}
+    SOURCES
+    ${SOURCES}
+    HEADERS
+    ${HEADERS}
+    LIBRARIES
+    ${LIBRARIES}
+)
 
 # ######################################################################################################################
 # Tests

src/core/cm/updatemanager/tests/CMakeLists.txt

@@ -16,7 +16,7 @@ set(SOURCES updatemanager.cpp)
 # Libraries
 # ######################################################################################################################
 
-set(LIBRARIES GTest::gmock_main)
+set(LIBRARIES aos::core::cm::updatemanager aos::core::common::tests::utils GTest::gmock_main)
 
 # ######################################################################################################################
 # Target

src/core/cm/updatemanager/tests/mocks/imagemanagermock.hpp

@@ -0,0 +1,36 @@
+/*
+ * Copyright (C) 2025 EPAM Systems, Inc.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#ifndef AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_IMAGEMANAGERMOCK_HPP_
+#define AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_IMAGEMANAGERMOCK_HPP_
+
+#include <gmock/gmock.h>
+
+#include <core/cm/imagemanager/itf/imagemanager.hpp>
+
+namespace aos::cm::imagemanager {
+
+/**
+ * Image manager mock.
+ */
+class ImageManagerMock : public ImageManagerItf {
+public:
+    MOCK_METHOD(Error, InstallUpdateItems,
+        (const Array<UpdateItemInfo>& itemsInfo, const Array<crypto::CertificateInfo>& certificates,
+            const Array<crypto::CertificateChainInfo>& certificateChains, Array<UpdateItemStatus>& statuses),
+        (override));
+    MOCK_METHOD(Error, UninstallUpdateItems,
+        (const Array<StaticString<cIDLen>>& ids, Array<UpdateItemStatus>& statuses), (override));
+    MOCK_METHOD(Error, RevertUpdateItems, (const Array<StaticString<cIDLen>>& ids, Array<UpdateItemStatus>& statuses),
+        (override));
+    MOCK_METHOD(Error, GetUpdateItemsStatuses, (Array<UpdateItemStatus> & statuses), (override));
+    MOCK_METHOD(Error, SubscribeListener, (ImageStatusListenerItf & listener), (override));
+    MOCK_METHOD(Error, UnsubscribeListener, (ImageStatusListenerItf & listener), (override));
+};
+
+} // namespace aos::cm::imagemanager
+
+#endif

src/core/cm/updatemanager/tests/mocks/instancestatusprovidermock.hpp

@@ -0,0 +1,28 @@
+/*
+ * Copyright (C) 2025 EPAM Systems, Inc.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#ifndef AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_INSTANCESTATUSPROVIDERMOCK_HPP_
+#define AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_INSTANCESTATUSPROVIDERMOCK_HPP_
+
+#include <gmock/gmock.h>
+
+#include <core/cm/launcher/itf/instancestatusprovider.hpp>
+
+namespace aos::cm::launcher {
+
+/**
+ * Instance status provider mock.
+ */
+class InstanceStatusProviderMock : public InstanceStatusProviderItf {
+public:
+    MOCK_METHOD(Error, GetInstancesStatuses, (Array<InstanceStatus> & statuses), (override));
+    MOCK_METHOD(Error, SubscribeListener, (InstanceStatusListenerItf & listener), (override));
+    MOCK_METHOD(Error, UnsubscribeListener, (InstanceStatusListenerItf & listener), (override));
+};
+
+} // namespace aos::cm::launcher
+
+#endif

src/core/cm/updatemanager/tests/mocks/unitconfigmock.hpp

@@ -0,0 +1,28 @@
+/*
+ * Copyright (C) 2025 EPAM Systems, Inc.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#ifndef AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_UNITCONFIGMOCK_HPP_
+#define AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_UNITCONFIGMOCK_HPP_
+
+#include <gmock/gmock.h>
+
+#include <core/cm/unitconfig/itf/unitconfig.hpp>
+
+namespace aos::cm::unitconfig {
+
+/**
+ * Unit config mock.
+ */
+class UnitConfigMock : public UnitConfigItf {
+public:
+    MOCK_METHOD(Error, CheckUnitConfig, (const UnitConfig&), (override));
+    MOCK_METHOD(Error, UpdateUnitConfig, (const UnitConfig&), (override));
+    MOCK_METHOD(Error, GetUnitConfigStatus, (UnitConfigStatus&), (override));
+};
+
+} // namespace aos::cm::unitconfig
+
+#endif

src/core/cm/updatemanager/tests/stubs/senderstub.hpp

@@ -0,0 +1,63 @@
+/*
+ * Copyright (C) 2025 EPAM Systems, Inc.
+ *
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+#ifndef AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_SENDER_HPP_
+#define AOS_CORE_CM_UPDATEMANAGER_TESTS_MOCKS_SENDER_HPP_
+
+#include <condition_variable>
+#include <mutex>
+
+#include <core/cm/updatemanager/itf/sender.hpp>
+
+namespace aos::cm::updatemanager {
+
+/**
+ * Sender stub.
+ */
+class SenderStub : public SenderItf {
+public:
+    /**
+     * Sends unit status.
+     *
+     * @param unitStatus unit status.
+     * @return Error.
+     */
+    Error SendUnitStatus(const UnitStatus& unitStatus) override
+    {
+        std::lock_guard lock {mMutex};
+
+        mUnitStatus = unitStatus;
+        mSendCalled = true;
+        mCondVar.notify_one();
+
+        return ErrorEnum::eNone;
+    }
+
+    const UnitStatus& WaitSendUnitStatus()
+    {
+        std::unique_lock lock {mMutex};
+
+        if (!mCondVar.wait_for(lock, cWaitTimeout, [this] { return mSendCalled; })) {
+            throw std::runtime_error("waiting for unit status timed out");
+        }
+
+        mSendCalled = false;
+
+        return mUnitStatus;
+    }
+
+private:
+    static constexpr std::chrono::milliseconds cWaitTimeout {5000};
+
+    UnitStatus              mUnitStatus;
+    bool                    mSendCalled {};
+    std::mutex              mMutex;
+    std::condition_variable mCondVar;
+};
+
+} // namespace aos::cm::updatemanager
+
+#endif