parent
6b7433352a
commit
9e4b3707ef
@ -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…
Reference in new issue