[iam] Implement CertReceiver subscription in ProvisionManager

Signed-off-by: Mykola Kobets <[email protected]>
aosedge · Oct 3, 2024 · 587e954 · 587e954
1 parent 21cb940
commit 587e954
Show file tree

Hide file tree

Showing 3 changed files with 62 additions and 0 deletions.
diff --git a/include/aos/iam/provisionmanager.hpp b/include/aos/iam/provisionmanager.hpp
@@ -119,6 +119,23 @@ class ProvisionManagerItf {
         certhandler::CertInfo& resCert)
         = 0;
 
+    /**
+     * Subscribes certificates receiver.
+     *
+     * @param certType certificate type.
+     * @param certReceiver certificate receiver.
+     * @returns Error.
+     */
+    virtual Error SubscribeCertReceiver(const String& certType, certhandler::CertReceiverItf& certReceiver) = 0;
+
+    /**
+     * Unsubscribes certificate receiver.
+     *
+     * @param certReceiver certificate receiver.
+     * @returns Error.
+     */
+    virtual Error UnsubscribeCertReceiver(certhandler::CertReceiverItf& certReceiver) = 0;
+
     /**
      * Finishes provisioning.
      *
@@ -204,6 +221,23 @@ class ProvisionManager : public ProvisionManagerItf {
     Error GetCert(const String& certType, const Array<uint8_t>& issuer, const Array<uint8_t>& serial,
         certhandler::CertInfo& resCert) override;
 
+    /**
+     * Subscribes certificates receiver.
+     *
+     * @param certType certificate type.
+     * @param certReceiver certificate receiver.
+     * @returns Error.
+     */
+    Error SubscribeCertReceiver(const String& certType, certhandler::CertReceiverItf& certReceiver) override;
+
+    /**
+     * Unsubscribes certificate receiver.
+     *
+     * @param certReceiver certificate receiver.
+     * @returns Error.
+     */
+    Error UnsubscribeCertReceiver(certhandler::CertReceiverItf& certReceiver) override;
+
     /**
      * Finishes provisioning.
      *

diff --git a/src/iam/provisionmanager/provisionmanager.cpp b/src/iam/provisionmanager/provisionmanager.cpp
@@ -145,4 +145,18 @@ Error ProvisionManager::GetCert(
     return AOS_ERROR_WRAP(mCertHandler->GetCertificate(certType, issuer, serial, resCert));
 }
 
+Error ProvisionManager::SubscribeCertReceiver(const String& certType, certhandler::CertReceiverItf& certReceiver)
+{
+    LOG_DBG() << "Subscribe cert receiver: type = " << certType;
+
+    return AOS_ERROR_WRAP(mCertHandler->SubscribeCertReceiver(certType, certReceiver));
+}
+
+Error ProvisionManager::UnsubscribeCertReceiver(certhandler::CertReceiverItf& certReceiver)
+{
+    LOG_DBG() << "Unsubscribe cert receiver";
+
+    return AOS_ERROR_WRAP(mCertHandler->UnsubscribeCertReceiver(certReceiver));
+}
+
 } // namespace aos::iam::provisionmanager
diff --git a/tests/iam/provisionmanager/provisionmanager_test.cpp b/tests/iam/provisionmanager/provisionmanager_test.cpp
@@ -10,6 +10,7 @@
 #include <aos/iam/provisionmanager.hpp>
 
 #include "mocks/certhandlermock.hpp"
+#include "mocks/certreceivermock.hpp"
 #include "mocks/provisioningcallbackmock.hpp"
 
 using namespace testing;
@@ -293,6 +294,19 @@ TEST_F(ProvisionManagerTest, GetCert)
     EXPECT_EQ(result, certInfo);
 }
 
+TEST_F(ProvisionManagerTest, SubscribeCertReceiver)
+{
+    const aos::String certType = "iam";
+
+    CertReceiverItfMock certReceiver;
+
+    EXPECT_CALL(mCertHandler, SubscribeCertReceiver(certType, _)).WillOnce(Return(aos::ErrorEnum::eNone));
+    ASSERT_TRUE(mProvisionManager.SubscribeCertReceiver(certType, certReceiver).IsNone());
+
+    EXPECT_CALL(mCertHandler, UnsubscribeCertReceiver(Ref(certReceiver))).WillOnce(Return(aos::ErrorEnum::eNone));
+    ASSERT_TRUE(mProvisionManager.UnsubscribeCertReceiver(certReceiver).IsNone());
+}
+
 TEST_F(ProvisionManagerTest, Deprovision)
 {
     EXPECT_CALL(mCallback, OnDeprovision).WillOnce(Return(aos::ErrorEnum::eNone));