// Copyright (c) 2010 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 "chrome/browser/extensions/extension_apitest.h"
#include "chrome/browser/prefs/pref_service.h"
#include "chrome/browser/prefs/proxy_prefs.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/common/chrome_switches.h"
#include "chrome/common/extensions/extension.h"
#include "chrome/common/pref_names.h"

// Tests direct connection settings.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxyDirectSettings) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/direct")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  int mode = pref_service->GetInteger(prefs::kProxyMode);
  EXPECT_EQ(ProxyPrefs::MODE_DIRECT, mode);

  // Other proxy prefs should also be set, so they're all controlled from one
  // place.
  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyPacUrl));

  // No manual proxy prefs were set.
  pref = pref_service->FindPreference(prefs::kProxyServer);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyServer));
}

// Tests auto-detect settings.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxyAutoSettings) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/auto")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  int mode = pref_service->GetInteger(prefs::kProxyMode);
  EXPECT_EQ(ProxyPrefs::MODE_AUTO_DETECT, mode);

  // Other proxy prefs should also be set, so they're all controlled from one
  // place.
  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyPacUrl));
}

// Tests PAC proxy settings.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxyPacScript) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/pac")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  int mode = pref_service->GetInteger(prefs::kProxyMode);
  EXPECT_EQ(ProxyPrefs::MODE_PAC_SCRIPT, mode);

  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  std::string pac_url = pref_service->GetString(prefs::kProxyPacUrl);
  EXPECT_EQ("http://wpad/windows.pac", pac_url);

  // No manual proxy prefs were set.
  pref = pref_service->FindPreference(prefs::kProxyServer);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyServer));
}

// Tests setting a single proxy to cover all schemes.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxyFixedSingle) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/single")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  // There should be no values superseding the extension-set proxy in this test.
  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyServer);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  std::string proxy_server = pref_service->GetString(prefs::kProxyServer);
  EXPECT_EQ("http=http://127.0.0.1:100;"
            "https=http://127.0.0.1:100;"
            "ftp=http://127.0.0.1:100;"
            "socks=http://9.9.9.9", proxy_server);

  // Other proxy prefs should also be set, so they're all controlled from one
  // place.
  pref = pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ(ProxyPrefs::MODE_FIXED_SERVERS,
            pref_service->GetInteger(prefs::kProxyMode));

  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyPacUrl));
}

// Tests setting to use the system's proxy settings.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxySystem) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/system")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  // There should be no values superseding the extension-set proxy in this test.
  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());
  int proxy_server_mode = pref_service->GetInteger(prefs::kProxyMode);
  EXPECT_EQ(ProxyPrefs::MODE_SYSTEM, proxy_server_mode);

  // Other proxy prefs should also be set, so they're all controlled from one
  // place.
  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyPacUrl));

  // No manual proxy prefs were set.
  pref = pref_service->FindPreference(prefs::kProxyServer);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyServer));
}

// Tests setting separate proxies for each scheme.
IN_PROC_BROWSER_TEST_F(ExtensionApiTest, ProxyFixedIndividual) {
  CommandLine::ForCurrentProcess()->AppendSwitch(
      switches::kEnableExperimentalExtensionApis);

  ASSERT_TRUE(RunExtensionTest("proxy/individual")) << message_;
  const Extension* extension = GetSingleLoadedExtension();
  ASSERT_TRUE(extension);

  PrefService* pref_service = browser()->profile()->GetPrefs();

  // There should be no values superseding the extension-set proxy in this test.
  const PrefService::Preference* pref =
      pref_service->FindPreference(prefs::kProxyServer);
  ASSERT_TRUE(pref != NULL);
  ASSERT_TRUE(pref->IsExtensionControlled());

  std::string proxy_server = pref_service->GetString(prefs::kProxyServer);
  EXPECT_EQ("http=http://1.1.1.1;"
            "https=socks://2.2.2.2;"
            "ftp=http://3.3.3.3:9000;"
            "socks=socks4://4.4.4.4:9090", proxy_server);

  // Other proxy prefs should also be set, so they're all controlled from one
  // place.
  pref = pref_service->FindPreference(prefs::kProxyMode);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ(ProxyPrefs::MODE_FIXED_SERVERS,
            pref_service->GetInteger(prefs::kProxyMode));

  pref = pref_service->FindPreference(prefs::kProxyPacUrl);
  ASSERT_TRUE(pref != NULL);
  EXPECT_TRUE(pref->IsExtensionControlled());
  EXPECT_EQ("", pref_service->GetString(prefs::kProxyPacUrl));
}