/* -*- 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)