package mtinfo

import (
	"testing"

	"github.com/stretchr/testify/assert"
)

func TestNewClient_withGRPCProtocol_returnsGRPCClient(t *testing.T) {
	cfg := Config{
		Auth: AuthConfig{
			PublicKey: "testdata/key_rsa.pub",
			Grpc: GrpcEndpoint{
				Host: "localhost",
				Port: "1234",
			},
		},
	}
	client, err := NewClient(GRPC, cfg)
	assert.Nil(t, err)
	assert.NotNil(t, client)
	assert.NotNil(t, client.Auth)
	assert.IsType(t, client.Auth, &grpcAuthServiceClient{})
}

func TestNewClient_withRESTProtocol_returnsNotYetImplemented(t *testing.T) {
	cfg := Config{}
	_, err := NewClient(REST, cfg)
	assert.EqualError(t, err, "not yet implemented")
}

func TestNewClient_withUnknownProtocol_returnsError(t *testing.T) {
	cfg := Config{}
	_, err := NewClient(Protocol("udp"), cfg)
	assert.EqualError(t, err, "protocol not supported")
}