1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/policy/messaging_layer/util/user_dm_token_retriever.h"
#include <memory>
#include <string_view>
#include "base/functional/bind.h"
#include "base/memory/raw_ptr.h"
#include "base/test/task_environment.h"
#include "chrome/browser/policy/dm_token_utils.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chrome/test/base/testing_profile_manager.h"
#include "components/policy/core/common/cloud/dm_token.h"
#include "components/reporting/util/statusor.h"
#include "components/reporting/util/test_support_callbacks.h"
#include "content/public/test/browser_task_environment.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace reporting {
namespace {
using ::testing::MockFunction;
using ::testing::Return;
using ::testing::StrEq;
constexpr char kDMToken[] = "dm-token";
constexpr char kUserId[] = "test-user";
class UserDMTokenRetrieverTest : public ::testing::Test {
protected:
UserDMTokenRetrieverTest()
: testing_profile_manager_(TestingBrowserProcess::GetGlobal()) {
// Ensure test profile manager is set up
CHECK(testing_profile_manager_.SetUp())
<< "TestingProfileManager not setup";
profile_ = CreateProfile(kUserId);
}
void SetUp() override {
policy::SetDMTokenForTesting(policy::DMToken::CreateValidToken(kDMToken));
// Use the mock profile retrieval callback for setting up the retriever
user_dm_token_retriever_ = UserDMTokenRetriever::CreateForTest(
base::BindRepeating(&MockFunction<Profile*()>::Call,
base::Unretained(&profile_retrieval_cb_)));
}
void TearDown() override {
// Unset mocks and test hooks
policy::SetDMTokenForTesting(policy::DMToken::CreateEmptyToken());
}
// Creates a new profile with the current context for testing purposes.
TestingProfile* CreateProfile(std::string_view id) {
TestingProfile* const profile =
testing_profile_manager_.CreateTestingProfile(std::string(id));
return profile;
}
content::BrowserTaskEnvironment task_environment_{
base::test::TaskEnvironment::MainThreadType::UI};
TestingProfileManager testing_profile_manager_;
raw_ptr<TestingProfile> profile_;
MockFunction<Profile*()> profile_retrieval_cb_;
std::unique_ptr<UserDMTokenRetriever> user_dm_token_retriever_;
};
TEST_F(UserDMTokenRetrieverTest, GetDMToken) {
EXPECT_CALL(profile_retrieval_cb_, Call()).WillOnce(Return(profile_.get()));
test::TestEvent<StatusOr<std::string>> dm_token_retrieved_event;
std::move(user_dm_token_retriever_)
->RetrieveDMToken(dm_token_retrieved_event.cb());
const auto dm_token_result = dm_token_retrieved_event.result();
ASSERT_TRUE(dm_token_result.has_value());
EXPECT_THAT(dm_token_result.value(), StrEq(kDMToken));
}
TEST_F(UserDMTokenRetrieverTest, GetDMTokenMultipleRequests) {
EXPECT_CALL(profile_retrieval_cb_, Call())
.Times(2)
.WillRepeatedly(Return(profile_.get()));
test::TestEvent<StatusOr<std::string>> dm_token_retrieved_event_1;
user_dm_token_retriever_->RetrieveDMToken(dm_token_retrieved_event_1.cb());
test::TestEvent<StatusOr<std::string>> dm_token_retrieved_event_2;
std::move(user_dm_token_retriever_)
->RetrieveDMToken(dm_token_retrieved_event_2.cb());
const auto dm_token_result_1 = dm_token_retrieved_event_1.result();
const auto dm_token_result_2 = dm_token_retrieved_event_2.result();
ASSERT_TRUE(dm_token_result_1.has_value());
ASSERT_TRUE(dm_token_result_2.has_value());
EXPECT_THAT(dm_token_result_1.value(), StrEq(kDMToken));
EXPECT_THAT(dm_token_result_2.value(), StrEq(kDMToken));
}
TEST_F(UserDMTokenRetrieverTest, GetDMTokenInvalid) {
policy::SetDMTokenForTesting(policy::DMToken::CreateInvalidToken());
EXPECT_CALL(profile_retrieval_cb_, Call()).WillOnce(Return(profile_.get()));
test::TestEvent<StatusOr<std::string>> dm_token_retrieved_event;
std::move(user_dm_token_retriever_)
->RetrieveDMToken(dm_token_retrieved_event.cb());
EXPECT_FALSE(dm_token_retrieved_event.result().has_value());
}
} // namespace
} // namespace reporting
|