mirror of
https://github.com/go-gitea/gitea.git
synced 2026-10-03 18:08:38 +09:00
873efed5a6
The `gitea.com/go-chi/captcha` package only exists for Gitea, so it moves into `modules/imagecaptcha`. - Reloading an expired challenge shows a new image instead of a broken one - Every answer is consumed on its first check - OpenID registration stops after a failed captcha --------- Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
67 lines
1.8 KiB
Go
67 lines
1.8 KiB
Go
// Copyright 2026 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package imagecaptcha
|
|
|
|
import (
|
|
"bytes"
|
|
"image"
|
|
"image/png"
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
"gitea.dev/modules/cache"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestImageCaptcha(t *testing.T) {
|
|
require.NoError(t, cache.Init())
|
|
createWithAnswer := func() (string, string) {
|
|
id, err := CreateNew()
|
|
require.NoError(t, err)
|
|
code, ok := cache.GetCache().Get(cacheKeyPrefix + id)
|
|
require.True(t, ok)
|
|
return id, code
|
|
}
|
|
renderImage := func(id string, refresh bool) *httptest.ResponseRecorder {
|
|
resp := httptest.NewRecorder()
|
|
reqLink := "/captcha?id=" + id
|
|
if refresh {
|
|
reqLink += "&reload=any"
|
|
}
|
|
ServeImage(resp, httptest.NewRequest(http.MethodGet, reqLink, nil))
|
|
return resp
|
|
}
|
|
id, answer := createWithAnswer()
|
|
assert.Len(t, answer, codeLength)
|
|
assert.True(t, Verify(id, answer))
|
|
assert.False(t, Verify(id, answer))
|
|
|
|
id, answer = createWithAnswer()
|
|
assert.False(t, Verify(id, "wrong"))
|
|
assert.False(t, Verify(id, answer))
|
|
assert.False(t, Verify("", ""))
|
|
assert.False(t, Verify("unknown", answer))
|
|
|
|
resp := renderImage("unknown", true)
|
|
assert.Equal(t, http.StatusNotFound, resp.Code)
|
|
_, exists := cache.GetCache().Get(cacheKeyPrefix + "unknown")
|
|
assert.False(t, exists)
|
|
|
|
id, _ = createWithAnswer()
|
|
first := renderImage(id, false)
|
|
decoded, err := png.Decode(bytes.NewReader(first.Body.Bytes()))
|
|
require.NoError(t, err)
|
|
assert.Equal(t, image.Rect(0, 0, imageWidth, imageHeight), decoded.Bounds())
|
|
second := renderImage(id, false)
|
|
assert.Equal(t, first.Body.Bytes(), second.Body.Bytes())
|
|
|
|
require.NoError(t, cache.GetCache().Delete(cacheKeyPrefix+id))
|
|
_ = renderImage(id, true)
|
|
_, exists = cache.GetCache().Get(cacheKeyPrefix + id)
|
|
assert.True(t, exists)
|
|
}
|