129 lines
3.5 KiB
Go
129 lines
3.5 KiB
Go
package user
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"strings"
|
|
"testing"
|
|
|
|
"git.golem15.com/golem15/sm-user-plugin/models"
|
|
"git.golem15.com/golem15/summercms/modules/backpack"
|
|
"git.golem15.com/golem15/summercms/modules/bonfire"
|
|
"git.golem15.com/golem15/summercms/modules/pact"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
func TestRequirePasswordChange(t *testing.T) {
|
|
app, gdb := sessionApp(t, "0s")
|
|
user := insertUser(t, gdb, "alice@example.com", "correct-horse")
|
|
p := &Plugin{app: app}
|
|
var _ pact.HasCommands = p
|
|
cmd := commandNamed(t, p.Commands(), "user:require-password-change")
|
|
var buf bytes.Buffer
|
|
out := bonfire.NewOutput(nil, &buf, &buf)
|
|
if err := cmd.Run(context.Background(), emailArg{email: "alice@example.com"}, out); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(buf.String(), "alice@example.com") {
|
|
t.Fatalf("output = %q", buf.String())
|
|
}
|
|
var fresh models.User
|
|
if err := gdb.First(&fresh, user.ID).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !fresh.MustChangePassword {
|
|
t.Fatal("must_change_password stayed false")
|
|
}
|
|
if err := cmd.Run(context.Background(), emailArg{email: "missing@example.com"}, out); err == nil {
|
|
t.Fatal("unknown email returned nil error")
|
|
}
|
|
}
|
|
|
|
func TestAPITokenCreateCommand(t *testing.T) {
|
|
app, gdb := sessionApp(t, "0s")
|
|
user := insertUser(t, gdb, "pat@example.com", "correct-horse")
|
|
cmd := commandNamed(t, (&Plugin{app: app}).Commands(), "user:token:create")
|
|
var buf bytes.Buffer
|
|
out := bonfire.NewOutput(nil, &buf, &buf)
|
|
if err := cmd.Run(context.Background(), tokenCreateArg{email: user.Email, name: "rome scripts", scopes: "read,write"}, out); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if !strings.Contains(buf.String(), "usr_") {
|
|
t.Fatalf("secret missing from output %q", buf.String())
|
|
}
|
|
var n int64
|
|
if err := gdb.Model(&models.APIToken{}).Where("user_id = ? AND revoked_at IS NULL", user.ID).Count(&n).Error; err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
if n != 1 {
|
|
t.Fatalf("tokens = %d", n)
|
|
}
|
|
}
|
|
|
|
func TestConsoleWithDBOpensFromConfigWhenUnpublished(t *testing.T) {
|
|
p := &Plugin{app: backpack.New(sessionConfig(t))}
|
|
err := p.withDB(context.Background(), func(*gorm.DB) error {
|
|
t.Fatal("fn should not run without a DSN")
|
|
return nil
|
|
})
|
|
if err == nil {
|
|
t.Fatal("expected OpenFromApp to fail")
|
|
}
|
|
if strings.Contains(err.Error(), "database is not configured") {
|
|
t.Fatalf("still the Lookup-only error: %v", err)
|
|
}
|
|
if !strings.Contains(err.Error(), "database.dsn is empty") {
|
|
t.Fatalf("got %v", err)
|
|
}
|
|
}
|
|
|
|
func commandNamed(t *testing.T, cmds []bonfire.Command, name string) bonfire.Command {
|
|
t.Helper()
|
|
for _, candidate := range cmds {
|
|
if candidate.Name == name {
|
|
return candidate
|
|
}
|
|
}
|
|
t.Fatalf("commands = %#v", cmds)
|
|
return bonfire.Command{}
|
|
}
|
|
|
|
type emailArg struct{ email string }
|
|
|
|
func (e emailArg) Args() []string { return []string{e.email} }
|
|
|
|
func (e emailArg) Argument(name string) (string, bool) {
|
|
if name == "email" {
|
|
return e.email, true
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
func (e emailArg) Flag(string) (string, bool) { return "", false }
|
|
|
|
func (e emailArg) Flags(string) []string { return nil }
|
|
|
|
type tokenCreateArg struct{ email, name, scopes string }
|
|
|
|
func (a tokenCreateArg) Args() []string { return []string{a.email} }
|
|
|
|
func (a tokenCreateArg) Argument(name string) (string, bool) {
|
|
if name == "email" {
|
|
return a.email, true
|
|
}
|
|
return "", false
|
|
}
|
|
|
|
func (a tokenCreateArg) Flag(name string) (string, bool) {
|
|
switch name {
|
|
case "name":
|
|
return a.name, true
|
|
case "scopes":
|
|
return a.scopes, true
|
|
default:
|
|
return "", false
|
|
}
|
|
}
|
|
|
|
func (a tokenCreateArg) Flags(string) []string { return nil }
|