feat: Added unit test for DeleteGroupMemberHash me

pull/1199/head
sweep-ai[bot] 2 years ago committed by GitHub
parent 6b7433352a
commit 9e4b3707ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -0,0 +1,42 @@
package controller_test
import (
"context"
"testing"
"github.com/OpenIMSDK/open-im-server/v3/pkg/common/db/controller"
"github.com/OpenIMSDK/open-im-server/v3/pkg/common/db/relation"
"github.com/OpenIMSDK/open-im-server/v3/pkg/common/db/table/relationtb"
"github.com/dtm-labs/rockscache"
"github.com/redis/go-redis/v9"
"go.mongodb.org/mongo-driver/mongo"
"gorm.io/gorm"
)
func TestDeleteGroupMemberHash(t *testing.T) {
db := gorm.DB{}
rdb := redis.UniversalClient{}
database := mongo.Database{}
hashCode := func(ctx context.Context, groupID string) (uint64, error) { return 0, nil }
groupDatabase := controller.InitGroupDatabase(&db, rdb, &database, hashCode)
group := &relationtb.GroupModel{GroupID: "testGroupID"}
member := &relationtb.GroupMemberModel{GroupID: "testGroupID", UserID: "testUserID"}
groupDatabase.CreateGroup(context.Background(), []*relationtb.GroupModel{group}, []*relationtb.GroupMemberModel{member})
err := groupDatabase.DeleteGroupMemberHash(context.Background(), "testGroupID", "testUserID")
if err != nil {
t.Errorf("Failed to delete group member hash: %v", err)
}
members, err := groupDatabase.FindGroupMember(context.Background(), []string{"testGroupID"}, []string{"testUserID"}, nil)
if err != nil {
t.Errorf("Failed to find group member: %v", err)
}
if len(members) != 0 {
t.Errorf("Group member hash was not deleted")
}
}
Loading…
Cancel
Save