Bug 798678 - Weakmap key preservation test (r=mccr8)

This commit is contained in:
Bill McCloskey 2012-10-17 18:22:54 -07:00
parent 0a9fd297da
commit 17957fa850
3 changed files with 90 additions and 1 deletions

View File

@ -271,6 +271,7 @@ WeakMap_set(JSContext *cx, unsigned argc, Value *vp)
JS_FRIEND_API(JSBool)
JS_NondeterministicGetWeakMapKeys(JSContext *cx, JSObject *obj, JSObject **ret)
{
obj = UnwrapObject(obj);
if (!obj || !obj->isWeakMap()) {
*ret = NULL;
return true;
@ -281,7 +282,10 @@ JS_NondeterministicGetWeakMapKeys(JSContext *cx, JSObject *obj, JSObject **ret)
ObjectValueMap *map = GetObjectMap(obj);
if (map) {
for (ObjectValueMap::Base::Range r = map->all(); !r.empty(); r.popFront()) {
if (!js_NewbornArrayPush(cx, arr, ObjectValue(*r.front().key)))
RootedObject key(cx, r.front().key);
if (!JS_WrapObject(cx, key.address()))
return false;
if (!js_NewbornArrayPush(cx, arr, ObjectValue(*key)))
return false;
}
}

View File

@ -63,6 +63,7 @@ MOCHITEST_CHROME_FILES = \
test_sandboxImport.xul \
test_weakmaps.xul \
test_weakmap_keys_preserved.xul \
test_weakmap_keys_preserved2.xul \
test_weakref.xul \
test_wrappers.xul \
$(NULL)

View File

@ -0,0 +1,84 @@
<?xml version="1.0"?>
<?xml-stylesheet type="text/css" href="chrome://global/skin"?>
<?xml-stylesheet type="text/css" href="chrome://mochikit/content/tests/SimpleTest/test.css"?>
<!--
https://bugzilla.mozilla.org/show_bug.cgi?id=673468
-->
<window title="Mozilla Bug "
xmlns="http://www.mozilla.org/keymaster/gatekeeper/there.is.only.xul">
<script type="application/javascript" src="chrome://mochikit/content/tests/SimpleTest/SimpleTest.js"/>
<!-- test results are displayed in the html:body -->
<body xmlns="http://www.w3.org/1999/xhtml">
<a id="testelem" href="https://bugzilla.mozilla.org/show_bug.cgi?id="
target="_blank">Mozilla Bug 673468</a>
</body>
<!-- test code goes here -->
<script type="application/javascript">
<![CDATA[
/** Test for Bug 673468 **/
let Cc = Components.classes;
let Cu = Components.utils;
let Ci = Components.interfaces;
SpecialPowers.DOMWindowUtils.garbageCollect();
let get_live_dom = function () {
return document.getElementById("testelem");
};
let wrappers_as_keys_test = function () {
let e = document.createEvent("MessageEvent");
e.initMessageEvent("foo", false, false, { dummy: document.createElement("foo") }, null, null, null);
window.eeeevent = e;
let live_dom = e.data.dummy;
let dead_dom = document.createElement("div");
let dead_child = document.createElement("div");
dead_dom.appendChild(dead_child);
is(dead_dom.children.length, 1, "children have wrong length");
let wrappee = new Object();
dead_dom.abcxyz = wrappee;
let system = Cc["@mozilla.org/systemprincipal;1"].createInstance();
let sandbox = Cu.Sandbox(system);
sandbox.wrapper = wrappee;
sandbox.value = dead_dom;
let map = Cu.evalInSandbox("wm = new WeakMap(); wm.set(wrapper, value); wm", sandbox);
sandbox.wrapper = null;
sandbox.value = null;
live_dom.xyzabc = {wrappee: wrappee, m: map, sb: sandbox};
let key = Cu.nondeterministicGetWeakMapKeys(map)[0];
let value = map.get(key);
is(value.children.length, 1, "children have wrong length");
}
wrappers_as_keys_test();
let check_wrappers_as_keys = function () {
let live_dom = window.eeeevent.data.dummy;
let live_map = live_dom.xyzabc.m;
let sandbox = live_dom.xyzabc.sb;
is(Cu.nondeterministicGetWeakMapKeys(live_map).length, 1,
"Map should not be empty.");
let key = Cu.nondeterministicGetWeakMapKeys(live_map)[0];
let value = live_map.get(key);
is(value.children.length, 1, "children have wrong length");
}
Cu.schedulePreciseGC(function () {
SpecialPowers.DOMWindowUtils.cycleCollect();
SpecialPowers.DOMWindowUtils.garbageCollect();
check_wrappers_as_keys();
});
]]>
</script>
</window>