// Copyright 2019 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include <memory>

#include "build/chromeos_buildflags.h"
#include "chrome/browser/extensions/extension_apitest.h"
#include "content/public/test/browser_test.h"
#include "extensions/browser/api/idle/idle_manager_factory.h"
#include "extensions/browser/api/idle/test_idle_provider.h"

namespace extensions {

using ContextType = ExtensionBrowserTest::ContextType;

class IdleApiTest : public ExtensionApiTest,
                    public testing::WithParamInterface<ContextType> {
 public:
  IdleApiTest() : ExtensionApiTest(GetParam()) {}
  ~IdleApiTest() override = default;
  IdleApiTest(const IdleApiTest& other) = delete;
  IdleApiTest& operator=(const IdleApiTest& other) = delete;

 protected:
  void SetTestIdleProvider(int idle_time, bool locked) {
    auto idle_provider = std::make_unique<TestIdleProvider>();
    idle_provider->set_idle_time(idle_time);
    idle_provider->set_locked(locked);
    IdleManagerFactory::GetForBrowserContext(profile())
        ->SetIdleTimeProviderForTest(std::move(idle_provider));
  }
};

INSTANTIATE_TEST_SUITE_P(PersistentBackground,
                         IdleApiTest,
                         ::testing::Values(ContextType::kPersistentBackground));

INSTANTIATE_TEST_SUITE_P(ServiceWorker,
                         IdleApiTest,
                         ::testing::Values(ContextType::kServiceWorker));

IN_PROC_BROWSER_TEST_P(IdleApiTest, QueryStateActive) {
  // Set up a test IdleProvider in the active state.
  SetTestIdleProvider(/*idle_time=*/0, /*locked=*/false);
  ASSERT_TRUE(
      RunExtensionTest("idle/query_state", {.custom_arg = "queryStateActive"}));
}

IN_PROC_BROWSER_TEST_P(IdleApiTest, QueryStateIdle) {
  // Set up a test IdleProvider in the idle state. The JS test uses the
  // value 15 for its "intervalInSeconds" value.
  SetTestIdleProvider(/*idle_time=*/15, /*locked=*/false);
  ASSERT_TRUE(
      RunExtensionTest("idle/query_state", {.custom_arg = "queryStateIdle"}));
}

IN_PROC_BROWSER_TEST_P(IdleApiTest, QueryStateAlmostIdle) {
  // Set up a test IdleProvider in the active state, just about to transition
  // to idle. The JS test uses the value 15 for its "intervalInSeconds" value.
  SetTestIdleProvider(/*idle_time=*/14, /*locked=*/false);
  ASSERT_TRUE(
      RunExtensionTest("idle/query_state", {.custom_arg = "queryStateActive"}));
}

IN_PROC_BROWSER_TEST_P(IdleApiTest, QueryStateLocked) {
  // Set up a test IdleProvider in the locked state.
  SetTestIdleProvider(/*idle_time=*/0, /*locked=*/true);
  ASSERT_TRUE(
      RunExtensionTest("idle/query_state", {.custom_arg = "queryStateLocked"}));
}

#if BUILDFLAG(IS_CHROMEOS_ASH) || BUILDFLAG(IS_CHROMEOS_LACROS)
IN_PROC_BROWSER_TEST_P(IdleApiTest, IdleGetAutoLockDelay) {
  ASSERT_TRUE(RunExtensionTest("idle/get_auto_lock_delay")) << message_;
}
#else
IN_PROC_BROWSER_TEST_P(IdleApiTest, UnsupportedIdleGetAutoLockDelay) {
  ASSERT_TRUE(RunExtensionTest("idle/unsupported_get_auto_lock_delay"));
}
#endif
}  // namespace extensions
