wakapi/services/alias_test.go

package services

import (
	"github.com/muety/wakapi/mocks"
	"github.com/muety/wakapi/models"
	"github.com/stretchr/testify/assert"
	"github.com/stretchr/testify/mock"
	"github.com/stretchr/testify/suite"
	"testing"
)

type AliasServiceTestSuite struct {
	suite.Suite
	TestUserId      string
	AliasRepository *mocks.AliasRepositoryMock
}

func (suite *AliasServiceTestSuite) SetupSuite() {
	suite.TestUserId = "johndoe@example.org"

	aliases := []*models.Alias{
		{
			Type:   models.SummaryProject,
			UserID: suite.TestUserId,
			Key:    "wakapi",
			Value:  "wakapi-mobile",
		},
	}

	aliasRepoMock := new(mocks.AliasRepositoryMock)
	aliasRepoMock.On("GetByUser", suite.TestUserId).Return(aliases, nil)
	aliasRepoMock.On("GetByUser", mock.AnythingOfType("string")).Return([]*models.Alias{}, assert.AnError)

	suite.AliasRepository = aliasRepoMock
}

func TestAliasServiceTestSuite(t *testing.T) {
	suite.Run(t, new(AliasServiceTestSuite))
}

func (suite *AliasServiceTestSuite) TestAliasService_GetAliasOrDefault() {
	sut := NewAliasService(suite.AliasRepository)

	result1, err1 := sut.GetAliasOrDefault(suite.TestUserId, models.SummaryProject, "wakapi-mobile")
	result2, err2 := sut.GetAliasOrDefault(suite.TestUserId, models.SummaryProject, "wakapi")
	result3, err3 := sut.GetAliasOrDefault(suite.TestUserId, models.SummaryProject, "anchr")

	assert.Equal(suite.T(), "wakapi", result1)
	assert.Nil(suite.T(), err1)
	assert.Equal(suite.T(), "wakapi", result2)
	assert.Nil(suite.T(), err2)
	assert.Equal(suite.T(), "anchr", result3)
	assert.Nil(suite.T(), err3)
}