summaryrefslogtreecommitdiff
path: root/chromium/third_party/blink/renderer/platform/testing/scoped_fake_plugin_registry.cc
blob: 96fc9c4089134a5ed9b24742b27a7f6f424ee801 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
// Copyright 2018 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 "third_party/blink/renderer/platform/testing/scoped_fake_plugin_registry.h"

#include "base/files/file_path.h"
#include "mojo/public/cpp/bindings/binding_set.h"
#include "services/service_manager/public/cpp/connector.h"
#include "third_party/blink/public/mojom/plugins/plugin_registry.mojom-blink.h"
#include "third_party/blink/public/platform/platform.h"
#include "third_party/blink/renderer/platform/wtf/functional.h"
#include "third_party/blink/renderer/platform/wtf/std_lib_extras.h"
#include "third_party/skia/include/core/SkColor.h"

namespace blink {

namespace {

class FakePluginRegistryImpl : public mojom::blink::PluginRegistry {
 public:
  static void Bind(mojo::ScopedMessagePipeHandle handle) {
    DEFINE_STATIC_LOCAL(FakePluginRegistryImpl, impl, ());
    impl.bindings_.AddBinding(
        &impl, mojom::blink::PluginRegistryRequest(std::move(handle)));
  }

  // PluginRegistry
  void GetPlugins(bool refresh,
                  const scoped_refptr<const SecurityOrigin>& origin,
                  GetPluginsCallback callback) override {
    auto mime = mojom::blink::PluginMimeType::New();
    mime->mime_type = "application/pdf";
    mime->description = "pdf";

    auto plugin = mojom::blink::PluginInfo::New();
    plugin->name = "pdf";
    plugin->description = "pdf";
    plugin->filename = base::FilePath(FILE_PATH_LITERAL("pdf-files"));
    plugin->background_color = SkColorSetRGB(38, 38, 38);
    plugin->mime_types.push_back(std::move(mime));

    Vector<mojom::blink::PluginInfoPtr> plugins;
    plugins.push_back(std::move(plugin));
    std::move(callback).Run(std::move(plugins));
  }

 private:
  mojo::BindingSet<PluginRegistry> bindings_;
};

}  // namespace

ScopedFakePluginRegistry::ScopedFakePluginRegistry() {
  const char* interface_name = mojom::blink::PluginRegistry::Name_;
  service_manager::Connector* connector = Platform::Current()->GetConnector();
  auto browser_service_filter = service_manager::ServiceFilter::ByName(
      Platform::Current()->GetBrowserServiceName());
  DCHECK(!connector->HasBinderOverrideForTesting(browser_service_filter,
                                                 interface_name));
  connector->OverrideBinderForTesting(
      browser_service_filter, interface_name,
      WTF::BindRepeating(&FakePluginRegistryImpl::Bind));
}

ScopedFakePluginRegistry::~ScopedFakePluginRegistry() {
  Platform::Current()->GetConnector()->ClearBinderOverrideForTesting(
      service_manager::ServiceFilter::ByName(
          Platform::Current()->GetBrowserServiceName()),
      mojom::blink::PluginRegistry::Name_);
}

}  // namespace blink