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

#include "build/build_config.h"
#include "chrome/browser/vr/test/mock_xr_device_hook_base.h"
#include "chrome/browser/vr/test/multi_class_browser_test.h"
#include "chrome/browser/vr/test/webxr_vr_browser_test.h"
#include "content/public/browser/service_process_host.h"
#include "content/public/test/browser_test_utils.h"
#include "device/vr/public/mojom/isolated_xr_service.mojom.h"

// Android doesn't use an isolated device service, so these tests don't need to
// be built here.
#if BUILDFLAG(IS_ANDROID)
#error "Android does not have an isolated device service"
#endif  // if !BUILDFLAG(IS_ANDROID)
namespace vr {

// Tests that we can recover from a crash/disconnect on the DeviceService
WEBXR_VR_ALL_RUNTIMES_BROWSER_TEST_F(TestDeviceServiceDisconnect) {
  MockXRDeviceHookBase device_hook;

  t->LoadFileAndAwaitInitialization("test_isolated_device_service_disconnect");

  t->EnterSessionWithUserGestureOrFail();

  // We don't care how many device changes we've received prior to this point.
  // We should now be at a steady state, so what we really care about is the
  // number of device changes after this point.
  t->RunJavaScriptOrFail("resetDeviceChanges()");

  for (const auto& info :
       content::ServiceProcessHost::GetRunningProcessInfo()) {
    if (info.IsService<device::mojom::XRDeviceService>()) {
      info.GetProcess().Terminate(/*exit_code=*/1, /*wait=*/false);
    }
  }

  // Ensure that we've actually exited the session.
  t->PollJavaScriptBooleanOrFail(
      "sessionInfos[sessionTypes.IMMERSIVE].currentSession === null",
      WebXrVrBrowserTestBase::kPollTimeoutLong);

  // We expect one change indicating the device was disconnected, and then
  // one more indicating that the device was re-connected.
  t->PollJavaScriptBooleanOrFail("deviceChanges === 2",
                                 WebXrVrBrowserTestBase::kPollTimeoutMedium);

  // One last check now that we have the device change that we can actually
  // still enter an immersive session.
  t->EnterSessionWithUserGestureOrFail();
}
}  // namespace vr