connect/repositories/session_group_repository_test.go

package repositories_test

import (
	"context"
	"fmt"
	"sikey.com/websocket/pkg/gid"
	"testing"

	"go.uber.org/zap"
	"sikey.com/websocket/models"
	"sikey.com/websocket/pkg/confx"
	"sikey.com/websocket/pkg/dbx"
	"sikey.com/websocket/repositories"
)

func TestSessionGroupRepository_Create(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	gid.SetNodeId(1)
	repository := repositories.NewSessionGroupRepository(source)
	group := &models.SessionGroup{
		ID:      "GROUP_" + gid.GetSnowflakeId(),
		ChildId: "f0e3d3d0-38fc-4365-86fd-5d97bfd49bfd",
		Members: models.NewMembers([]models.Member2{
			{UserId: "dce34294-467d-4fcb-a550-b729f7167f69", Identity: 4},
			{UserId: "f0e3d3d0-38fc-4365-86fd-5d97bfd49bfd", Identity: 0},
		}),
	}

	err := repository.Create(context.Background(), group)
	if err != nil {
		zap.S().Error(err)
		return
	}
	zap.S().Info("success")
}

func TestSessionGroupRepository_Get(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	repository := repositories.NewSessionGroupRepository(source)
	group, _ := repository.Get(context.Background(), "ID")
	members := group.Members.ToSlice()
	fmt.Println(members)
}

func TestSessionGroupRepository_Delete(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	repository := repositories.NewSessionGroupRepository(source)
	err := repository.Delete(context.Background(), "ID")
	if err != nil {
		zap.S().Error(err)
		return
	}
	zap.S().Info("success")
}

func TestSessionGroupRepository_Update(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	repository := repositories.NewSessionGroupRepository(source)
	group, _ := repository.Get(context.Background(), "ID")
	group.ChildId = group.ChildId + "_1"
	members := group.Members.ToSlice()
	for i := range len(members) {
		members[i].UserId = members[i].UserId + "_"
	}
	group.Members = models.NewMembers(members)
	_ = repository.Update(context.Background(), group)
}