Add TestDeleteEmptyChats()

JiawenZou · Mar 3, 2024 · d434386 · d434386
1 parent 26c7c0e
commit d434386
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/object/chat_test.go b/object/chat_test.go
@@ -18,6 +18,7 @@
 package object
 
 import (
+	"fmt"
 	"testing"
 
 	"github.com/casibase/casibase/model"
@@ -95,6 +96,26 @@ func TestUpdateMessageTokens(t *testing.T) {
 	}
 }
 
+func TestDeleteEmptyChats(t *testing.T) {
+	InitConfig()
+
+	chats, err := GetGlobalChats()
+	if err != nil {
+		panic(err)
+	}
+
+	for i, chat := range chats {
+		if chat.MessageCount == 0 {
+			_, err = DeleteChat(chat)
+			if err != nil {
+				panic(err)
+			}
+
+			fmt.Printf("[%d/%d] deleted chat: %s, user: %s, clientIP: %s (%s), userAgent: %s (%s)\n", i+1, len(chats), chat.Name, chat.User, chat.ClientIpDesc, chat.ClientIp, chat.UserAgentDesc, chat.UserAgent)
+		}
+	}
+}
+
 func TestUpdateChatDescs(t *testing.T) {
 	InitConfig()
 	util.InitIpDb()

diff --git a/object/message_test.go b/object/message_test.go
@@ -124,3 +124,9 @@ func TestUpdateMessages(t *testing.T) {
 		}
 	}
 }
+
+func TestUpdateMessagesAndChats(t *testing.T) {
+	TestUpdateMessages(t)
+	TestUpdateMessageCounts(t)
+	TestUpdateMessageTokens(t)
+}