110 lines
2.8 KiB
Go
110 lines
2.8 KiB
Go
|
package db
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"os"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"code.lila.network/adoralaura/go-urlsh/internal/config"
|
||
|
|
"code.lila.network/adoralaura/go-urlsh/models"
|
||
|
|
"github.com/rs/zerolog"
|
||
|
|
"github.com/uptrace/bun"
|
||
|
|
)
|
||
|
|
|
||
|
|
// createTestingUser creates the necessary testing users for testing MFA.
|
||
|
|
func createTestingUser(t *testing.T, db *bun.DB, user *models.User) {
|
||
|
|
mfaconfig := models.MFAConfig{
|
||
|
|
UserName: user.UserName,
|
||
|
|
TOTPSecret: "abcd",
|
||
|
|
ExpiresAt: time.Now().Add(5 * time.Minute),
|
||
|
|
Active: false,
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err := db.NewInsert().Model(user).Exec(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Couldn't insert test user into DB: %q\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = db.NewInsert().Model(&mfaconfig).Exec(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Couldn't insert test mfa config into DB: %q\n", err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// removeTestingUser is used as deferred cleanup in multifactor tests.
|
||
|
|
// It removes models.User and models.MFAConfig with given username from DB.
|
||
|
|
func removeTestingUser(t *testing.T, db *bun.DB, username string) {
|
||
|
|
_, err := db.NewInsert().Model((*models.User)(nil)).
|
||
|
|
Where("username = ?", username).Exec(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Couldn't remove test user %q from DB: %q\n", username, err)
|
||
|
|
}
|
||
|
|
|
||
|
|
_, err = db.NewInsert().Model((*models.MFAConfig)(nil)).
|
||
|
|
Where("username = ?", username).Exec(context.Background())
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Couldn't remove test mfa config for %q from DB: %q\n", username, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUserHasFailedMFAAttemptTrue(t *testing.T) {
|
||
|
|
cfgfile := os.Getenv("GOURLSH_TEST_CONFIGFILE")
|
||
|
|
logger := config.NewLogger(zerolog.InfoLevel)
|
||
|
|
cfg := config.NewConfig(cfgfile)
|
||
|
|
// setup test environment
|
||
|
|
testdb := InitializeDB(cfg, logger)
|
||
|
|
testUserName := "test_mfafailedtrue"
|
||
|
|
|
||
|
|
mfauser := models.User{
|
||
|
|
UserName: testUserName,
|
||
|
|
Created: time.Now(),
|
||
|
|
}
|
||
|
|
|
||
|
|
createTestingUser(t, testdb, &mfauser)
|
||
|
|
|
||
|
|
defer removeTestingUser(t, testdb, testUserName)
|
||
|
|
|
||
|
|
answer, err := UserHasFailedMFAAttempt(mfauser)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Error running UserHasFailedMFAAttempt in tests: %q\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if answer == false {
|
||
|
|
t.Fatalf("db.UserHasFailedMFAAttempt() = false, want true")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestUserHasFailedMFAAttemptFalse(t *testing.T) {
|
||
|
|
cfgfile := os.Getenv("GOURLSH_TEST_CONFIGFILE")
|
||
|
|
cfg := config.NewConfig(cfgfile)
|
||
|
|
logger := config.NewLogger(zerolog.InfoLevel)
|
||
|
|
// setup test environment
|
||
|
|
testdb := InitializeDB(cfg, logger)
|
||
|
|
testUserName := "test_mfafailed"
|
||
|
|
|
||
|
|
mfauser := models.User{
|
||
|
|
UserName: testUserName,
|
||
|
|
Created: time.Now(),
|
||
|
|
}
|
||
|
|
|
||
|
|
createTestingUser(t, testdb, &mfauser)
|
||
|
|
|
||
|
|
defer removeTestingUser(t, testdb, testUserName)
|
||
|
|
|
||
|
|
answer, err := UserHasFailedMFAAttempt(mfauser)
|
||
|
|
if err != nil {
|
||
|
|
t.Fatalf("Error running UserHasFailedMFAAttempt in tests: %q\n", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if answer == true {
|
||
|
|
t.Fatalf("db.UserHasFailedMFAAttempt() = true, want false")
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestGenerateNewScratchcodes(t *testing.T) {
|
||
|
|
|
||
|
|
}
|