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 <vector>
   6
   7 #include "base/message_loop/message_loop.h"
   8 #include "extensions/common/extension.h"
   9 #include "extensions/common/extension_set.h"
  10 #include "extensions/common/features/feature.h"
  11 #include "extensions/renderer/scoped_web_frame.h"
  12 #include "extensions/renderer/script_context.h"
  13 #include "extensions/renderer/script_context_set.h"
  14 #include "gin/public/context_holder.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16 #include "third_party/WebKit/public/web/WebFrame.h"
  17 #include "v8/include/v8.h"
  18
  19 namespace extensions {
  20
  21 TEST(ScriptContextSetTest, Lifecycle) {
  22   base::MessageLoop loop;
  23   ScopedWebFrame web_frame;
  24
  25   // Do this after construction of the webview, since it may construct the
  26   // Isolate.
  27   v8::Isolate* isolate = v8::Isolate::GetCurrent();
  28
  29   v8::HandleScope handle_scope(isolate);
  30   v8::Local<v8::Context> v8_context = v8::Context::New(isolate);
  31   v8::Context::Scope context_scope(v8_context);
  32   // ScriptContext relies on gin, it just doesn't look like it from here.
  33   gin::ContextHolder context_holder(isolate);
  34   context_holder.SetContext(v8_context);
  35
  36   ExtensionIdSet active_extensions;
  37   ScriptContextSet context_set(&active_extensions);
  38   ScriptContext* context = context_set.Register(
  39       web_frame.frame(), v8_context, 0, 0);  // no extension group or world ID
  40
  41   // Context is valid and resembles correctness.
  42   EXPECT_TRUE(context->is_valid());
  43   EXPECT_EQ(web_frame.frame(), context->web_frame());
  44   EXPECT_EQ(v8_context, context->v8_context());
  45
  46   // Context has been correctly added.
  47   EXPECT_EQ(1u, context_set.size());
  48   EXPECT_EQ(context, context_set.GetByV8Context(v8_context));
  49
  50   // Test context is correctly removed.
  51   context_set.Remove(context);
  52   EXPECT_EQ(0u, context_set.size());
  53   EXPECT_EQ(nullptr, context_set.GetByV8Context(v8_context));
  54
  55   // After removal, the context should be marked for destruction.
  56   EXPECT_FALSE(context->is_valid());
  57
  58   // Run loop to do the actual deletion.
  59   loop.RunUntilIdle();
  60 }
  61
  62 }  // namespace extensions