package parser

import (
	"sort"
	"testing"

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

func TestDefaultProtoParse(t *testing.T) {
	p := NewDefaultProtoParser()
	data, err := p.Parse("./test.proto")
	assert.Nil(t, err)
	assert.Equal(t, "base.proto", func() string {
		ip := data.Import[0]
		return ip.Filename
	}())
	assert.Equal(t, "test", data.Package.Name)
	assert.Equal(t, true, data.GoPackage == "go")
	assert.Equal(t, true, data.PbPackage == "_go")
	assert.Equal(t, []string{"Inline", "Inner", "TestMessage", "TestReply", "TestReq"},
		func() []string {
			var list []string
			for _, item := range data.Message {
				list = append(list, item.Name)
			}
			sort.Strings(list)
			return list
		}())

	assert.Equal(t, true, func() bool {
		if len(data.Service) != 1 {
			return false
		}

		s := data.Service[0]
		if s.Name != "TestService" {
			return false
		}
		rpcOne := s.RPC[0]

		return rpcOne.Name == "TestRpcOne" && rpcOne.RequestType == "TestReq" && rpcOne.ReturnsType == "TestReply"
	}())
}

func TestDefaultProtoParseDottedRequestType(t *testing.T) {
	// Dotted types (e.g. "base.Req") are now valid — they refer to messages in
	// imported protos. Parsing should succeed.
	p := NewDefaultProtoParser()
	data, err := p.Parse("./test_invalid_request.proto")
	assert.NoError(t, err)
	assert.Equal(t, "base.Req", data.Service[0].RPC[0].RequestType)
}

func TestDefaultProtoParseDottedResponseType(t *testing.T) {
	// Dotted return types (e.g. "base.Reply") are now valid.
	p := NewDefaultProtoParser()
	data, err := p.Parse("./test_invalid_response.proto")
	assert.NoError(t, err)
	assert.Equal(t, "base.Reply", data.Service[0].RPC[0].ReturnsType)
}

func TestDefaultProtoParseError(t *testing.T) {
	p := NewDefaultProtoParser()
	_, err := p.Parse("./nil.proto")
	assert.NotNil(t, err)
}

func TestDefaultProtoParse_Option(t *testing.T) {
	p := NewDefaultProtoParser()
	data, err := p.Parse("./test_option.proto")
	assert.Nil(t, err)
	assert.Equal(t, "github.com/zeromicro/go-zero", data.GoPackage)
	assert.Equal(t, "go_zero", data.PbPackage)
}

func TestDefaultProtoParse_Option2(t *testing.T) {
	p := NewDefaultProtoParser()
	data, err := p.Parse("./test_option2.proto")
	assert.Nil(t, err)
	assert.Equal(t, "stream", data.GoPackage)
	assert.Equal(t, "stream", data.PbPackage)
}