// Copyright 2020 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/renderer/chrome_content_settings_agent_delegate.h"
#include "chrome/test/base/chrome_render_view_test.h"
#include "content/public/renderer/render_frame.h"
#include "content/public/renderer/render_view.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/blink/public/common/associated_interfaces/associated_interface_registry.h"

class ChromeContentSettingsAgentDelegateBrowserTest
    : public ChromeRenderViewTest {
 protected:
  void SetUp() override {
    ChromeRenderViewTest::SetUp();

    // Set up a fake url loader factory to ensure that script loader can create
    // a WebURLLoader.
    CreateFakeWebURLLoaderFactory();

    // Unbind the ContentSettingsAgent interface that would be registered by
    // the ChromeContentSettingsAgent created when the render frame is created.
    GetMainRenderFrame()->GetAssociatedInterfaceRegistry()->RemoveInterface(
        content_settings::mojom::ContentSettingsAgent::Name_);
  }
};

TEST_F(ChromeContentSettingsAgentDelegateBrowserTest,
       PluginsTemporarilyAllowed) {
  // Load some HTML.
  LoadHTML("<html>Foo</html>");

  std::string foo_plugin = "foo";
  std::string bar_plugin = "bar";

  auto* delegate =
      ChromeContentSettingsAgentDelegate::Get(GetMainRenderFrame());
  EXPECT_FALSE(delegate->IsPluginTemporarilyAllowed(foo_plugin));

  // Temporarily allow the "foo" plugin.
  delegate->AllowPluginTemporarily(foo_plugin);
  EXPECT_TRUE(delegate->IsPluginTemporarilyAllowed(foo_plugin));
  EXPECT_FALSE(delegate->IsPluginTemporarilyAllowed(bar_plugin));

  // Simulate same document navigation.
  OnSameDocumentNavigation(GetMainFrame(), true);
  EXPECT_TRUE(delegate->IsPluginTemporarilyAllowed(foo_plugin));
  EXPECT_FALSE(delegate->IsPluginTemporarilyAllowed(bar_plugin));

  // Navigate to a different page.
  LoadHTML("<html>Bar</html>");
  EXPECT_FALSE(delegate->IsPluginTemporarilyAllowed(foo_plugin));
  EXPECT_FALSE(delegate->IsPluginTemporarilyAllowed(bar_plugin));

  // Temporarily allow all plugins.
  delegate->AllowPluginTemporarily(std::string());
  EXPECT_TRUE(delegate->IsPluginTemporarilyAllowed(foo_plugin));
  EXPECT_TRUE(delegate->IsPluginTemporarilyAllowed(bar_plugin));
}