1 /*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include <gtest/gtest.h>
17
18 #include "account_log_wrapper.h"
19 #include "app_account_common.h"
20 #define private public
21 #include "app_account_authenticator_callback.h"
22 #include "app_account_authenticator_callback_stub.h"
23 #include "app_account_authenticator_callback_proxy.h"
24 #include "iapp_account_authenticator_callback.h"
25 #undef private
26
27 using namespace testing::ext;
28 using namespace OHOS;
29 using namespace OHOS::AccountSA;
30
31 bool g_status = false;
32
33 class MockAppAccountAuthenticatorCallbackStub : public AppAccountAuthenticatorCallbackStub {
34 public:
OnResult(int32_t resultCode,const AAFwk::Want & result)35 void OnResult(int32_t resultCode, const AAFwk::Want &result)
36 {
37 g_status = true;
38 }
OnRequestRedirected(AAFwk::Want & request)39 virtual void OnRequestRedirected(AAFwk::Want &request)
40 {
41 g_status = true;
42 }
OnRequestContinued()43 virtual void OnRequestContinued()
44 {
45 g_status = true;
46 }
47 };
48
49 class AppAccountAuthenticatorCallbackModuleTest : public testing::Test {
50 public:
51 static void SetUpTestCase(void);
52 static void TearDownTestCase(void);
53 void SetUp(void) override;
54 void TearDown(void) override;
55 };
56
SetUpTestCase(void)57 void AppAccountAuthenticatorCallbackModuleTest::SetUpTestCase(void)
58 {}
59
TearDownTestCase(void)60 void AppAccountAuthenticatorCallbackModuleTest::TearDownTestCase(void)
61 {}
62
SetUp(void)63 void AppAccountAuthenticatorCallbackModuleTest::SetUp(void) __attribute__((no_sanitize("cfi")))
64 {
65 testing::UnitTest *test = testing::UnitTest::GetInstance();
66 ASSERT_NE(test, nullptr);
67 const testing::TestInfo *testinfo = test->current_test_info();
68 ASSERT_NE(testinfo, nullptr);
69 string testCaseName = string(testinfo->name());
70 ACCOUNT_LOGI("[SetUp] %{public}s start", testCaseName.c_str());
71 }
72
TearDown(void)73 void AppAccountAuthenticatorCallbackModuleTest::TearDown(void)
74 {}
75
76 /**
77 * @tc.name: AppAccountAuthenticatorCallbackTest_OnResult_0100
78 * @tc.desc: test authenticatecallback func OnResult.
79 * @tc.type: FUNC
80 * @tc.require:
81 */
82 HWTEST_F(AppAccountAuthenticatorCallbackModuleTest, AppAccountAuthenticatorCallbackTest_OnResult_0100, TestSize.Level1)
83 {
84 ACCOUNT_LOGI("AppAccountAuthenticatorCallbackTest_OnResult_0100");
85
86 AAFwk::Want result;
87 int32_t testResultCode = 1;
88 sptr<IAppAccountAuthenticatorCallback> oauthCallbackPtr =
89 new (std::nothrow) MockAppAccountAuthenticatorCallbackStub();
90 ASSERT_NE(oauthCallbackPtr, nullptr);
91 sptr<IRemoteObject> callback = oauthCallbackPtr->AsObject();
92 AppAccountAuthenticatorCallbackProxy testCallbackProxy(callback);
93 testCallbackProxy.OnResult(testResultCode, result);
94 EXPECT_EQ(g_status, true);
95 g_status = false;
96
97 AAFwk::Want request;
98 testCallbackProxy.OnRequestRedirected(request);
99 EXPECT_EQ(g_status, true);
100 g_status = false;
101
102 testCallbackProxy.OnRequestContinued();
103 EXPECT_EQ(g_status, true);
104 g_status = false;
105 }
106
107 /**
108 * @tc.name: AppAccountAuthenticatorCallbackTest_func_0200
109 * @tc.desc: test authenticatecallback func OnRequestRedirected and OnRequestContinued.
110 * @tc.type: FUNC
111 * @tc.require:
112 */
113 HWTEST_F(AppAccountAuthenticatorCallbackModuleTest, AppAccountAuthenticatorCallbackTest_func_0200, TestSize.Level1)
114 {
115 const std::string sessionId = "testsessionid";
116 auto AppAccountAuthenticatorCallbackPtr = std::make_shared<AppAccountAuthenticatorCallback>(sessionId);
117 ASSERT_NE(nullptr, AppAccountAuthenticatorCallbackPtr);
118 AAFwk::Want reques;
119 AppAccountAuthenticatorCallbackPtr->OnRequestRedirected(reques);
120 ASSERT_EQ(AppAccountAuthenticatorCallbackPtr->sessionId_, sessionId);
121 AppAccountAuthenticatorCallbackPtr->OnRequestContinued();
122 ASSERT_EQ(AppAccountAuthenticatorCallbackPtr->sessionId_, sessionId);
123 }