src/third_party/WebKit/Source/core/testing/v8/WebCoreTestSupport.cpp

   1 /*
   2  * Copyright (C) 2011 Google Inc. All rights reserved.
   3  *
   4  * Redistribution and use in source and binary forms, with or without
   5  * modification, are permitted provided that the following conditions
   6  * are met:
   7  *
   8  * 1.  Redistributions of source code must retain the above copyright
   9  *     notice, this list of conditions and the following disclaimer.
  10  * 2.  Redistributions in binary form must reproduce the above copyright
  11  *     notice, this list of conditions and the following disclaimer in the
  12  *     documentation and/or other materials provided with the distribution.
  13  *
  14  * THIS SOFTWARE IS PROVIDED BY APPLE AND ITS CONTRIBUTORS "AS IS" AND ANY
  15  * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
  16  * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
  17  * DISCLAIMED. IN NO EVENT SHALL APPLE OR ITS CONTRIBUTORS BE LIABLE FOR ANY
  18  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
  19  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  20  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
  21  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  22  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  23  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  24  */
  25
  26 #include "config.h"
  27 #include "core/testing/v8/WebCoreTestSupport.h"
  28
  29 #include "bindings/core/v8/V8Internals.h"
  30 #include "core/dom/Document.h"
  31 #include "core/dom/ExecutionContext.h"
  32 #include "core/frame/LocalFrame.h"
  33 #include "core/testing/InternalSettings.h"
  34 #include "core/testing/Internals.h"
  35
  36 using namespace blink;
  37
  38 namespace WebCoreTestSupport {
  39
  40 void injectInternalsObject(v8::Local<v8::Context> context)
  41 {
  42     ScriptState* scriptState = ScriptState::from(context);
  43     ScriptState::Scope scope(scriptState);
  44     v8::Handle<v8::Object> global = scriptState->context()->Global();
  45     ExecutionContext* executionContext = scriptState->executionContext();
  46     if (executionContext->isDocument())
  47         global->Set(v8::String::NewFromUtf8(scriptState->isolate(), Internals::internalsId), toV8(Internals::create(toDocument(executionContext)), global, scriptState->isolate()));
  48 }
  49
  50 void resetInternalsObject(v8::Local<v8::Context> context)
  51 {
  52     // This can happen if JavaScript is disabled in the main frame.
  53     if (context.IsEmpty())
  54         return;
  55
  56     ScriptState* scriptState = ScriptState::from(context);
  57     ScriptState::Scope scope(scriptState);
  58     Page* page = toDocument(scriptState->executionContext())->frame()->page();
  59     ASSERT(page);
  60     Internals::resetToConsistentState(page);
  61     InternalSettings::from(*page)->resetToConsistentState();
  62 }
  63
  64 }