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 "bindings/v8/IDBBindingUtilities.h"
28
29 #include "bindings/v8/V8Binding.h"
30 #include "bindings/v8/V8PerIsolateData.h"
31 #include "bindings/v8/V8Utilities.h"
32 #include "modules/indexeddb/IDBKey.h"
33 #include "modules/indexeddb/IDBKeyPath.h"
34
35 #include <gtest/gtest.h>
36
37 using namespace WebCore;
38
39 namespace {
40
checkKeyFromValueAndKeyPathInternal(const ScriptValue & value,const String & keyPath)41 PassRefPtr<IDBKey> checkKeyFromValueAndKeyPathInternal(const ScriptValue& value, const String& keyPath)
42 {
43 IDBKeyPath idbKeyPath(keyPath);
44 EXPECT_TRUE(idbKeyPath.isValid());
45
46 return createIDBKeyFromScriptValueAndKeyPath(0, value, idbKeyPath);
47 }
48
checkKeyPathNullValue(const ScriptValue & value,const String & keyPath)49 void checkKeyPathNullValue(const ScriptValue& value, const String& keyPath)
50 {
51 RefPtr<IDBKey> idbKey = checkKeyFromValueAndKeyPathInternal(value, keyPath);
52 ASSERT_FALSE(idbKey.get());
53 }
54
injectKey(PassRefPtr<IDBKey> key,ScriptValue & value,const String & keyPath)55 bool injectKey(PassRefPtr<IDBKey> key, ScriptValue& value, const String& keyPath)
56 {
57 IDBKeyPath idbKeyPath(keyPath);
58 EXPECT_TRUE(idbKeyPath.isValid());
59 ScriptValue keyValue = idbKeyToScriptValue(0, key);
60 return injectV8KeyIntoV8Value(keyValue.v8Value(), value.v8Value(), idbKeyPath, v8::Isolate::GetCurrent());
61 }
62
checkInjection(PassRefPtr<IDBKey> prpKey,ScriptValue & value,const String & keyPath)63 void checkInjection(PassRefPtr<IDBKey> prpKey, ScriptValue& value, const String& keyPath)
64 {
65 RefPtr<IDBKey> key = prpKey;
66 bool result = injectKey(key, value, keyPath);
67 ASSERT_TRUE(result);
68 RefPtr<IDBKey> extractedKey = checkKeyFromValueAndKeyPathInternal(value, keyPath);
69 EXPECT_TRUE(key->isEqual(extractedKey.get()));
70 }
71
checkInjectionFails(PassRefPtr<IDBKey> key,ScriptValue & value,const String & keyPath)72 void checkInjectionFails(PassRefPtr<IDBKey> key, ScriptValue& value, const String& keyPath)
73 {
74 EXPECT_FALSE(injectKey(key, value, keyPath));
75 }
76
checkKeyPathStringValue(const ScriptValue & value,const String & keyPath,const String & expected)77 void checkKeyPathStringValue(const ScriptValue& value, const String& keyPath, const String& expected)
78 {
79 RefPtr<IDBKey> idbKey = checkKeyFromValueAndKeyPathInternal(value, keyPath);
80 ASSERT_TRUE(idbKey.get());
81 ASSERT_EQ(IDBKey::StringType, idbKey->type());
82 ASSERT_TRUE(expected == idbKey->string());
83 }
84
checkKeyPathNumberValue(const ScriptValue & value,const String & keyPath,int expected)85 void checkKeyPathNumberValue(const ScriptValue& value, const String& keyPath, int expected)
86 {
87 RefPtr<IDBKey> idbKey = checkKeyFromValueAndKeyPathInternal(value, keyPath);
88 ASSERT_TRUE(idbKey.get());
89 ASSERT_EQ(IDBKey::NumberType, idbKey->type());
90 ASSERT_TRUE(expected == idbKey->number());
91 }
92
93 class IDBKeyFromValueAndKeyPathTest : public testing::Test {
94 public:
IDBKeyFromValueAndKeyPathTest()95 IDBKeyFromValueAndKeyPathTest()
96 : m_handleScope(v8::Isolate::GetCurrent())
97 , m_scope(v8::Context::New(v8::Isolate::GetCurrent()))
98 {
99 }
100
101 private:
102 v8::HandleScope m_handleScope;
103 v8::Context::Scope m_scope;
104 };
105
TEST_F(IDBKeyFromValueAndKeyPathTest,TopLevelPropertyStringValue)106 TEST_F(IDBKeyFromValueAndKeyPathTest, TopLevelPropertyStringValue)
107 {
108 v8::Local<v8::Object> object = v8::Object::New();
109 object->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "foo"), v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "zoo"));
110
111 ScriptValue scriptValue(object, v8::Isolate::GetCurrent());
112
113 checkKeyPathStringValue(scriptValue, "foo", "zoo");
114 checkKeyPathNullValue(scriptValue, "bar");
115 }
116
TEST_F(IDBKeyFromValueAndKeyPathTest,TopLevelPropertyNumberValue)117 TEST_F(IDBKeyFromValueAndKeyPathTest, TopLevelPropertyNumberValue)
118 {
119 v8::Local<v8::Object> object = v8::Object::New();
120 object->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "foo"), v8::Number::New(456));
121
122 ScriptValue scriptValue(object, v8::Isolate::GetCurrent());
123
124 checkKeyPathNumberValue(scriptValue, "foo", 456);
125 checkKeyPathNullValue(scriptValue, "bar");
126 }
127
TEST_F(IDBKeyFromValueAndKeyPathTest,SubProperty)128 TEST_F(IDBKeyFromValueAndKeyPathTest, SubProperty)
129 {
130 v8::Local<v8::Object> object = v8::Object::New();
131 v8::Local<v8::Object> subProperty = v8::Object::New();
132 subProperty->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "bar"), v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "zee"));
133 object->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "foo"), subProperty);
134
135 ScriptValue scriptValue(object, v8::Isolate::GetCurrent());
136
137 checkKeyPathStringValue(scriptValue, "foo.bar", "zee");
138 checkKeyPathNullValue(scriptValue, "bar");
139 }
140
141 class InjectIDBKeyTest : public IDBKeyFromValueAndKeyPathTest {
142 };
143
TEST_F(InjectIDBKeyTest,TopLevelPropertyStringValue)144 TEST_F(InjectIDBKeyTest, TopLevelPropertyStringValue)
145 {
146 v8::Local<v8::Object> object = v8::Object::New();
147 object->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "foo"), v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "zoo"));
148
149 ScriptValue foozoo(object, v8::Isolate::GetCurrent());
150 checkInjection(IDBKey::createString("myNewKey"), foozoo, "bar");
151 checkInjection(IDBKey::createNumber(1234), foozoo, "bar");
152
153 checkInjectionFails(IDBKey::createString("key"), foozoo, "foo.bar");
154 }
155
TEST_F(InjectIDBKeyTest,SubProperty)156 TEST_F(InjectIDBKeyTest, SubProperty)
157 {
158 v8::Local<v8::Object> object = v8::Object::New();
159 v8::Local<v8::Object> subProperty = v8::Object::New();
160 subProperty->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "bar"), v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "zee"));
161 object->Set(v8::String::NewFromUtf8(v8::Isolate::GetCurrent(), "foo"), subProperty);
162
163 ScriptValue scriptObject(object, v8::Isolate::GetCurrent());
164 checkInjection(IDBKey::createString("myNewKey"), scriptObject, "foo.baz");
165 checkInjection(IDBKey::createNumber(789), scriptObject, "foo.baz");
166 checkInjection(IDBKey::createDate(4567), scriptObject, "foo.baz");
167 checkInjection(IDBKey::createDate(4567), scriptObject, "bar");
168 checkInjection(IDBKey::createArray(IDBKey::KeyArray()), scriptObject, "foo.baz");
169 checkInjection(IDBKey::createArray(IDBKey::KeyArray()), scriptObject, "bar");
170
171 checkInjectionFails(IDBKey::createString("zoo"), scriptObject, "foo.bar.baz");
172 checkInjection(IDBKey::createString("zoo"), scriptObject, "foo.xyz.foo");
173 }
174
175 } // namespace
176