// Copyright 2022 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/test/media_router/access_code_cast/access_code_cast_integration_browsertest.h"
#include "components/signin/public/base/consent_level.h"
#include "components/signin/public/identity_manager/identity_manager.h"
#include "google_apis/gaia/google_service_auth_error.h"

#if BUILDFLAG(IS_WIN)
#include "base/win/windows_version.h"
#endif
#include "chrome/browser/media/router/discovery/access_code/access_code_cast_constants.h"
#include "components/sessions/content/session_tab_helper.h"
#include "services/network/test/test_network_connection_tracker.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace media_router {

class AccessCodeCastHandlerBrowserTest
    : public AccessCodeCastIntegrationBrowserTest {};

// TODO(b/235275754): Add a case for when the network is connected and we
// surface a server error.
//
// TODO(crbug.com/513366591): Re-enable this test on Linux once the flakiness is
// fixed.
#if BUILDFLAG(IS_LINUX)
#define MAYBE_ExpectNetworkErrorWhenNoNetwork \
  DISABLED_ExpectNetworkErrorWhenNoNetwork
#else
#define MAYBE_ExpectNetworkErrorWhenNoNetwork ExpectNetworkErrorWhenNoNetwork
#endif
IN_PROC_BROWSER_TEST_F(AccessCodeCastHandlerBrowserTest,
                       MAYBE_ExpectNetworkErrorWhenNoNetwork) {
  EnableAccessCodeCasting();

  // This tests that if the network is not present (we are not connected to the
  // internet), we will see a server error in the access code dialog box.
  SetUpPrimaryAccountWithHostedDomain(browser()->GetProfile());

  network::TestNetworkConnectionTracker::GetInstance()->SetConnectionType(
      net::NetworkChangeNotifier::ConnectionType::CONNECTION_NONE);

  auto* dialog_contents = ShowDialog();
  SetAccessCode("abcdef", dialog_contents);
  PressSubmit(dialog_contents);

  // This error code corresponds to
  // ErrorMessage.NETWORK::AddSinkResultCode.SERVICE_NOT_PRESENT. This error
  // code 3 refers to the ErrorMessage described within
  // chrome/browser/resources/access_code_cast/error_message/error_message.ts
  EXPECT_EQ(3, WaitForAddSinkErrorCode(dialog_contents));
  CloseDialog(dialog_contents);
}

IN_PROC_BROWSER_TEST_F(AccessCodeCastHandlerBrowserTest,
                       ReturnSuccessfulResponse) {
  AddScreenplayTag(AccessCodeCastIntegrationBrowserTest::
                       kAccessCodeCastNewDeviceScreenplayTag);

  const char kEndpointResponseSuccess[] =
      R"({
      "device": {
        "displayName": "test_device",
        "id": "1234",
        "deviceCapabilities": {
          "videoOut": true,
          "videoIn": true,
          "audioOut": true,
          "audioIn": true,
          "devMode": true
        },
        "networkInfo": {
          "hostName": "GoogleNet",
          "port": "666",
          "ipV4Address": "192.0.2.146",
          "ipV6Address": "2001:0db8:85a3:0000:0000:8a2e:0370:7334"
        }
      }
    })";

  // Mock a successful fetch from our server.
  SetEndpointFetcherMockResponse(kEndpointResponseSuccess, net::HTTP_OK,
                                 net::OK);

  // Simulate a successful opening of the channel.
  SetMockOpenChannelCallbackResponse(true);

  EnableAccessCodeCasting();

  SetUpPrimaryAccountWithHostedDomain(browser()->GetProfile());

  auto* dialog_contents = ShowDialog();
  SetAccessCode("abcdef", dialog_contents);
  ExpectStartRouteCallFromTabMirroring(
      "cast:<1234>",
      MediaSource::ForTab(
          sessions::SessionTabHelper::IdForTab(web_contents()).id())
          .id(),
      web_contents());

  PressSubmitAndWaitForClose(dialog_contents);
}

IN_PROC_BROWSER_TEST_F(AccessCodeCastHandlerBrowserTest,
                       ExpectErrorWhenNoSigninError) {
  EnableAccessCodeCasting();

  // This tests that an account with auth error will throw a generic error.
  SetUpPrimaryAccountWithHostedDomain(browser()->GetProfile());
  signin::IdentityManager* identity_manager =
      AccessCodeCastSinkServiceFactory::GetForProfile(browser()->GetProfile())
          ->GetIdentityManager();
  signin::UpdatePersistentErrorOfRefreshTokenForAccount(
      identity_manager,
      identity_manager->GetPrimaryAccountId(signin::ConsentLevel::kSignin),
      GoogleServiceAuthError::FromInvalidGaiaCredentialsReason(
          GoogleServiceAuthError::InvalidGaiaCredentialsReason::UNKNOWN));

  auto* dialog_contents = ShowDialog();
  SetAccessCode("abcdef", dialog_contents);
  PressSubmit(dialog_contents);

  // This error code corresponds to
  // ErrorMessage.PROFILE_SYNC_ERROR::AddSinkResultCode.PROFILE_SYNC_ERROR. This
  // error code 6 refers to the ErrorMessage described within
  // chrome/browser/resources/access_code_cast/error_message/error_message.ts
  EXPECT_EQ(6, WaitForAddSinkErrorCode(dialog_contents));
  CloseDialog(dialog_contents);
}

IN_PROC_BROWSER_TEST_F(AccessCodeCastHandlerBrowserTest,
                       ExpectDifferentNetworkErrorWhenChannelError) {
  const char kEndpointResponseSuccess[] =
      R"({
      "device": {
        "displayName": "test_device",
        "id": "1234",
        "deviceCapabilities": {
          "videoOut": true,
          "videoIn": true,
          "audioOut": true,
          "audioIn": true,
          "devMode": true
        },
        "networkInfo": {
          "hostName": "GoogleNet",
          "port": "666",
          "ipV4Address": "192.0.2.146",
          "ipV6Address": "2001:0db8:85a3:0000:0000:8a2e:0370:7334"
        }
      }
    })";

  // Mock a successful fetch from our server.
  SetEndpointFetcherMockResponse(kEndpointResponseSuccess, net::HTTP_OK,
                                 net::OK);

  // Simulate the failure of opening a channel. This will trigger the correct
  // error code below.
  SetMockOpenChannelCallbackResponse(false);

  EnableAccessCodeCasting();

  SetUpPrimaryAccountWithHostedDomain(browser()->GetProfile());

  auto* dialog_contents = ShowDialog();
  SetAccessCode("abcdef", dialog_contents);

  PressSubmit(dialog_contents);

  // This error code corresponds to
  // ErrorMessage.DIFFERENT_NETWORK::AddSinkResultCode.CHANNEL_OPEN_ERROR. This
  // error code 7 refers to the ErrorMessage described within
  // chrome/browser/resources/access_code_cast/error_message/error_message.ts
  EXPECT_EQ(7, WaitForAddSinkErrorCode(dialog_contents));
  CloseDialog(dialog_contents);
}

// TODO(b/260627828): This test is flaky on all platforms.
IN_PROC_BROWSER_TEST_F(AccessCodeCastHandlerBrowserTest,
                       DISABLED_ReturnSuccessfulResponseUsingKeyPress) {
  const char kEndpointResponseSuccess[] =
      R"({
      "device": {
        "displayName": "test_device",
        "id": "1234",
        "deviceCapabilities": {
          "videoOut": true,
          "videoIn": true,
          "audioOut": true,
          "audioIn": true,
          "devMode": true
        },
        "networkInfo": {
          "hostName": "GoogleNet",
          "port": "666",
          "ipV4Address": "192.0.2.146",
          "ipV6Address": "2001:0db8:85a3:0000:0000:8a2e:0370:7334"
        }
      }
    })";

  // Mock a successful fetch from our server.
  SetEndpointFetcherMockResponse(kEndpointResponseSuccess, net::HTTP_OK,
                                 net::OK);

  // Simulate a successful opening of the channel.
  SetMockOpenChannelCallbackResponse(true);

  EnableAccessCodeCasting();

  SetUpPrimaryAccountWithHostedDomain(browser()->GetProfile());

  auto* dialog_contents = ShowDialog();
  SetAccessCodeUsingKeyPress("ABCDEF");
  ExpectStartRouteCallFromTabMirroring(
      "cast:<1234>",
      MediaSource::ForTab(
          sessions::SessionTabHelper::IdForTab(web_contents()).id())
          .id(),
      web_contents());

  PressSubmitAndWaitForCloseUsingKeyPress(dialog_contents);
}

}  // namespace media_router