| // Copyright (c) 2012 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 "base/message_loop/message_loop.h" |
| #include "chrome/common/extensions/extension.h" |
| #include "chrome/common/extensions/features/feature.h" |
| #include "chrome/renderer/extensions/chrome_v8_context.h" |
| #include "chrome/renderer/extensions/chrome_v8_context_set.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "third_party/WebKit/public/web/WebFrame.h" |
| #include "v8/include/v8.h" |
| |
| namespace extensions { |
| |
| TEST(ChromeV8ContextSet, Lifecycle) { |
| base::MessageLoop loop; |
| |
| ChromeV8ContextSet context_set; |
| |
| v8::HandleScope handle_scope; |
| v8::Handle<v8::Context> v8_context( |
| v8::Context::New(v8::Isolate::GetCurrent())); |
| |
| // Dirty hack, but we don't actually need the frame, and this is easier than |
| // creating a whole webview. |
| WebKit::WebFrame* frame = reinterpret_cast<WebKit::WebFrame*>(1); |
| const Extension* extension = NULL; |
| ChromeV8Context* context = new ChromeV8Context( |
| v8_context, |
| frame, |
| extension, |
| Feature::BLESSED_EXTENSION_CONTEXT); |
| |
| context_set.Add(context); |
| EXPECT_EQ(1u, context_set.GetAll().count(context)); |
| EXPECT_EQ(context, context_set.GetByV8Context(context->v8_context())); |
| |
| // Adding the same item multiple times should be OK and deduped. |
| context_set.Add(context); |
| EXPECT_EQ(1u, context_set.GetAll().count(context)); |
| |
| // GetAll() returns a copy so removing from one should not remove from others. |
| ChromeV8ContextSet::ContextSet set_copy = context_set.GetAll(); |
| EXPECT_EQ(1u, set_copy.count(context)); |
| |
| context_set.Remove(context); |
| EXPECT_EQ(0, context_set.size()); |
| EXPECT_FALSE(context_set.GetByV8Context(context->v8_context())); |
| EXPECT_EQ(1u, set_copy.size()); |
| |
| // After removal, the context should be marked for destruction. |
| EXPECT_FALSE(context->web_frame()); |
| |
| // Run loop to do the actual deletion. |
| loop.RunUntilIdle(); |
| } |
| |
| } // namespace extensions |