1 files changed, 44 insertions, 0 deletions
diff --git a/js/src/jsapi-tests/testMemoryAssociation.cpp b/js/src/jsapi-tests/testMemoryAssociation.cpp
new file mode 100644
index 0000000000..fb57e54715
--- /dev/null
+++ b/js/src/jsapi-tests/testMemoryAssociation.cpp
@@ -0,0 +1,44 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*-
+ * vim: set ts=8 sts=2 et sw=2 tw=80:
+ */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include "jsapi.h"
+#include "jspubtd.h"
+
+#include "js/CompilationAndEvaluation.h"
+#include "jsapi-tests/tests.h"
+
+static const JS::MemoryUse TestUse1 = JS::MemoryUse::XPCWrappedNative;
+static const JS::MemoryUse TestUse2 = JS::MemoryUse::DOMBinding;
+
+BEGIN_TEST(testMemoryAssociation) {
+  JS::RootedObject obj(cx, JS_NewPlainObject(cx));
+  CHECK(obj);
+
+  // No association is allowed for nursery objects.
+  JS_GC(cx);
+
+  // Test object/memory association.
+  JS::AddAssociatedMemory(obj, 100, TestUse1);
+  JS::RemoveAssociatedMemory(obj, 100, TestUse1);
+
+  // Test association when object moves due to compacting GC.
+  void* initial = obj;
+  JS::AddAssociatedMemory(obj, 300, TestUse1);
+  JS::PrepareForFullGC(cx);
+  JS::NonIncrementalGC(cx, GC_SHRINK, JS::GCReason::DEBUG_GC);
+  CHECK(obj != initial);
+  JS::RemoveAssociatedMemory(obj, 300, TestUse1);
+
+  // Test multiple associations.
+  JS::AddAssociatedMemory(obj, 400, TestUse1);
+  JS::AddAssociatedMemory(obj, 500, TestUse2);
+  JS::RemoveAssociatedMemory(obj, 400, TestUse1);
+  JS::RemoveAssociatedMemory(obj, 500, TestUse2);
+
+  return true;
+}
+END_TEST(testMemoryAssociation)