WindowsUIElementTests.cpp

Enable keyboard shortcuts

/* This Source Code Form is subject to the terms of the Mozilla Public

 * License, v. 2.0. If a copy of the MPL was not distributed with this

 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#include <gtest/gtest.h>

#include <gmock/gmock.h>

#include <cstdint>

#include <windows.h>

#include <ole2.h>

#include <uiautomation.h>

#include "WindowsDefaultBrowser.h"

#include "WindowsDefaultBrowserTests.h"

#include "WindowsUIElement.h"

#include "mozilla/RefPtr.h"

#include "mozilla/SpinEventLoopUntil.h"

#include "mozilla/WindowsVersion.h"

#include "mozilla/gtest/MozAssertions.h"

#include "nsCOMPtr.h"

#include "nsISerialEventTarget.h"

#include "nsITimer.h"

#include "nsLiteralString.h"

#include "nsThreadUtils.h"

static bool IsElementFocus(const UIElement& aElement) {

  BOOL isFocus{FALSE};

  aElement->get_CurrentHasKeyboardFocus(&isFocus);

  return isFocus;

class WindowsUIElementTests : public FindSetDefaultBrowserButtonTests {};

TEST_F(WindowsUIElementTests, DefaultBrowserButtonFocused) {

  ASSERT_TRUE(LaunchModernSettingsDialogDefaultApps());

  auto [window, button]{WaitForSetDefaultBrowserButton()};

  ASSERT_THAT(window, testing::NotNull());

  ASSERT_THAT(button, testing::NotNull());

  RefPtr<mozilla::WindowsUIElement> element{

      new mozilla::WindowsUIElement(window, button)};

  element->Focus();

  ASSERT_TRUE(IsElementFocus(button));

TEST_F(WindowsUIElementTests, DefaultBrowserButtonStopsMoving) {

  if (!mozilla::IsWin11OrLater()) {

    // This test is intended to run only on Win11 since it's part of the feature

    // that displays the Kit image behind the Set Default Browser button

    return;

  ASSERT_TRUE(LaunchModernSettingsDialogDefaultApps());

  auto [window, button]{WaitForSetDefaultBrowserButton()};

  ASSERT_THAT(window, testing::NotNull());

  ASSERT_THAT(button, testing::NotNull());

  RefPtr<mozilla::WindowsUIElement> element{

      new mozilla::WindowsUIElement(window, button)};

  nsCOMPtr<nsISerialEventTarget> queue;

  ASSERT_NS_SUCCEEDED(NS_CreateBackgroundTaskQueue(

      "WindowsUIElementTests::DefaultBrowserButtonStopsMovingQueue",

      getter_AddRefs(queue)));

  int ticks{0};

  bool elementIsMoving{true};

  bool done{false};

  nsCOMPtr<nsITimer> timer;

  auto callback{[element, &ticks, &elementIsMoving, &done](nsITimer* aTimer) {

    elementIsMoving = element->IsMoving().valueOr(true);

    constexpr int kMaxTicks{10};

    if (!elementIsMoving || ++ticks >= kMaxTicks) {

      aTimer->Cancel();

      NS_DispatchToMainThread(NS_NewRunnableFunction(

          "WindowsUIElementTests::DefaultBrowserButtonStopsMovingDone",

          [&done] { done = true; }));

}};

  const uint32_t kDelayMs{500};

  ASSERT_NS_SUCCEEDED(NS_NewTimerWithCallback(

      getter_AddRefs(timer), callback, kDelayMs, nsITimer::TYPE_REPEATING_SLACK,

      "WindowsUIElementTests::DefaultBrowserButtonStopsMovingTimer"_ns, queue));

  mozilla::SpinEventLoopUntil(

      "WindowsUIElementTests::DefaultBrowserButtonStopsMovingLoop"_ns,

      [&done] { return bool(done); });

  ASSERT_FALSE(elementIsMoving);