mirror of
https://github.com/elyby/chrly.git
synced 2024-08-25 14:20:00 +05:30
164 lines
4.5 KiB
Go
164 lines
4.5 KiB
Go
package http
|
|
|
|
import (
|
|
"bytes"
|
|
"image"
|
|
"image/png"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"github.com/golang/mock/gomock"
|
|
testify "github.com/stretchr/testify/assert"
|
|
|
|
"github.com/elyby/chrly/db"
|
|
"github.com/elyby/chrly/model"
|
|
)
|
|
|
|
type capesTestCase struct {
|
|
Name string
|
|
RequestUrl string
|
|
ExpectedLogKey string
|
|
ExistsInLocalStorage bool
|
|
ExistsInMojang bool
|
|
HasCapeInMojangResp bool
|
|
AssertResponse func(assert *testify.Assertions, resp *http.Response)
|
|
}
|
|
|
|
var capesTestCases = []*capesTestCase{
|
|
{
|
|
Name: "Obtain cape for known username",
|
|
ExistsInLocalStorage: true,
|
|
AssertResponse: func(assert *testify.Assertions, resp *http.Response) {
|
|
assert.Equal(200, resp.StatusCode)
|
|
responseData, _ := ioutil.ReadAll(resp.Body)
|
|
assert.Equal(createCape(), responseData)
|
|
assert.Equal("image/png", resp.Header.Get("Content-Type"))
|
|
},
|
|
},
|
|
{
|
|
Name: "Obtain cape for unknown username that exists in Mojang and has a cape",
|
|
ExistsInLocalStorage: false,
|
|
ExistsInMojang: true,
|
|
HasCapeInMojangResp: true,
|
|
AssertResponse: func(assert *testify.Assertions, resp *http.Response) {
|
|
assert.Equal(301, resp.StatusCode)
|
|
assert.Equal("http://mojang/cape.png", resp.Header.Get("Location"))
|
|
},
|
|
},
|
|
{
|
|
Name: "Obtain cape for unknown username that exists in Mojang, but don't has a cape",
|
|
ExistsInLocalStorage: false,
|
|
ExistsInMojang: true,
|
|
HasCapeInMojangResp: false,
|
|
AssertResponse: func(assert *testify.Assertions, resp *http.Response) {
|
|
assert.Equal(404, resp.StatusCode)
|
|
},
|
|
},
|
|
{
|
|
Name: "Obtain cape for unknown username that doesn't exists in Mojang",
|
|
ExistsInLocalStorage: false,
|
|
ExistsInMojang: false,
|
|
AssertResponse: func(assert *testify.Assertions, resp *http.Response) {
|
|
assert.Equal(404, resp.StatusCode)
|
|
},
|
|
},
|
|
}
|
|
|
|
func TestConfig_Cape(t *testing.T) {
|
|
performTest := func(t *testing.T, testCase *capesTestCase) {
|
|
assert := testify.New(t)
|
|
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
config, mocks := setupMocks(ctrl)
|
|
|
|
mocks.Log.EXPECT().IncCounter(testCase.ExpectedLogKey, int64(1))
|
|
if testCase.ExistsInLocalStorage {
|
|
mocks.Capes.EXPECT().FindByUsername("mock_username").Return(&model.Cape{
|
|
File: bytes.NewReader(createCape()),
|
|
}, nil)
|
|
} else {
|
|
mocks.Capes.EXPECT().FindByUsername("mock_username").Return(nil, &db.CapeNotFoundError{Who: "mock_username"})
|
|
}
|
|
|
|
if testCase.ExistsInMojang {
|
|
textures := createTexturesResponse(false, testCase.HasCapeInMojangResp)
|
|
mocks.Queue.On("GetTexturesForUsername", "mock_username").Return(textures)
|
|
} else {
|
|
mocks.Queue.On("GetTexturesForUsername", "mock_username").Return(nil)
|
|
}
|
|
|
|
req := httptest.NewRequest("GET", testCase.RequestUrl, nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
config.CreateHandler().ServeHTTP(w, req)
|
|
|
|
resp := w.Result()
|
|
testCase.AssertResponse(assert, resp)
|
|
}
|
|
|
|
t.Run("Normal API", func(t *testing.T) {
|
|
for _, testCase := range capesTestCases {
|
|
testCase.RequestUrl = "http://chrly/cloaks/mock_username"
|
|
testCase.ExpectedLogKey = "capes.request"
|
|
t.Run(testCase.Name, func(t *testing.T) {
|
|
performTest(t, testCase)
|
|
})
|
|
}
|
|
})
|
|
|
|
t.Run("GET fallback API", func(t *testing.T) {
|
|
for _, testCase := range capesTestCases {
|
|
testCase.RequestUrl = "http://chrly/cloaks?name=mock_username"
|
|
testCase.ExpectedLogKey = "capes.get_request"
|
|
t.Run(testCase.Name, func(t *testing.T) {
|
|
performTest(t, testCase)
|
|
})
|
|
}
|
|
|
|
t.Run("Should trim trailing slash", func(t *testing.T) {
|
|
assert := testify.New(t)
|
|
|
|
req := httptest.NewRequest("GET", "http://chrly/cloaks/?name=notch", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
(&Config{}).CreateHandler().ServeHTTP(w, req)
|
|
|
|
resp := w.Result()
|
|
assert.Equal(301, resp.StatusCode)
|
|
assert.Equal("http://chrly/cloaks?name=notch", resp.Header.Get("Location"))
|
|
})
|
|
|
|
t.Run("Return error when name is not provided", func(t *testing.T) {
|
|
assert := testify.New(t)
|
|
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
|
|
config, mocks := setupMocks(ctrl)
|
|
mocks.Log.EXPECT().IncCounter("capes.get_request", int64(1))
|
|
|
|
req := httptest.NewRequest("GET", "http://chrly/cloaks", nil)
|
|
w := httptest.NewRecorder()
|
|
|
|
config.CreateHandler().ServeHTTP(w, req)
|
|
|
|
resp := w.Result()
|
|
assert.Equal(400, resp.StatusCode)
|
|
})
|
|
})
|
|
}
|
|
|
|
// Cape md5: 424ff79dce9940af89c28ad80de8aaad
|
|
func createCape() []byte {
|
|
img := image.NewAlpha(image.Rect(0, 0, 64, 32))
|
|
writer := &bytes.Buffer{}
|
|
_ = png.Encode(writer, img)
|
|
pngBytes, _ := ioutil.ReadAll(writer)
|
|
|
|
return pngBytes
|
|
}
|