55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package credentials
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
|
|
"git.pcux.cn/xuans/Smail/internal/passwordtest"
|
|
"git.pcux.cn/xuans/Smail/models"
|
|
"git.pcux.cn/xuans/Smail/utils/password"
|
|
)
|
|
|
|
func TestCredentialBusyDoesNotConsumeOrReplaceCodes(t *testing.T) {
|
|
service := newFixture(t)
|
|
if err := service.engine.Sync2(&models.ClientCredential{}); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.SetSecurityQuestion(1, "问题", "answer"); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.SetProtocolEnabled(1, ProtocolIMAP, true); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
protocolCode, err := service.GenerateProtocolCode(1, ProtocolIMAP)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
codes, err := service.GenerateRecoveryCodes(1, 5)
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
release := passwordtest.Hold(t)
|
|
if _, err := service.VerifySecurityQuestion(1, "answer"); !errors.Is(err, password.ErrBusy) {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := service.CheckProtocolCode(1, ProtocolIMAP, protocolCode); !errors.Is(err, password.ErrBusy) {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := service.CheckClientOrProtocolCode(1, ProtocolIMAP, protocolCode); !errors.Is(err, password.ErrBusy) {
|
|
t.Fatal(err)
|
|
}
|
|
if err := service.ConsumeRecoveryCode(1, codes[0]); !errors.Is(err, password.ErrBusy) {
|
|
t.Fatal(err)
|
|
}
|
|
if _, err := service.GenerateRecoveryCodes(1, 5); !errors.Is(err, password.ErrBusy) {
|
|
t.Fatal(err)
|
|
}
|
|
release()
|
|
if err := service.ConsumeRecoveryCode(1, codes[0]); err != nil {
|
|
t.Fatalf("繁忙后恢复码已被错误消费或撤销：%v", err)
|
|
}
|
|
if _, err := service.CheckProtocolCode(1, ProtocolIMAP, protocolCode); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
}
|