package turbo

import (
	"testing"

	logger "github.com/sirupsen/logrus"
	"github.com/stretchr/testify/assert"
	"google.golang.org/grpc"
)

func TestGrpcInit(t *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			assert.Fail(t, "should not panic")
		}
	}()
	s := &GrpcServer{gClient: new(grpcClient)}
	s.gClient.grpcServiceMap = make(map[string]interface{})
	s.gClient.init("", func(*grpc.ClientConn) map[string]interface{} { return nil })
}

func TestGrpcClose(t *testing.T) {
	s := &GrpcServer{gClient: new(grpcClient)}
	err := s.gClient.close()
	assert.Nil(t, err)
}

func TestGrpcService(t *testing.T) {
	defer func() {
		if err := recover(); err != nil {
			assert.Equal(t, "grpc connection not initiated!", err.(*logger.Entry).Message)
		} else {
			t.Errorf("The code did not panic")
		}
	}()
	s := &GrpcServer{gClient: new(grpcClient)}
	s.Service("")
}