// Copyright 2017 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/ui/ash/wallpaper/wallpaper_controller_client_impl.h"
#include "ash/webui/personalization_app/mojom/personalization_app.mojom.h"
#include "base/containers/contains.h"
#include "base/strings/string_number_conversions.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "base/test/test_future.h"
#include "chrome/browser/ash/settings/scoped_cros_settings_test_helper.h"
#include "chrome/browser/ash/wallpaper_handlers/test_wallpaper_fetcher_delegate.h"
#include "chrome/browser/ui/ash/wallpaper/test_wallpaper_controller.h"
#include "chrome/test/base/scoped_testing_local_state.h"
#include "chrome/test/base/testing_browser_process.h"
#include "chromeos/ash/components/cryptohome/system_salt_getter.h"
#include "components/user_manager/fake_user_manager.h"
#include "components/user_manager/known_user.h"
#include "components/user_manager/scoped_user_manager.h"
#include "components/value_store/testing_value_store.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
class WallpaperControllerClientImplTest : public testing::Test {
public:
WallpaperControllerClientImplTest()
: local_state_(TestingBrowserProcess::GetGlobal()) {
auto fake_user_manager = std::make_unique<user_manager::FakeUserManager>();
user_manager_ = std::make_unique<user_manager::ScopedUserManager>(
std::move(fake_user_manager));
}
void SetUp() override {
client_.InitForTesting(&controller_);
ash::SystemSaltGetter::Initialize();
ash::SystemSaltGetter::Get()->SetRawSaltForTesting({0x01, 0x02, 0x03});
}
void TearDown() override { ash::SystemSaltGetter::Shutdown(); }
TestWallpaperController* controller() { return &controller_; }
WallpaperControllerClientImpl* client() { return &client_; }
WallpaperControllerClientImplTest(const WallpaperControllerClientImplTest&) =
delete;
WallpaperControllerClientImplTest& operator=(
const WallpaperControllerClientImplTest&) = delete;
~WallpaperControllerClientImplTest() override = default;
void OnGooglePhotosDailyAlbumFetched(
const AccountId& account_id,
WallpaperControllerClientImpl::FetchGooglePhotosPhotoCallback callback,
ash::personalization_app::mojom::FetchGooglePhotosPhotosResponsePtr
response) {
client_.OnGooglePhotosDailyAlbumFetched(account_id, std::move(callback),
std::move(response));
}
private:
ScopedTestingLocalState local_state_;
ash::ScopedCrosSettingsTestHelper cros_settings_test_helper_;
base::test::TaskEnvironment task_environment_;
std::unique_ptr<user_manager::ScopedUserManager> user_manager_;
TestWallpaperController controller_;
WallpaperControllerClientImpl client_{
std::make_unique<wallpaper_handlers::TestWallpaperFetcherDelegate>()};
};
TEST_F(WallpaperControllerClientImplTest, Construction) {
// Singleton was initialized.
EXPECT_EQ(client(), WallpaperControllerClientImpl::Get());
// Object was set as client.
EXPECT_TRUE(controller()->was_client_set());
}
TEST_F(WallpaperControllerClientImplTest, IsWallpaperSyncEnabledNoProfile) {
AccountId account_id =
AccountId::FromUserEmailGaiaId("[email protected]", "444444");
EXPECT_FALSE(client()->WallpaperControllerClientImpl::IsWallpaperSyncEnabled(
account_id));
}
TEST_F(WallpaperControllerClientImplTest, GetFilesId) {
const AccountId account_id = AccountId::FromUserEmail("[email protected]");
user_manager::KnownUser known_user(g_browser_process->local_state());
// Make a fake entry to register `account_id` as existing user.
known_user.SetPath(account_id, "test", base::Value(""));
EXPECT_TRUE(known_user.UserExists(account_id));
EXPECT_FALSE(known_user.FindPath(account_id, "wallpaper-files-id"));
base::test::TestFuture<const std::string&> future;
client()->GetFilesId(account_id, future.GetCallback());
ASSERT_TRUE(future.Wait());
const auto* value = known_user.FindPath(account_id, "wallpaper-files-id");
ASSERT_TRUE(value);
const auto* files_id = value->GetIfString();
ASSERT_TRUE(files_id);
EXPECT_EQ(*files_id, future.Get<0>());
}
TEST_F(WallpaperControllerClientImplTest, GetFilesIdForRemovedUser) {
const AccountId account_id = AccountId::FromUserEmail("[email protected]");
base::test::TestFuture<const std::string&> future;
client()->GetFilesId(account_id, future.GetCallback());
ASSERT_TRUE(future.Wait());
user_manager::KnownUser known_user(g_browser_process->local_state());
EXPECT_FALSE(known_user.UserExists(account_id));
}
TEST_F(WallpaperControllerClientImplTest, DailyGooglePhotosDoNotRepeat) {
using ash::personalization_app::mojom::GooglePhotosPhotoPtr;
AccountId account_id =
AccountId::FromUserEmailGaiaId("[email protected]", "444444");
// Size big enough to have not hit the minimum cache size logic, but small
// enough to make sure we're very unlikely to pass if the logic is broken,
// since there's randomness involved here.
const int photos_in_album = 20;
// Create a fake response with fake photos (the id is the only parameter that
// matters here).
auto response =
ash::personalization_app::mojom::FetchGooglePhotosPhotosResponse::New(
std::vector<GooglePhotosPhotoPtr>(), std::nullopt);
for (int i = 0; i < photos_in_album; i++) {
response->photos->push_back(
ash::personalization_app::mojom::GooglePhotosPhoto::New(
"id" + base::NumberToString(i), /*dedup_key=*/std::nullopt,
/*name=*/"", /*date=*/u"",
/*url=*/GURL(""), /*location=*/std::nullopt));
}
std::deque<std::string> last_ten;
auto handle_photo = [&last_ten](GooglePhotosPhotoPtr photo, bool success) {
ASSERT_TRUE(success);
EXPECT_FALSE(base::Contains(last_ten, photo->id));
last_ten.push_back(photo->id);
if (last_ten.size() > 10)
last_ten.pop_front();
};
for (int i = 0; i < 20; i++) {
OnGooglePhotosDailyAlbumFetched(account_id,
base::BindLambdaForTesting(handle_photo),
response->Clone());
}
}
} // namespace
Codebase Browser
chromium