extensions/renderer/script_context_set_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "base/message_loop/message_loop.h"
   6 #include "extensions/common/extension.h"
   7 #include "extensions/common/features/feature.h"
   8 #include "extensions/renderer/script_context.h"
   9 #include "extensions/renderer/script_context_set.h"
  10 #include "gin/public/context_holder.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "third_party/WebKit/public/web/WebFrame.h"
  13 #include "third_party/WebKit/public/web/WebLocalFrame.h"
  14 #include "third_party/WebKit/public/web/WebView.h"
  15 #include "v8/include/v8.h"
  16
  17 namespace extensions {
  18
  19 TEST(ScriptContextSet, Lifecycle) {
  20   base::MessageLoop loop;
  21
  22   ScriptContextSet context_set;
  23
  24   v8::Isolate* isolate = v8::Isolate::GetCurrent();
  25   v8::HandleScope handle_scope(isolate);
  26   gin::ContextHolder context_holder(isolate);
  27   context_holder.SetContext(v8::Context::New(isolate));
  28
  29   blink::WebView* webview = blink::WebView::create(nullptr);
  30   blink::WebFrame* frame = blink::WebLocalFrame::create(nullptr);
  31   webview->setMainFrame(frame);
  32   const Extension* extension = NULL;
  33   ScriptContext* context =
  34       new ScriptContext(context_holder.context(),
  35                         frame,
  36                         extension,
  37                         Feature::BLESSED_EXTENSION_CONTEXT,
  38                         extension,
  39                         Feature::BLESSED_EXTENSION_CONTEXT);
  40
  41   context_set.Add(context);
  42   EXPECT_EQ(1u, context_set.GetAll().count(context));
  43   EXPECT_EQ(context, context_set.GetByV8Context(context->v8_context()));
  44
  45   // Adding the same item multiple times should be OK and deduped.
  46   context_set.Add(context);
  47   EXPECT_EQ(1u, context_set.GetAll().count(context));
  48
  49   // GetAll() returns a copy so removing from one should not remove from others.
  50   ScriptContextSet::ContextSet set_copy = context_set.GetAll();
  51   EXPECT_EQ(1u, set_copy.count(context));
  52
  53   context_set.Remove(context);
  54   EXPECT_EQ(0, context_set.size());
  55   EXPECT_FALSE(context_set.GetByV8Context(context->v8_context()));
  56   EXPECT_EQ(1u, set_copy.size());
  57
  58   // After removal, the context should be marked for destruction.
  59   EXPECT_FALSE(context->web_frame());
  60
  61   // Run loop to do the actual deletion.
  62   loop.RunUntilIdle();
  63 }
  64
  65 }  // namespace extensions