mix
15baa1e5d0
test / test (push) Has been cancelled
Co-authored-by: Cursor <cursoragent@cursor.com>
102 lines
2.5 KiB
Go
102 lines
2.5 KiB
Go
package store
|
|
|
|
import (
|
|
"errors"
|
|
"testing"
|
|
)
|
|
|
|
func TestUpdateUser(t *testing.T) {
|
|
st := openTestStore(t)
|
|
|
|
if err := st.CreateGlobalUser("admin", "hash-one"); err != nil {
|
|
t.Fatalf("CreateGlobalUser: %v", err)
|
|
}
|
|
u, err := st.GetUserByUsername("admin")
|
|
if err != nil {
|
|
t.Fatalf("GetUserByUsername: %v", err)
|
|
}
|
|
if err := st.UpdateUser(u.ID, "operator", "hash-two", "reports@hub.example"); err != nil {
|
|
t.Fatalf("UpdateUser: %v", err)
|
|
}
|
|
|
|
got, err := st.GetUser(u.ID)
|
|
if err != nil {
|
|
t.Fatalf("GetUser: %v", err)
|
|
}
|
|
if got.DMARCReportEmail != "reports@hub.example" {
|
|
t.Fatalf("dmarc email = %q", got.DMARCReportEmail)
|
|
}
|
|
email, err := st.GlobalDMARCReportEmail()
|
|
if err != nil {
|
|
t.Fatalf("GlobalDMARCReportEmail: %v", err)
|
|
}
|
|
if email != "reports@hub.example" {
|
|
t.Fatalf("settings dmarc = %q", email)
|
|
}
|
|
|
|
if err := st.UpdateUser(u.ID, "operator", "hash-three", ""); err != nil {
|
|
t.Fatalf("clear dmarc email: %v", err)
|
|
}
|
|
got, err = st.GetUser(u.ID)
|
|
if err != nil {
|
|
t.Fatalf("GetUser: %v", err)
|
|
}
|
|
if got.Username != "operator" || got.PasswordHash != "hash-three" {
|
|
t.Fatalf("unexpected user after update: %+v", got)
|
|
}
|
|
if got.CreatedAt.IsZero() {
|
|
t.Fatal("update dropped created_at")
|
|
}
|
|
}
|
|
|
|
func TestUpdateUserWithoutUser(t *testing.T) {
|
|
st := openTestStore(t)
|
|
|
|
if err := st.UpdateUser(1, "operator", "hash", ""); !errors.Is(err, ErrUserNotFound) {
|
|
t.Fatalf("UpdateUser without user = %v, want ErrUserNotFound", err)
|
|
}
|
|
exists, err := st.UserExists()
|
|
if err != nil {
|
|
t.Fatalf("UserExists: %v", err)
|
|
}
|
|
if exists {
|
|
t.Fatal("UpdateUser created a user")
|
|
}
|
|
}
|
|
|
|
func TestCreateDomainAdminUser(t *testing.T) {
|
|
st := openTestStore(t)
|
|
if err := st.CreateGlobalUser("admin", "hash"); err != nil {
|
|
t.Fatalf("CreateGlobalUser: %v", err)
|
|
}
|
|
d, err := st.AddDomain("example.com", "s1")
|
|
if err != nil {
|
|
t.Fatalf("AddDomain: %v", err)
|
|
}
|
|
id, err := st.CreateUser("domainop", "hash2", RoleDomainAdmin, []int64{d.ID})
|
|
if err != nil {
|
|
t.Fatalf("CreateUser: %v", err)
|
|
}
|
|
u, err := st.GetUser(id)
|
|
if err != nil {
|
|
t.Fatalf("GetUser: %v", err)
|
|
}
|
|
if len(u.DomainIDs) != 1 || u.DomainIDs[0] != d.ID {
|
|
t.Fatalf("domain ids = %v, want [%d]", u.DomainIDs, d.ID)
|
|
}
|
|
}
|
|
|
|
func TestDeleteLastGlobalUser(t *testing.T) {
|
|
st := openTestStore(t)
|
|
if err := st.CreateGlobalUser("admin", "hash"); err != nil {
|
|
t.Fatalf("CreateGlobalUser: %v", err)
|
|
}
|
|
u, err := st.GetUserByUsername("admin")
|
|
if err != nil {
|
|
t.Fatalf("GetUserByUsername: %v", err)
|
|
}
|
|
if err := st.DeleteUser(u.ID); !errors.Is(err, ErrLastGlobal) {
|
|
t.Fatalf("DeleteUser = %v, want ErrLastGlobal", err)
|
|
}
|
|
}
|