connect/repositories/session_single_repository_test.go

package repositories_test

import (
	"context"
	"go.uber.org/zap"
	"sikey.com/websocket/models"
	"sikey.com/websocket/pkg/confx"
	"sikey.com/websocket/pkg/dbx"
	"sikey.com/websocket/pkg/gid"
	"sikey.com/websocket/repositories"
	"testing"
)

func TestSessionSingleRepository_Create(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	gid.SetNodeId(0)
	repository := repositories.NewSessionSingleRepository(source)
	err := repository.Create(context.Background(), &models.SessionSingle{
		ID:           "SINGLE_" + gid.GetSnowflakeId(),
		TargetUserID: "f0e3d3d0-38fc-4365-86fd-5d97bfd49bfd",
		ToUserID:     "dce34294-467d-4fcb-a550-b729f7167f69",
	})
	if err != nil {
		zap.S().Error(err)
		return
	}
	zap.S().Info("success")
}

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

func TestSessionSingleRepository_GetByJoin(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	repository := repositories.NewSessionSingleRepository(source)
	singles, _ := repository.GetByJoin(context.Background(), "f28ea995-fc67-4ca4-bfdd-235d08cf6167")
	zap.S().Info("success", zap.Any("singles", singles))
}

func TestSessionSingleRepository_Update(t *testing.T) {
	confx.LoadConfig("../etc/servers.debug.yaml")
	source := dbx.GetConnect()
	gid.SetNodeId(0)
	repository := repositories.NewSessionSingleRepository(source)
	single, _ := repository.Get(context.Background(), "SINGLE_1777608344781979649")
	single.ToUserID = "f28ea995-fc67-4ca4-bfdd-235d08cf6167"
	_ = repository.Update(context.Background(), single)
}