// Copyright (c) 2011 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 <string>

#include "chrome/browser/tab_contents/render_view_context_menu.h"

#include "chrome/browser/extensions/extension_prefs.h"
#include "chrome/common/extensions/url_pattern.h"
#include "googleurl/src/gurl.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/glue/context_menu.h"

#include "third_party/WebKit/Source/WebKit/chromium/public/WebContextMenuData.h"

class RenderViewContextMenuTest : public testing::Test {
 public:
  RenderViewContextMenuTest() { }

 protected:
  // Proxy defined here to minimize friend classes in RenderViewContextMenu
  static bool ExtensionContextAndPatternMatch(
      const ContextMenuParams& params,
      ExtensionMenuItem::ContextList contexts,
      const URLPatternSet& patterns) {
    return RenderViewContextMenu::ExtensionContextAndPatternMatch(params,
        contexts, patterns);
  }

 private:
  DISALLOW_COPY_AND_ASSIGN(RenderViewContextMenuTest);
};

// Generates a ContextMenuParams that matches the specified contexts.
static ContextMenuParams CreateParams(int contexts) {
  ContextMenuParams rv;
  rv.is_editable = false;
  rv.media_type = WebKit::WebContextMenuData::MediaTypeNone;
  rv.page_url = GURL("http://test.page/");

  static const char16 selected_text[] = { 's', 'e', 'l', 0 };
  if (contexts & ExtensionMenuItem::SELECTION)
    rv.selection_text = selected_text;

  if (contexts & ExtensionMenuItem::LINK)
    rv.link_url = GURL("http://test.link/");

  if (contexts & ExtensionMenuItem::EDITABLE)
    rv.is_editable = true;

  if (contexts & ExtensionMenuItem::IMAGE) {
    rv.src_url = GURL("http://test.image/");
    rv.media_type = WebKit::WebContextMenuData::MediaTypeImage;
  }

  if (contexts & ExtensionMenuItem::VIDEO) {
    rv.src_url = GURL("http://test.video/");
    rv.media_type = WebKit::WebContextMenuData::MediaTypeVideo;
  }

  if (contexts & ExtensionMenuItem::AUDIO) {
    rv.src_url = GURL("http://test.audio/");
    rv.media_type = WebKit::WebContextMenuData::MediaTypeAudio;
  }

  if (contexts & ExtensionMenuItem::FRAME)
    rv.frame_url = GURL("http://test.frame/");

  return rv;
}

// Generates a URLPatternSet with a single pattern
static URLPatternSet CreatePatternSet(const std::string& pattern) {
  URLPattern target(URLPattern::SCHEME_HTTP);
  target.Parse(pattern, URLPattern::IGNORE_PORTS);

  URLPatternSet rv;
  rv.AddPattern(target);

  return rv;
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForPage) {
  ContextMenuParams params = CreateParams(0);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::PAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetCheckedForLink) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::LINK);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::PAGE);
  contexts.Add(ExtensionMenuItem::LINK);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetCheckedForImage) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::IMAGE);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::PAGE);
  contexts.Add(ExtensionMenuItem::IMAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetCheckedForVideo) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::VIDEO);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::PAGE);
  contexts.Add(ExtensionMenuItem::VIDEO);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetCheckedForAudio) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::AUDIO);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::PAGE);
  contexts.Add(ExtensionMenuItem::AUDIO);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, MatchWhenLinkedImageMatchesTarget) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::IMAGE |
                                          ExtensionMenuItem::LINK);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::LINK);
  contexts.Add(ExtensionMenuItem::IMAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.link/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, MatchWhenLinkedImageMatchesSource) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::IMAGE |
                                          ExtensionMenuItem::LINK);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::LINK);
  contexts.Add(ExtensionMenuItem::IMAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.image/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, NoMatchWhenLinkedImageMatchesNeither) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::IMAGE |
                                          ExtensionMenuItem::LINK);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::LINK);
  contexts.Add(ExtensionMenuItem::IMAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_FALSE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForFrame) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::FRAME);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::FRAME);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForEditable) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::EDITABLE);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::EDITABLE);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelection) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::SELECTION);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::SELECTION);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelectionOnLink) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::SELECTION |
                                          ExtensionMenuItem::LINK);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::SELECTION);
  contexts.Add(ExtensionMenuItem::LINK);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}

TEST_F(RenderViewContextMenuTest, TargetIgnoredForSelectionOnImage) {
  ContextMenuParams params = CreateParams(ExtensionMenuItem::SELECTION |
                                          ExtensionMenuItem::IMAGE);

  ExtensionMenuItem::ContextList contexts;
  contexts.Add(ExtensionMenuItem::SELECTION);
  contexts.Add(ExtensionMenuItem::IMAGE);

  URLPatternSet patterns = CreatePatternSet("*://test.none/*");

  EXPECT_TRUE(ExtensionContextAndPatternMatch(params, contexts, patterns));
}