From f935aebfb7646dd7d5a6518add8271cdc5828628 Mon Sep 17 00:00:00 2001
From: godotg <jaysunxiao@gmail.com>
Date: Wed, 3 Jul 2024 10:49:13 +0800
Subject: [PATCH] test[orm]: test

---
 .../java/com/zfoo/orm/accessor/AccessorTest.java    |  5 ++---
 .../java/com/zfoo/orm/accessor/DocumentTest.java    |  2 +-
 .../java/com/zfoo/orm/cache/EntityCachesTest.java   | 13 +++++++------
 3 files changed, 10 insertions(+), 10 deletions(-)

diff --git a/orm/src/test/java/com/zfoo/orm/accessor/AccessorTest.java b/orm/src/test/java/com/zfoo/orm/accessor/AccessorTest.java
index 131efdd3..815a7edc 100644
--- a/orm/src/test/java/com/zfoo/orm/accessor/AccessorTest.java
+++ b/orm/src/test/java/com/zfoo/orm/accessor/AccessorTest.java
@@ -20,7 +20,6 @@ import org.junit.Test;
 import org.springframework.context.support.ClassPathXmlApplicationContext;
 
 import java.util.ArrayList;
-import java.util.Collections;
 
 /**
  * @author godotg
@@ -31,7 +30,7 @@ public class AccessorTest {
     @Test
     public void oneTest() {
         var context = new ClassPathXmlApplicationContext("application.xml");
-        OrmContext.getOrmManager().getCollection(UserEntity.class).drop();
+        delete();
         insert();
         delete();
         insert();
@@ -42,7 +41,7 @@ public class AccessorTest {
     @Test
     public void batchTest() {
         var context = new ClassPathXmlApplicationContext("application.xml");
-        OrmContext.getOrmManager().getCollection(UserEntity.class).drop();
+        deleteAll();
         batchInsert();
         batchUpdate();
         queryAll();
diff --git a/orm/src/test/java/com/zfoo/orm/accessor/DocumentTest.java b/orm/src/test/java/com/zfoo/orm/accessor/DocumentTest.java
index 1a475c64..d12e06a3 100644
--- a/orm/src/test/java/com/zfoo/orm/accessor/DocumentTest.java
+++ b/orm/src/test/java/com/zfoo/orm/accessor/DocumentTest.java
@@ -31,7 +31,7 @@ public class DocumentTest {
     @Test
     public void collectionTest() {
         var context = new ClassPathXmlApplicationContext("application.xml");
-
+        OrmContext.getAccessor().delete(1L, UserEntity.class);
         var collection = OrmContext.getOrmManager().getCollection(UserEntity.class);
         var result = collection.updateOne(Filters.eq("_id", 1), new Document("$inc", new Document("c", 1L)));
         System.out.println(result);
diff --git a/orm/src/test/java/com/zfoo/orm/cache/EntityCachesTest.java b/orm/src/test/java/com/zfoo/orm/cache/EntityCachesTest.java
index d44514d4..5a37b0aa 100644
--- a/orm/src/test/java/com/zfoo/orm/cache/EntityCachesTest.java
+++ b/orm/src/test/java/com/zfoo/orm/cache/EntityCachesTest.java
@@ -36,9 +36,7 @@ public class EntityCachesTest {
     public void loadTest() {
         var context = new ClassPathXmlApplicationContext("application.xml");
 
-        // 每次运行前先删除数据库
-        var collection = OrmContext.getOrmManager().getCollection(UserEntity.class);
-        collection.drop();
+        batchDelete();
 
         // 再插入
         batchInsert();
@@ -65,9 +63,7 @@ public class EntityCachesTest {
     public void loadOrCreateTest() {
         var context = new ClassPathXmlApplicationContext("application.xml");
 
-        // 每次运行前先删除数据库
-        var collection = OrmContext.getOrmManager().getCollection(UserEntity.class);
-        collection.drop();
+        batchDelete();
 
         @SuppressWarnings("unchecked")
         var userEntityCaches = (IEntityCache<Long, UserEntity>) OrmContext.getOrmManager().getEntityCaches(UserEntity.class);
@@ -95,4 +91,9 @@ public class EntityCachesTest {
         OrmContext.getAccessor().batchInsert(listUser);
     }
 
+    public void batchDelete() {
+        for (var i = 1; i <= 10; i++) {
+            OrmContext.getAccessor().delete((long) i, UserEntity.class);
+        }
+    }
 }