diff --git a/test/unittest/mock/if_system_ability_manager_mock.h b/test/unittest/mock/if_system_ability_manager_mock.h
index 8d4c0d3a17f7971acf396fa272a4859448b9268e..2c40e25ef75de83c9355c122d7f80b388afd3638 100644
--- a/test/unittest/mock/if_system_ability_manager_mock.h
+++ b/test/unittest/mock/if_system_ability_manager_mock.h
@@ -64,6 +64,8 @@ public:
         std::vector<SaExtensionInfo>& infoList));
     MOCK_METHOD3(GetCommonEventExtraDataIdlist, int32_t(int32_t systemAbilityId,
         std::vector<int64_t>& extraDataIdList, const std::string& eventName));
+    MOCK_METHOD2(GetSaAttributeInfo, int32_t(int32_t saId,
+        SaAttributeInfo& saAttributeInfo));
 };
 } // namespace OHOS