// 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 "base/basictypes.h"
#include "base/scoped_ptr.h"
#include "chrome/browser/browser_about_handler.h"
#include "chrome/common/about_handler.h"
#include "chrome/common/url_constants.h"
#include "googleurl/src/gurl.h"
#include "testing/gtest/include/gtest/gtest.h"

TEST(BrowserAboutHandlerTest, WillHandleBrowserAboutURL) {
  struct AboutURLTestData {
    GURL test_url;
    GURL result_url;
    bool about_handled;
    bool browser_handled;
  } test_data[] = {
      {
        GURL("http://google.com"),
        GURL("http://google.com"),
        false,
        false
      },
      {
        GURL(chrome::kAboutBlankURL),
        GURL(chrome::kAboutBlankURL),
        false,
        false
      },
      {
        GURL(std::string(chrome::kAboutCacheURL) + "/mercury"),
        GURL(std::string(chrome::kNetworkViewCacheURL) + "mercury"),
        false,
        true
      },
      {
        GURL(std::string(chrome::kAboutNetInternalsURL) + "/venus"),
        GURL(std::string(chrome::kNetworkViewInternalsURL) + "venus"),
        false,
        true
      },
      {
        GURL(std::string(chrome::kAboutAppCacheInternalsURL) + "/earth"),
        GURL(std::string(chrome::kAppCacheViewInternalsURL) + "earth"),
        false,
        true
      },
      {
        GURL(chrome::kAboutPluginsURL),
        GURL(chrome::kChromeUIPluginsURL),
        false,
        true
      },
      {
        GURL(chrome::kAboutCrashURL),
        GURL(chrome::kAboutCrashURL),
        true,
        false
      },
      {
        GURL(chrome::kAboutHangURL),
        GURL(chrome::kAboutHangURL),
        true,
        false
      },
      {
        GURL(chrome::kAboutShorthangURL),
        GURL(chrome::kAboutShorthangURL),
        true,
        false
      },
      {
        GURL("about:memory"),
        GURL("chrome://about/memory-redirect"),
        false,
        true
      },
      {
        GURL("about:mars"),
        GURL("chrome://about/mars"),
        false,
        true
      },
  };

  for (size_t i = 0; i < ARRAYSIZE_UNSAFE(test_data); ++i) {
    GURL url(test_data[i].test_url);
    EXPECT_EQ(test_data[i].about_handled,
              chrome_about_handler::WillHandle(url));
    EXPECT_EQ(test_data[i].browser_handled,
              WillHandleBrowserAboutURL(&url, NULL));
    EXPECT_EQ(test_data[i].result_url, url);
  }

  // Crash the browser process for about:inducebrowsercrashforrealz.
  GURL url(chrome::kAboutBrowserCrash);
  EXPECT_DEATH(WillHandleBrowserAboutURL(&url, NULL), "");
}