forked from NYANDEV/forgejo
9886aec9f8
(cherry picked from commit 7b0549cd70aa7cafec853e15b25270847c59850b)
(cherry picked from commit 13e10a65d974c7b594681bfa36402a6144862116)
(cherry picked from commit 89982e6c4a
)
(cherry picked from commit a4acf6343d9f5c6dedeb261c524cd7ac5ae1b3c1)
23 lines
511 B
Go
23 lines
511 B
Go
// SPDX-License-Identifier: MIT
|
|
|
|
package context
|
|
|
|
import (
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestGetOtpHeader(t *testing.T) {
|
|
header := http.Header{}
|
|
assert.EqualValues(t, "", getOtpHeader(header))
|
|
// Gitea
|
|
giteaOtp := "123456"
|
|
header.Set("X-Gitea-OTP", giteaOtp)
|
|
assert.EqualValues(t, giteaOtp, getOtpHeader(header))
|
|
// Forgejo has precedence
|
|
forgejoOtp := "abcdef"
|
|
header.Set("X-Forgejo-OTP", forgejoOtp)
|
|
assert.EqualValues(t, forgejoOtp, getOtpHeader(header))
|
|
}
|