improve robustness

apache · Oct 29, 2024 · d00a8df · d00a8df
1 parent b77346b
commit d00a8df
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/...erver/hugegraph-test/src/main/java/org/apache/hugegraph/core/memory/MemoryManageTest.java b/...erver/hugegraph-test/src/main/java/org/apache/hugegraph/core/memory/MemoryManageTest.java
@@ -20,6 +20,7 @@
 import org.apache.hugegraph.memory.MemoryManager;
 import org.apache.hugegraph.memory.pool.MemoryPool;
 import org.apache.hugegraph.memory.pool.impl.TaskMemoryPool;
+import org.apache.hugegraph.memory.util.RoundUtil;
 import org.apache.hugegraph.testutil.Assert;
 import org.apache.hugegraph.util.Bytes;
 import org.junit.After;
@@ -70,6 +71,9 @@ public class MemoryManageTest {
     public void setUp() {
         memoryManager = MemoryManager.getInstance();
         MemoryManager.setMemoryMode(MemoryManager.MemoryMode.ENABLE_OFF_HEAP_MANAGEMENT);
+        MemoryManager.setMaxMemoryCapacityInBytes(Bytes.GB);
+        MemoryManager.setMaxMemoryCapacityForOneQuery(Bytes.MB * 100);
+        RoundUtil.setAlignment(8);
         query1MemoryPool = memoryManager.addQueryMemoryPool();
         query2MemoryPool = memoryManager.addQueryMemoryPool();