mirror of
https://github.com/elyby/chrly.git
synced 2024-12-12 08:19:15 +05:30
521 lines
23 KiB
Go
521 lines
23 KiB
Go
package queue
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"encoding/base64"
|
|
"errors"
|
|
"net"
|
|
"net/url"
|
|
"strings"
|
|
"syscall"
|
|
"time"
|
|
|
|
"github.com/elyby/chrly/api/mojang"
|
|
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/mock"
|
|
"github.com/stretchr/testify/suite"
|
|
|
|
mocks "github.com/elyby/chrly/tests"
|
|
)
|
|
|
|
type mojangApiMocks struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (o *mojangApiMocks) UsernamesToUuids(usernames []string) ([]*mojang.ProfileInfo, error) {
|
|
args := o.Called(usernames)
|
|
var result []*mojang.ProfileInfo
|
|
if casted, ok := args.Get(0).([]*mojang.ProfileInfo); ok {
|
|
result = casted
|
|
}
|
|
|
|
return result, args.Error(1)
|
|
}
|
|
|
|
func (o *mojangApiMocks) UuidToTextures(uuid string, signed bool) (*mojang.SignedTexturesResponse, error) {
|
|
args := o.Called(uuid, signed)
|
|
var result *mojang.SignedTexturesResponse
|
|
if casted, ok := args.Get(0).(*mojang.SignedTexturesResponse); ok {
|
|
result = casted
|
|
}
|
|
|
|
return result, args.Error(1)
|
|
}
|
|
|
|
type mockStorage struct {
|
|
mock.Mock
|
|
}
|
|
|
|
func (m *mockStorage) GetUuid(username string) (string, error) {
|
|
args := m.Called(username)
|
|
return args.String(0), args.Error(1)
|
|
}
|
|
|
|
func (m *mockStorage) StoreUuid(username string, uuid string) error {
|
|
args := m.Called(username, uuid)
|
|
return args.Error(0)
|
|
}
|
|
|
|
func (m *mockStorage) GetTextures(uuid string) (*mojang.SignedTexturesResponse, error) {
|
|
args := m.Called(uuid)
|
|
var result *mojang.SignedTexturesResponse
|
|
if casted, ok := args.Get(0).(*mojang.SignedTexturesResponse); ok {
|
|
result = casted
|
|
}
|
|
|
|
return result, args.Error(1)
|
|
}
|
|
|
|
func (m *mockStorage) StoreTextures(uuid string, textures *mojang.SignedTexturesResponse) {
|
|
m.Called(uuid, textures)
|
|
}
|
|
|
|
type queueTestSuite struct {
|
|
suite.Suite
|
|
Queue *JobsQueue
|
|
Storage *mockStorage
|
|
MojangApi *mojangApiMocks
|
|
Logger *mocks.WdMock
|
|
Iterate func()
|
|
|
|
iterateChan chan bool
|
|
done func()
|
|
}
|
|
|
|
func (suite *queueTestSuite) SetupSuite() {
|
|
uuidsQueueIterationDelay = 0
|
|
}
|
|
|
|
func (suite *queueTestSuite) SetupTest() {
|
|
suite.Storage = &mockStorage{}
|
|
suite.Logger = &mocks.WdMock{}
|
|
|
|
suite.Queue = &JobsQueue{Storage: suite.Storage, Logger: suite.Logger}
|
|
|
|
suite.iterateChan = make(chan bool)
|
|
forever = func() bool {
|
|
return <-suite.iterateChan
|
|
}
|
|
|
|
suite.Iterate = func() {
|
|
suite.iterateChan <- true
|
|
}
|
|
|
|
suite.done = func() {
|
|
suite.iterateChan <- false
|
|
}
|
|
|
|
suite.MojangApi = new(mojangApiMocks)
|
|
usernamesToUuids = suite.MojangApi.UsernamesToUuids
|
|
uuidToTextures = suite.MojangApi.UuidToTextures
|
|
}
|
|
|
|
func (suite *queueTestSuite) TearDownTest() {
|
|
suite.done()
|
|
time.Sleep(10 * time.Millisecond) // Add delay to let finish all goroutines before assert mocks calls
|
|
suite.MojangApi.AssertExpectations(suite.T())
|
|
suite.Storage.AssertExpectations(suite.T())
|
|
suite.Logger.AssertExpectations(suite.T())
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForOneUsernameWithoutAnyCache() {
|
|
expectedResult := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything)
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.request", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.textures.request_time", mock.Anything).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_hit", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", expectedResult).Once()
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(expectedResult, nil)
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
suite.Iterate()
|
|
|
|
result := <-resultChan
|
|
suite.Assert().Equal(expectedResult, result)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForFewUsernamesWithoutAnyCache() {
|
|
expectedResult1 := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
expectedResult2 := &mojang.SignedTexturesResponse{Id: "4566e69fc90748ee8d71d7ba5aa00d20", Name: "Thinkofdeath"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Twice()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Twice()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(2)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything)
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.request", int64(1)).Twice()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.textures.request_time", mock.Anything).Twice()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_hit", int64(1)).Twice()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Twice()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("", &ValueNotFound{})
|
|
suite.Storage.On("GetUuid", "Thinkofdeath").Once().Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil)
|
|
suite.Storage.On("StoreUuid", "Thinkofdeath", "4566e69fc90748ee8d71d7ba5aa00d20").Once().Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("GetTextures", "4566e69fc90748ee8d71d7ba5aa00d20").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", expectedResult1).Once()
|
|
suite.Storage.On("StoreTextures", "4566e69fc90748ee8d71d7ba5aa00d20", expectedResult2).Once()
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb", "Thinkofdeath"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
{Id: "4566e69fc90748ee8d71d7ba5aa00d20", Name: "Thinkofdeath"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(expectedResult1, nil)
|
|
suite.MojangApi.On("UuidToTextures", "4566e69fc90748ee8d71d7ba5aa00d20", true).Once().Return(expectedResult2, nil)
|
|
|
|
resultChan1 := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
resultChan2 := suite.Queue.GetTexturesForUsername("Thinkofdeath")
|
|
|
|
suite.Iterate()
|
|
|
|
suite.Assert().Equal(expectedResult1, <-resultChan1)
|
|
suite.Assert().Equal(expectedResult2, <-resultChan2)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForUsernameWithCachedUuid() {
|
|
expectedResult := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.cache_hit", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.request", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.textures.request_time", mock.Anything).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("0d252b7218b648bfb86c2ae476954d32", nil)
|
|
// Storage.StoreUuid shouldn't be called
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", expectedResult).Once()
|
|
|
|
// MojangApi.UsernamesToUuids shouldn't be called
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(expectedResult, nil)
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
// Note that there is no iteration
|
|
|
|
result := <-resultChan
|
|
suite.Assert().Equal(expectedResult, result)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForUsernameWithFullyCachedResult() {
|
|
expectedResult := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.cache_hit", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.cache_hit", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("0d252b7218b648bfb86c2ae476954d32", nil)
|
|
// Storage.StoreUuid shouldn't be called
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(expectedResult, nil)
|
|
// Storage.StoreTextures shouldn't be called
|
|
|
|
// MojangApi.UsernamesToUuids shouldn't be called
|
|
// MojangApi.UuidToTextures shouldn't be called
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
// Note that there is no iteration
|
|
|
|
result := <-resultChan
|
|
suite.Assert().Equal(expectedResult, result)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForUsernameWithCachedUnknownUuid() {
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.cache_hit_nil", int64(1)).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("", nil)
|
|
// Storage.StoreUuid shouldn't be called
|
|
// Storage.GetTextures shouldn't be called
|
|
// Storage.StoreTextures shouldn't be called
|
|
|
|
// MojangApi.UsernamesToUuids shouldn't be called
|
|
// MojangApi.UuidToTextures shouldn't be called
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
// Note that there is no iteration
|
|
|
|
suite.Assert().Nil(<-resultChan)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForMoreThan100Usernames() {
|
|
usernames := make([]string, 120)
|
|
for i := 0; i < 120; i++ {
|
|
usernames[i] = randStr(8)
|
|
}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Times(120)
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Times(120)
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(100)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(20)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(20)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything).Twice()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_miss", int64(1)).Times(120)
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Times(120)
|
|
|
|
suite.Storage.On("GetUuid", mock.Anything).Times(120).Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", mock.Anything, "").Times(120).Return(nil) // should be called with "" if username is not compared to uuid
|
|
// Storage.GetTextures and Storage.SetTextures shouldn't be called
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", usernames[0:100]).Once().Return([]*mojang.ProfileInfo{}, nil)
|
|
suite.MojangApi.On("UsernamesToUuids", usernames[100:120]).Once().Return([]*mojang.ProfileInfo{}, nil)
|
|
|
|
channels := make([]chan *mojang.SignedTexturesResponse, 120)
|
|
for i, username := range usernames {
|
|
channels[i] = suite.Queue.GetTexturesForUsername(username)
|
|
}
|
|
|
|
suite.Iterate()
|
|
suite.Iterate()
|
|
|
|
for _, channel := range channels {
|
|
<-channel
|
|
}
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForTheSameUsernames() {
|
|
expectedResult := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Twice()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.already_in_queue", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything)
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.request", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.textures.request_time", mock.Anything).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_hit", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Twice().Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", expectedResult).Once()
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(expectedResult, nil)
|
|
|
|
resultChan1 := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
resultChan2 := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
suite.Iterate()
|
|
|
|
suite.Assert().Equal(expectedResult, <-resultChan1)
|
|
suite.Assert().Equal(expectedResult, <-resultChan2)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForUsernameThatAlreadyProcessing() {
|
|
expectedResult := &mojang.SignedTexturesResponse{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"}
|
|
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Twice()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.already_in_queue", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything)
|
|
suite.Logger.On("IncCounter", "mojang_textures.textures.request", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.textures.request_time", mock.Anything).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_hit", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Twice().Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Once().Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", expectedResult).Once()
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).
|
|
Once().
|
|
After(10*time.Millisecond). // Simulate long round trip
|
|
Return(expectedResult, nil)
|
|
|
|
resultChan1 := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
// Note that for entire test there is only one iteration
|
|
suite.Iterate()
|
|
|
|
// Let it meet delayed UuidToTextures request
|
|
time.Sleep(5 * time.Millisecond)
|
|
|
|
resultChan2 := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
suite.Assert().Equal(expectedResult, <-resultChan1)
|
|
suite.Assert().Equal(expectedResult, <-resultChan2)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestDoNothingWhenNoTasks() {
|
|
suite.Logger.On("IncCounter", "mojang_textures.request", int64(1)).Once()
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.queued", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.iteration_size", int64(1)).Once()
|
|
suite.Logger.On("UpdateGauge", "mojang_textures.usernames.queue_size", int64(0)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.usernames.round_time", mock.Anything)
|
|
suite.Logger.On("IncCounter", "mojang_textures.usernames.uuid_miss", int64(1)).Once()
|
|
suite.Logger.On("RecordTimer", "mojang_textures.result_time", mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Once().Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "").Once().Return(nil)
|
|
// Storage.GetTextures and Storage.StoreTextures shouldn't be called
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{}, nil)
|
|
|
|
// Perform first iteration and await it finish
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
|
|
suite.Iterate()
|
|
|
|
suite.Assert().Nil(<-resultChan)
|
|
|
|
// Let it to perform a few more iterations to ensure, that there is no calls to external APIs
|
|
suite.Iterate()
|
|
suite.Iterate()
|
|
}
|
|
|
|
type timeoutError struct {
|
|
}
|
|
|
|
func (*timeoutError) Error() string { return "timeout error" }
|
|
func (*timeoutError) Timeout() bool { return true }
|
|
func (*timeoutError) Temporary() bool { return false }
|
|
|
|
var expectedErrors = []error{
|
|
&mojang.BadRequestError{},
|
|
&mojang.TooManyRequestsError{},
|
|
&mojang.ServerError{},
|
|
&timeoutError{},
|
|
&url.Error{Op: "GET", URL: "http://localhost"},
|
|
&net.OpError{Op: "read"},
|
|
&net.OpError{Op: "dial"},
|
|
syscall.ECONNREFUSED,
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestShouldNotLogErrorWhenExpectedErrorReturnedFromUsernameToUuidRequest() {
|
|
suite.Logger.On("IncCounter", mock.Anything, mock.Anything)
|
|
suite.Logger.On("UpdateGauge", mock.Anything, mock.Anything)
|
|
suite.Logger.On("RecordTimer", mock.Anything, mock.Anything)
|
|
suite.Logger.On("Debug", ":name: Got response error :err", mock.Anything, mock.Anything).Times(len(expectedErrors))
|
|
suite.Logger.On("Warning", ":name: Got 429 Too Many Requests :err", mock.Anything, mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Return("", &ValueNotFound{})
|
|
|
|
for _, err := range expectedErrors {
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return(nil, err)
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
suite.Iterate()
|
|
suite.Assert().Nil(<-resultChan)
|
|
suite.MojangApi.AssertExpectations(suite.T())
|
|
suite.MojangApi.ExpectedCalls = nil // https://github.com/stretchr/testify/issues/558#issuecomment-372112364
|
|
}
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestShouldLogEmergencyOnUnexpectedErrorReturnedFromUsernameToUuidRequest() {
|
|
suite.Logger.On("IncCounter", mock.Anything, mock.Anything)
|
|
suite.Logger.On("UpdateGauge", mock.Anything, mock.Anything)
|
|
suite.Logger.On("RecordTimer", mock.Anything, mock.Anything)
|
|
suite.Logger.On("Debug", ":name: Got response error :err", mock.Anything, mock.Anything).Once()
|
|
suite.Logger.On("Emergency", ":name: Unknown Mojang response error: :err", mock.Anything, mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Return("", &ValueNotFound{})
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return(nil, errors.New("unexpected error"))
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
suite.Iterate()
|
|
suite.Assert().Nil(<-resultChan)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestShouldNotLogErrorWhenExpectedErrorReturnedFromUuidToTexturesRequest() {
|
|
suite.Logger.On("IncCounter", mock.Anything, mock.Anything)
|
|
suite.Logger.On("UpdateGauge", mock.Anything, mock.Anything)
|
|
suite.Logger.On("RecordTimer", mock.Anything, mock.Anything)
|
|
suite.Logger.On("Debug", ":name: Got response error :err", mock.Anything, mock.Anything).Times(len(expectedErrors))
|
|
suite.Logger.On("Warning", ":name: Got 429 Too Many Requests :err", mock.Anything, mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", (*mojang.SignedTexturesResponse)(nil))
|
|
|
|
for _, err := range expectedErrors {
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(nil, err)
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
suite.Iterate()
|
|
suite.Assert().Nil(<-resultChan)
|
|
suite.MojangApi.AssertExpectations(suite.T())
|
|
suite.MojangApi.ExpectedCalls = nil // https://github.com/stretchr/testify/issues/558#issuecomment-372112364
|
|
}
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestShouldLogEmergencyOnUnexpectedErrorReturnedFromUuidToTexturesRequest() {
|
|
suite.Logger.On("IncCounter", mock.Anything, mock.Anything)
|
|
suite.Logger.On("UpdateGauge", mock.Anything, mock.Anything)
|
|
suite.Logger.On("RecordTimer", mock.Anything, mock.Anything)
|
|
suite.Logger.On("Debug", ":name: Got response error :err", mock.Anything, mock.Anything).Once()
|
|
suite.Logger.On("Emergency", ":name: Unknown Mojang response error: :err", mock.Anything, mock.Anything).Once()
|
|
|
|
suite.Storage.On("GetUuid", "maksimkurb").Return("", &ValueNotFound{})
|
|
suite.Storage.On("StoreUuid", "maksimkurb", "0d252b7218b648bfb86c2ae476954d32").Return(nil)
|
|
suite.Storage.On("GetTextures", "0d252b7218b648bfb86c2ae476954d32").Return(nil, &ValueNotFound{})
|
|
suite.Storage.On("StoreTextures", "0d252b7218b648bfb86c2ae476954d32", (*mojang.SignedTexturesResponse)(nil))
|
|
|
|
suite.MojangApi.On("UsernamesToUuids", []string{"maksimkurb"}).Once().Return([]*mojang.ProfileInfo{
|
|
{Id: "0d252b7218b648bfb86c2ae476954d32", Name: "maksimkurb"},
|
|
}, nil)
|
|
suite.MojangApi.On("UuidToTextures", "0d252b7218b648bfb86c2ae476954d32", true).Once().Return(nil, errors.New("unexpected error"))
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("maksimkurb")
|
|
suite.Iterate()
|
|
suite.Assert().Nil(<-resultChan)
|
|
}
|
|
|
|
func (suite *queueTestSuite) TestReceiveTexturesForNotAllowedMojangUsername() {
|
|
suite.Logger.On("IncCounter", "mojang_textures.invalid_username", int64(1)).Once()
|
|
|
|
resultChan := suite.Queue.GetTexturesForUsername("Not allowed")
|
|
suite.Assert().Nil(<-resultChan)
|
|
}
|
|
|
|
func TestJobsQueueSuite(t *testing.T) {
|
|
suite.Run(t, new(queueTestSuite))
|
|
}
|
|
|
|
var replacer = strings.NewReplacer("-", "_", "=", "")
|
|
|
|
// https://stackoverflow.com/a/50581165
|
|
func randStr(len int) string {
|
|
buff := make([]byte, len)
|
|
_, _ = rand.Read(buff)
|
|
str := replacer.Replace(base64.URLEncoding.EncodeToString(buff))
|
|
|
|
// Base 64 can be longer than len
|
|
return str[:len]
|
|
}
|