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 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147
|
// Copyright 2024 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "components/user_manager/account_id_util.h"
#include <memory>
#include <optional>
#include <utility>
#include "base/values.h"
#include "components/account_id/account_id.h"
#include "google_apis/gaia/gaia_id.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace user_manager {
namespace {
constexpr char kUserEmail[] = "default_account@gmail.com";
constexpr char kOtherEmail[] = "renamed_account@gmail.com";
constexpr GaiaId::Literal kGaiaID("fake-gaia-id");
} // namespace
// Base class for tests of known_user.
// Sets up global objects necessary for known_user to be able to access
// local_state.
class AccountIdUtilTest : public testing::Test {
public:
AccountIdUtilTest() = default;
~AccountIdUtilTest() override = default;
AccountIdUtilTest(const AccountIdUtilTest& other) = delete;
AccountIdUtilTest& operator=(const AccountIdUtilTest& other) = delete;
protected:
const AccountId kDefaultAccountId =
AccountId::FromUserEmailGaiaId(kUserEmail, kGaiaID);
};
TEST_F(AccountIdUtilTest, LoadGoogleAccountWithGaiaId) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
std::optional<AccountId> result = LoadAccountId(dict);
ASSERT_TRUE(result);
ASSERT_TRUE(result->is_valid());
EXPECT_EQ(result->GetAccountType(), AccountType::GOOGLE);
EXPECT_EQ(result->GetUserEmail(), kUserEmail);
ASSERT_TRUE(result->HasAccountIdKey());
ASSERT_EQ(result->GetGaiaId(), kGaiaID);
}
TEST_F(AccountIdUtilTest, LoadGoogleAccountWithoutGaiaId) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail);
std::optional<AccountId> result = LoadAccountId(dict);
ASSERT_TRUE(result);
ASSERT_TRUE(result->is_valid());
EXPECT_EQ(result->GetAccountType(), AccountType::GOOGLE);
EXPECT_EQ(result->GetUserEmail(), kUserEmail);
ASSERT_FALSE(result->HasAccountIdKey());
}
// Death test for some reason fails on MSAN bots.
// TODO(b/325904498): Investigate how DEATH tests should be used.
TEST_F(AccountIdUtilTest, DISABLED_LoadUnknownAccount) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "unknown")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
ASSERT_DEATH({ LoadAccountId(dict); }, "Unknown account type");
}
TEST_F(AccountIdUtilTest, LoadAccountEmailOnly) {
base::Value::Dict dict = base::Value::Dict().Set("email", kUserEmail);
std::optional<AccountId> result = LoadAccountId(dict);
ASSERT_TRUE(result);
ASSERT_TRUE(result->is_valid());
// Assume that accounts are Google accounts by default.
EXPECT_EQ(result->GetAccountType(), AccountType::GOOGLE);
EXPECT_EQ(result->GetUserEmail(), kUserEmail);
ASSERT_FALSE(result->HasAccountIdKey());
}
TEST_F(AccountIdUtilTest, MatchByCorrectEmail) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
AccountId id = AccountId::FromUserEmail(kUserEmail);
ASSERT_TRUE(AccountIdMatches(id, dict));
}
TEST_F(AccountIdUtilTest, MatchByIncorrectEmail) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
AccountId id = AccountId::FromUserEmail(kOtherEmail);
ASSERT_FALSE(AccountIdMatches(id, dict));
}
TEST_F(AccountIdUtilTest, MatchByGaiaIdSameEmail) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
AccountId id = AccountId::FromUserEmailGaiaId(kUserEmail, kGaiaID);
ASSERT_TRUE(AccountIdMatches(id, dict));
}
TEST_F(AccountIdUtilTest, MatchByGaiaIdOtherEmail) {
base::Value::Dict dict = base::Value::Dict()
.Set("account_type", "google")
.Set("email", kUserEmail)
.Set("gaia_id", kGaiaID.ToString());
AccountId id = AccountId::FromUserEmailGaiaId(kOtherEmail, kGaiaID);
ASSERT_TRUE(AccountIdMatches(id, dict));
}
TEST_F(AccountIdUtilTest, MatchByEmailOnly) {
base::Value::Dict dict = base::Value::Dict().Set("email", kUserEmail);
AccountId id = AccountId::FromUserEmail(kUserEmail);
ASSERT_TRUE(AccountIdMatches(id, dict));
}
TEST_F(AccountIdUtilTest, StoreEmailOnly) {
AccountId id = AccountId::FromUserEmail(kUserEmail);
base::Value::Dict dict;
StoreAccountId(id, dict);
EXPECT_EQ(dict.Find("account_type"), nullptr);
EXPECT_EQ(dict.Find("email")->GetString(), kUserEmail);
EXPECT_EQ(dict.Find("gaia_id"), nullptr);
}
TEST_F(AccountIdUtilTest, StoreGoogleAccount) {
AccountId id = AccountId::FromUserEmailGaiaId(kUserEmail, kGaiaID);
base::Value::Dict dict;
StoreAccountId(id, dict);
EXPECT_EQ(dict.Find("account_type")->GetString(), "google");
EXPECT_EQ(dict.Find("email")->GetString(), kUserEmail);
EXPECT_EQ(dict.Find("gaia_id")->GetString(), kGaiaID.ToString());
}
} // namespace user_manager
|