tests: Set JSON tl heap for tests

Signed-off-by: Abhijat Malviya <[email protected]>

Author: abhijat

src/core/compact_object_test.cc

@@ -88,6 +88,7 @@ static void InitThreadStructs() {
   SmallString::InitThreadLocal(tlh);
   thread_local MiMemoryResource mi_resource(tlh);
   CompactObj::InitThreadLocal(&mi_resource);
+  InitJSONTLHeap(&mi_resource);
 };
 
 static void CheckEverythingDeallocated() {

src/core/json/jsonpath_test.cc

@@ -91,6 +91,7 @@ class ScannerTest : public ::testing::Test {
  protected:
   ScannerTest() {
     driver_.lexer()->set_debug(1);
+    InitJSONTLHeap(PMR_NS::get_default_resource());
   }
 
   void SetInput(const std::string& str) {

src/core/page_usage_stats_test.cc

@@ -45,6 +45,7 @@ class PageUsageStatsTest : public ::testing::Test {
   }
 
   PageUsageStatsTest() : m_(mi_heap_get_backing()) {
+    InitJSONTLHeap(&m_);
   }
 
   void SetUp() override {

src/server/json_family_memory_test.cc

@@ -6,7 +6,6 @@
 #include "base/logging.h"
 #include "facade/facade_test.h"
 #include "server/command_registry.h"
-#include "server/json_family.h"
 #include "server/test_utils.h"
 
 using namespace testing;
@@ -19,25 +18,30 @@ namespace dfly {
 
 class JsonFamilyMemoryTest : public BaseFamilyTest {
  public:
-  static dfly::MiMemoryResource* GetMemoryResource() {
-    static thread_local mi_heap_t* heap = mi_heap_new();
-    static thread_local dfly::MiMemoryResource memory_resource{heap};
+  static MiMemoryResource* GetMemoryResource() {
+    thread_local mi_heap_t* heap = mi_heap_new();
+    thread_local MiMemoryResource memory_resource{heap};
     return &memory_resource;
   }
 
  protected:
+  void SetUp() override {
+    BaseFamilyTest::SetUp();
+    InitJSONTLHeap(GetMemoryResource());
+  }
+
   auto GetJsonMemoryUsageFromDb(std::string_view key) {
     return Run({"MEMORY", "USAGE", key, "WITHOUTKEY"});
   }
 };
 
 size_t GetMemoryUsage() {
-  return static_cast<MiMemoryResource*>(JsonFamilyMemoryTest::GetMemoryResource())->used();
+  return JsonFamilyMemoryTest::GetMemoryResource()->used();
 }
 
 size_t GetJsonMemoryUsageFromString(std::string_view json_str) {
   size_t start = GetMemoryUsage();
-  auto json = dfly::JsonFromString(json_str);
+  auto json = JsonFromString(json_str);
   if (!json) {
     return 0;
   }
@@ -111,7 +115,7 @@ TEST_F(JsonFamilyMemoryTest, JsonConsDelTest) {
 
   size_t start = GetMemoryUsage();
 
-  auto json = dfly::JsonFromString(start_json);
+  auto json = JsonFromString(start_json);
   void* ptr =
       JsonFamilyMemoryTest::GetMemoryResource()->allocate(sizeof(JsonType), alignof(JsonType));
   JsonType* json_on_heap = new (ptr) JsonType(std::move(json).value());