mirror of
https://github.com/mjl-/mox.git
synced 2025-07-13 01:34:39 +03:00
ac4b006ecd
The intent to remove the account is stored in the database. At startup, if there are any such referenes, they are applied by removing the account directories and the entry in the database. This ensures the account directory is properly removed, even on incomplete shutdown. Don't add an account when its directory already exits.
124 lines
3.0 KiB
Go
124 lines
3.0 KiB
Go
package store
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/mjl-/bstore"
|
|
|
|
"github.com/mjl-/mox/mox-"
|
|
)
|
|
|
|
func TestLoginAttempt(t *testing.T) {
|
|
os.RemoveAll("../testdata/store/data")
|
|
mox.ConfigStaticPath = filepath.FromSlash("../testdata/store/mox.conf")
|
|
mox.MustLoadConfig(true, false)
|
|
|
|
xctx, xcancel := context.WithCancel(ctxbg)
|
|
defer xcancel() // Stop clearing of LoginAttempts.
|
|
err := Init(xctx)
|
|
tcheck(t, err, "store init")
|
|
stopc := make(chan struct{})
|
|
writeLoginAttemptStop <- stopc
|
|
<-stopc
|
|
defer func() {
|
|
// Ensure Close() below finishes
|
|
go func() {
|
|
c := <-writeLoginAttemptStop
|
|
c <- struct{}{}
|
|
}()
|
|
|
|
err := Close()
|
|
tcheck(t, err, "store close")
|
|
}()
|
|
|
|
a1 := LoginAttempt{
|
|
Last: time.Now(),
|
|
First: time.Now(),
|
|
AccountName: "mjl1",
|
|
UserAgent: "0", // "0" so we update instead of insert when testing automatic cleanup below.
|
|
Result: AuthError,
|
|
}
|
|
a2 := a1
|
|
a2.AccountName = "mjl2"
|
|
a3 := a1
|
|
a3.AccountName = "mjl3"
|
|
a3.Last = a3.Last.Add(-31 * 24 * time.Hour) // Will be cleaned up.
|
|
a3.First = a3.Last
|
|
LoginAttemptAdd(ctxbg, pkglog, a1)
|
|
LoginAttemptAdd(ctxbg, pkglog, a2)
|
|
LoginAttemptAdd(ctxbg, pkglog, a3)
|
|
|
|
// Ensure there are no LoginAttempts that still need to be written.
|
|
loginAttemptDrain := func() {
|
|
for {
|
|
select {
|
|
case la := <-writeLoginAttempt:
|
|
loginAttemptWrite(la)
|
|
default:
|
|
return
|
|
}
|
|
}
|
|
}
|
|
|
|
loginAttemptDrain()
|
|
|
|
l, err := LoginAttemptList(ctxbg, "", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 3)
|
|
|
|
// Test limit.
|
|
l, err = LoginAttemptList(ctxbg, "", 2)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 2)
|
|
|
|
// Test account filter.
|
|
l, err = LoginAttemptList(ctxbg, "mjl1", 2)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 1)
|
|
|
|
// Cleanup will remove the entry for mjl3 and leave others.
|
|
err = LoginAttemptCleanup(ctxbg)
|
|
tcheck(t, err, "cleanup login attempt")
|
|
l, err = LoginAttemptList(ctxbg, "", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 2)
|
|
|
|
// Removing account will keep last entry for mjl2.
|
|
err = AuthDB.Write(ctxbg, func(tx *bstore.Tx) error {
|
|
return loginAttemptRemoveAccount(tx, "mjl1")
|
|
})
|
|
tcheck(t, err, "remove login attempt account")
|
|
l, err = LoginAttemptList(ctxbg, "", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 1)
|
|
|
|
l, err = LoginAttemptList(ctxbg, "mjl2", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), 1)
|
|
|
|
// Insert 3 failing entries. Then add another and see we still have 3.
|
|
loginAttemptsMaxPerAccount = 3
|
|
for i := range loginAttemptsMaxPerAccount {
|
|
a := a2
|
|
a.UserAgent = fmt.Sprintf("%d", i)
|
|
LoginAttemptAdd(ctxbg, pkglog, a)
|
|
}
|
|
loginAttemptDrain()
|
|
l, err = LoginAttemptList(ctxbg, "", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), loginAttemptsMaxPerAccount)
|
|
|
|
a := a2
|
|
a.UserAgent = fmt.Sprintf("%d", loginAttemptsMaxPerAccount)
|
|
LoginAttemptAdd(ctxbg, pkglog, a)
|
|
loginAttemptDrain()
|
|
l, err = LoginAttemptList(ctxbg, "", 0)
|
|
tcheck(t, err, "list login attempts")
|
|
tcompare(t, len(l), loginAttemptsMaxPerAccount)
|
|
}
|