// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "ash/constants/ash_features.h"
#include "ash/public/cpp/system/toast_data.h"
#include "ash/public/cpp/system/toast_manager.h"
#include "ash/wm/desks/desks_controller.h"
#include "ash/wm/desks/desks_test_api.h"
#include "ash/wm/desks/desks_test_util.h"
#include "base/memory/scoped_refptr.h"
#include "base/test/metrics/histogram_tester.h"
#include "base/test/spin_wait.h"
#include "base/time/time.h"
#include "base/uuid.h"
#include "build/build_config.h"
#include "chrome/browser/chromeos/extensions/wm/wm_desks_private_api.h"
#include "chrome/browser/extensions/extension_apitest.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "content/public/test/browser_test.h"
#include "extensions/browser/api_test_utils.h"
namespace extensions {
class WmDesksPrivateApiTest : public ExtensionApiTest {
public:
WmDesksPrivateApiTest() {
scoped_feature_list.InitWithFeatures(
/*enabled_features=*/{ash::features::kDesksTemplates},
/*disabled_features=*/{ash::features::kDeskTemplateSync});
}
~WmDesksPrivateApiTest() override = default;
private:
base::test::ScopedFeatureList scoped_feature_list;
};
// Use API test for tests require other chrome API.
// API test is flaky when involves animation. For APIs involving animation use
// browser test instead.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, WmDesksPrivateApiTest) {
// This loads and runs an extension from
// chrome/test/data/extensions/api_test/wm_desks_private.
ASSERT_TRUE(RunExtensionTest("wm_desks_private")) << message_;
}
// Tests launch and close a desk.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, LaunchAndCloseDeskTest) {
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
// The check is necessary as both desk animation and extension function is
// async. There is no guarantee which ones execute first.
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter remove_waiter;
// Remove a desk.
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
remove_desk_function.get(),
R"([")" + desk_id->GetString() +
R"(", { "combineDesks": false, "allowUndo":false }])",
browser()->profile());
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 1, 1);
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
remove_waiter.Wait();
}
histogram_tester.ExpectUniqueSample("Ash.DeskApi.RemoveDeskType",
ash::DeskCloseType::kCloseAllWindows, 1);
}
// Tests launch and removal of a desk. Makes sure desk cannot be undone after
// time has passed.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, LaunchAndAttemptUndo) {
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
// The check is necessary as both desk animation and extension function is
// async. There is no guarantee which ones execute first.
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter remove_waiter;
// Remove a desk.
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
remove_desk_function.get(),
R"([")" + desk_id->GetString() +
R"(", { "combineDesks": false, "allowUndo": true }])",
browser()->profile());
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
remove_waiter.Wait();
}
EXPECT_TRUE(ash::DesksTestApi::DesksControllerCanUndoDeskRemoval());
// Checks for if there are any other toasts running besides
// the undo toast. Waits for other toasts to expire.
if (!ash::ToastManager::Get()->IsToastShown("UndoCloseAllToast_1")) {
LOG(INFO) << "Non-undo toast running, must wait for other toasts :(";
SPIN_FOR_TIMEDELTA_OR_UNTIL_TRUE(
base::Seconds(45),
ash::ToastManager::Get()->IsToastShown("UndoCloseAllToast_1"));
}
ash::WaitForMilliseconds(
ash::ToastData::kDefaultToastDuration.InMilliseconds() +
ash::DesksTestApi::GetCloseAllWindowCloseTimeout().InMilliseconds());
EXPECT_FALSE(ash::DesksTestApi::DesksControllerCanUndoDeskRemoval());
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 1, 1);
}
// TODO(crbug.com/40927214): Re-enable test that flakily fails
#if defined(ADDRESS_SANITIZER) && defined(LEAK_SANITIZER)
#define MAYBE_LaunchAndUndo DISABLED_LaunchAndUndo
#else
#define MAYBE_LaunchAndUndo LaunchAndUndo
#endif
// Tests launch and removal of a desk. Tries to undo the removal.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, MAYBE_LaunchAndUndo) {
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
// The check is necessary as both desk animation and extension function is
// async. There is no guarantee which ones execute first.
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter remove_waiter;
// Remove a desk.
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
remove_desk_function.get(),
R"([")" + desk_id->GetString() +
R"(", { "combineDesks": false, "allowUndo": true }])",
browser()->profile());
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
remove_waiter.Wait();
}
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 1, 1);
histogram_tester.ExpectUniqueSample(
"Ash.DeskApi.RemoveDeskType", ash::DeskCloseType::kCloseAllWindowsAndWait,
1);
EXPECT_TRUE(ash::DesksTestApi::DesksControllerCanUndoDeskRemoval());
ash::DesksController::Get()->MaybeCancelDeskRemoval();
histogram_tester.ExpectTotalCount("Ash.DeskApi.CloseAllUndo", 1);
EXPECT_FALSE(ash::DesksTestApi::DesksControllerCanUndoDeskRemoval());
EXPECT_EQ(2, ash::DesksController::Get()->GetNumberOfDesks());
}
// Tests launch and removal of a desk. Should combine desks if both allowUndo
// and combineDesks are true.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, LaunchAndCombineUndoTrue) {
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
// The check is necessary as both desk animation and extension function is
// async. There is no guarantee which ones execute first.
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter remove_waiter;
// Remove a desk.
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
remove_desk_function.get(),
R"([")" + desk_id->GetString() +
R"(", { "combineDesks": true, "allowUndo": true }])",
browser()->profile());
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
remove_waiter.Wait();
}
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 1, 1);
histogram_tester.ExpectUniqueSample("Ash.DeskApi.RemoveDeskType",
ash::DeskCloseType::kCombineDesks, 1);
EXPECT_FALSE(ash::DesksTestApi::DesksControllerCanUndoDeskRemoval());
EXPECT_EQ(1, ash::DesksController::Get()->GetNumberOfDesks());
}
// Tests launch and removal of a desk. Desk removal results in desks combining.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, LaunchAndRemoveCombine) {
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
// The check is necessary as both desk animation and extension function is
// async. There is no guarantee which ones execute first.
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter remove_waiter;
// Remove a desk.
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
remove_desk_function.get(),
R"([")" + desk_id->GetString() +
R"(", { "combineDesks": true, "allowUndo": false }])",
browser()->profile());
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
remove_waiter.Wait();
}
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 1, 1);
histogram_tester.ExpectUniqueSample("Ash.DeskApi.RemoveDeskType",
ash::DeskCloseType::kCombineDesks, 1);
EXPECT_EQ(1, ash::DesksController::Get()->GetNumberOfDesks());
}
// Tests launch and list all desk.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, ListDesksTest) {
ash::DeskSwitchAnimationWaiter waiter;
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
// Asserts no error.
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
if (ash::DesksController::Get()->AreDesksBeingModified()) {
waiter.Wait();
}
// List All Desks.
auto list_desks_function =
base::MakeRefCounted<WmDesksPrivateGetAllDesksFunction>();
auto all_desks = api_test_utils::RunFunctionAndReturnSingleResult(
list_desks_function.get(), "[]", browser()->profile());
EXPECT_TRUE(all_desks->is_list());
EXPECT_EQ(2u, all_desks->GetList().size());
}
// Tests switch to different desk show trigger animation.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, SwitchToDifferentDeskTest) {
base::HistogramTester histogram_tester;
// Get the active desk.
auto get_active_desk_function =
base::MakeRefCounted<WmDesksPrivateGetActiveDeskFunction>();
// Asserts no error.
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
get_active_desk_function.get(), "[]", browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
ash::DeskSwitchAnimationWaiter launch_waiter;
// Asserts no error.
auto desk_id_1 = api_test_utils::RunFunctionAndReturnSingleResult(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_TRUE(desk_id_1->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id_1->GetString()).is_valid());
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 1, 1);
// Waiting for desk launch animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
launch_waiter.Wait();
}
ash::DeskSwitchAnimationWaiter switch_waiter;
// Switches to the previous desk.
auto switch_desk_function =
base::MakeRefCounted<WmDesksPrivateSwitchDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
switch_desk_function.get(), R"([")" + desk_id->GetString() + R"("])",
browser()->profile());
// Waiting for desk launch animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
switch_waiter.Wait();
}
auto get_active_desk_function_ =
base::MakeRefCounted<WmDesksPrivateGetActiveDeskFunction>();
// Asserts no error.
auto desk_id_2 = api_test_utils::RunFunctionAndReturnSingleResult(
get_active_desk_function_.get(), "[]", browser()->profile());
EXPECT_TRUE(desk_id_2->is_string());
EXPECT_EQ(desk_id->GetString(), desk_id_2->GetString());
histogram_tester.ExpectBucketCount("Ash.DeskApi.SwitchDesk.Result", 1, 1);
}
// Tests switch to current desk should skip animation.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, SwitchToCurrentDeskTest) {
// Get the desk desk.
auto get_active_desk_function =
base::MakeRefCounted<WmDesksPrivateGetActiveDeskFunction>();
// Asserts no error.
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
get_active_desk_function.get(), "[]", browser()->profile());
EXPECT_TRUE(desk_id->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id->GetString()).is_valid());
// Switches to the current desk.
auto switch_desk_function =
base::MakeRefCounted<WmDesksPrivateSwitchDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
switch_desk_function.get(), R"([")" + desk_id->GetString() + R"("])",
browser()->profile());
// Get the current desk.
auto get_active_desk_function_ =
base::MakeRefCounted<WmDesksPrivateGetActiveDeskFunction>();
auto desk_id_1 = api_test_utils::RunFunctionAndReturnSingleResult(
get_active_desk_function_.get(), "[]", browser()->profile());
EXPECT_TRUE(desk_id_1->is_string());
EXPECT_EQ(desk_id->GetString(), desk_id_1->GetString());
}
// Tests launch desks failed.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest,
LaunchDeskWhenMaxNumberExceedTest) {
// Max number of desks allowed is 8.
const int kMaxDeskIndex = 7;
for (int i = 0; i < kMaxDeskIndex; i++) {
ash::DesksController::Get()->NewDesk(ash::DesksCreationRemovalSource::kApi);
}
// Launch a desk.
auto launch_desk_function =
base::MakeRefCounted<WmDesksPrivateLaunchDeskFunction>();
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto error = api_test_utils::RunFunctionAndReturnError(
launch_desk_function.get(), R"([{"deskName":"test"}])",
browser()->profile());
EXPECT_EQ(error, "DesksCountCheckFailedError");
histogram_tester.ExpectBucketCount("Ash.DeskApi.LaunchDesk.Result", 0, 1);
}
// Tests remove desks failed.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, RemoveDeskWithInvalidIdTest) {
base::HistogramTester histogram_tester;
// The RunFunctionAndReturnSingleResult already asserts no error
auto remove_desk_function =
base::MakeRefCounted<WmDesksPrivateRemoveDeskFunction>();
auto error = api_test_utils::RunFunctionAndReturnError(
remove_desk_function.get(), R"(["invalid-id"])", browser()->profile());
EXPECT_EQ(error, "InvalidIdError");
histogram_tester.ExpectBucketCount("Ash.DeskApi.RemoveDesk.Result", 0, 1);
}
// Tests switch desks failed.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, SwitchDeskWithInvalidIdTest) {
base::HistogramTester histogram_tester;
// Switches to the current desk.
auto switch_desk_function =
base::MakeRefCounted<WmDesksPrivateSwitchDeskFunction>();
auto error = api_test_utils::RunFunctionAndReturnError(
switch_desk_function.get(), R"(["invalid-id"])", browser()->profile());
EXPECT_EQ(error, "InvalidIdError");
histogram_tester.ExpectBucketCount("Ash.DeskApi.SwitchDesk.Result", 0, 1);
}
// Tests set all desks with invalid ID.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest,
SetAllDesksWindowWithInvalidIdTest) {
base::HistogramTester histogram_tester;
// Switches to the current desk.
auto all_desk_function =
base::MakeRefCounted<WmDesksPrivateSetWindowPropertiesFunction>();
auto error = api_test_utils::RunFunctionAndReturnError(
all_desk_function.get(), R"([123,{"allDesks":true}])",
browser()->profile());
EXPECT_EQ(error, "ResourceNotFoundError");
histogram_tester.ExpectBucketCount("Ash.DeskApi.AllDesk.Result", 0, 1);
}
// Tests save and recall a desk.
// TODO(crbug.com/40902046): Test is flaky.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, DISABLED_SaveAndRecallDeskTest) {
// Save a desk.
auto save_desk_function =
base::MakeRefCounted<WmDesksPrivateSaveActiveDeskFunction>();
ash::DeskSwitchAnimationWaiter save_desk_waiter;
// Asserts no error.
auto result = api_test_utils::RunFunctionAndReturnSingleResult(
save_desk_function.get(), R"([])", browser()->profile());
EXPECT_TRUE(result->is_dict());
auto desk_id = result->GetDict().Find("savedDeskUuid")->GetString();
EXPECT_TRUE(base::Uuid::ParseCaseInsensitive(desk_id).is_valid());
// Waiting for desk launch animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
save_desk_waiter.Wait();
}
// List saved desks.
auto list_desk_function =
base::MakeRefCounted<WmDesksPrivateGetSavedDesksFunction>();
// Asserts no error.
auto result_1 = api_test_utils::RunFunctionAndReturnSingleResult(
list_desk_function.get(), R"([])", browser()->profile());
EXPECT_TRUE(result_1->is_list());
EXPECT_EQ(1u, result_1->GetList().size());
EXPECT_TRUE(result_1->GetList().front().GetDict().Find("savedDeskUuid"));
EXPECT_TRUE(result_1->GetList().front().GetDict().Find("savedDeskName"));
EXPECT_TRUE(result_1->GetList().front().GetDict().Find("savedDeskType"));
ash::DeskSwitchAnimationWaiter recall_desk_waiter;
// Recall a desk.
auto recall_desk_function =
base::MakeRefCounted<WmDesksPrivateRecallSavedDeskFunction>();
auto desk_id_1 = api_test_utils::RunFunctionAndReturnSingleResult(
recall_desk_function.get(), R"([")" + desk_id + R"("])",
browser()->profile());
EXPECT_TRUE(desk_id_1->is_string());
EXPECT_TRUE(
base::Uuid::ParseCaseInsensitive(desk_id_1->GetString()).is_valid());
// Waiting for desk removal animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
recall_desk_waiter.Wait();
}
}
// Tests save and delete a desk.
// TODO(crbug.com/40902046): Flaky on linux-chromeos-rel.
#if defined(NDEBUG)
#define MAYBE_SaveAndDeleteDeskTest DISABLED_SaveAndDeleteDeskTest
#else
#define MAYBE_SaveAndDeleteDeskTest SaveAndDeleteDeskTest
#endif
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, MAYBE_SaveAndDeleteDeskTest) {
// Save a desk.
auto save_desk_function =
base::MakeRefCounted<WmDesksPrivateSaveActiveDeskFunction>();
ash::DeskSwitchAnimationWaiter save_desk_waiter;
// Asserts no error.
auto result = api_test_utils::RunFunctionAndReturnSingleResult(
save_desk_function.get(), R"([])", browser()->profile());
EXPECT_TRUE(result->is_dict());
auto desk_id = result->GetDict().Find("savedDeskUuid")->GetString();
EXPECT_TRUE(base::Uuid::ParseCaseInsensitive(desk_id).is_valid());
// Waiting for desk launch animation to settle
if (ash::DesksController::Get()->AreDesksBeingModified()) {
save_desk_waiter.Wait();
}
// Delete a saved desk.
auto deleted_saved_desk_function =
base::MakeRefCounted<WmDesksPrivateDeleteSavedDeskFunction>();
api_test_utils::RunFunctionAndReturnSingleResult(
deleted_saved_desk_function.get(), R"([")" + desk_id + R"("])",
browser()->profile());
}
// Tests retrieve desk with deskID.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, GetDeskByIDTest) {
// Get current desk id.
auto desk_id = api_test_utils::RunFunctionAndReturnSingleResult(
base::MakeRefCounted<WmDesksPrivateGetActiveDeskFunction>().get(), "[]",
browser()->profile());
// Retrieve desk by Id.
auto get_desk_by_id_function =
base::MakeRefCounted<WmDesksPrivateGetDeskByIDFunction>();
auto result = api_test_utils::RunFunctionAndReturnSingleResult(
get_desk_by_id_function.get(), R"([")" + desk_id->GetString() + R"("])",
browser()->profile());
EXPECT_TRUE(result->is_dict());
auto* desk_id_1 = result->GetDict().Find("deskUuid");
auto* desk_name = result->GetDict().Find("deskName");
ASSERT_TRUE(desk_id_1->is_string());
EXPECT_EQ(desk_id->GetString(), desk_id_1->GetString());
EXPECT_TRUE(desk_name->is_string());
}
// Tests retrieve desk with invalid deskID.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, GetDeskByInvalidIDTest) {
// Retrieve desk by Id.
auto get_desk_by_id_function =
base::MakeRefCounted<WmDesksPrivateGetDeskByIDFunction>();
auto error = api_test_utils::RunFunctionAndReturnError(
get_desk_by_id_function.get(), R"(["invalid-id"])", browser()->profile());
EXPECT_EQ(error, "InvalidIdError");
}
// Tests retrieve desk with non-exist deskID.
IN_PROC_BROWSER_TEST_F(WmDesksPrivateApiTest, GetDeskByNonExistIDTest) {
auto desk_id = base::Uuid::GenerateRandomV4().AsLowercaseString();
// Retrieve desk by Id.
auto get_desk_by_id_function =
base::MakeRefCounted<WmDesksPrivateGetDeskByIDFunction>();
auto error = api_test_utils::RunFunctionAndReturnError(
get_desk_by_id_function.get(), R"([")" + desk_id + R"("])",
browser()->profile());
EXPECT_EQ(error, "ResourceNotFoundError");
}
} // namespace extensions
Codebase Browser
chromium