starter/nats_connect/nats_connect_test.go

package nats_connect

import (
	"context"
	"github.com/ManyakRus/starter/config_main"
	"github.com/ManyakRus/starter/constants"
	"github.com/ManyakRus/starter/contextmain"
	"github.com/ManyakRus/starter/micro"
	"testing"
	"time"

	//"github.com/ManyakRus/starter/common/v0/logger"
	"github.com/ManyakRus/starter/stopapp"
)

func TestConnect_err(t *testing.T) {
	//ProgramDir := micro.ProgramDir_Common()
	config_main.LoadEnv()
	err := Connect_err(constants.SERVICE_NAME + "_test")
	if err != nil {
		t.Error("nats_connect.TestConnect_err() error: ", err)
	}
	CloseConnection()
}

func TestCloseConnection(t *testing.T) {
	//ProgramDir := micro.ProgramDir_Common()
	config_main.LoadEnv()
	Connect(constants.SERVICE_NAME + "_test")
	CloseConnection()
}

func TestStartNats(t *testing.T) {
	//ProgramDir := micro.ProgramDir_Common()
	config_main.LoadEnv()
	StartNats(constants.SERVICE_NAME + "_test")
	micro.Pause(20)

	_ = contextmain.GetContext()
	contextmain.CancelContext()

	stopapp.GetWaitGroup_Main().Wait()

	contextmain.GetNewContext()
}

func TestWaitStop(t *testing.T) {

	stopapp.GetWaitGroup_Main().Add(1)
	go WaitStop()

	_ = contextmain.GetContext()
	contextmain.CancelContext()

	stopapp.GetWaitGroup_Main().Wait()

	contextmain.GetNewContext()
}

func TestConnect(t *testing.T) {
	config_main.LoadEnv()
	Connect(constants.SERVICE_NAME + "_test")
	defer CloseConnection()
}

func TestSendMessageCtx(t *testing.T) {

	config_main.LoadEnv()
	Connect(constants.SERVICE_NAME + "_test")
	defer CloseConnection()

	ctxMain := contextmain.GetContext()
	ctx, cancel := context.WithTimeout(ctxMain, 60*time.Second)
	defer cancel()

	Subject := "TESTING"
	Data := []byte("testing")
	err := SendMessageCtx(ctx, Subject, Data)
	if err != nil {
		t.Error("TestSendMessageCtx() error: ", err)
	}

}