fix: open Postgres for one-shot console commands

console_test.go

@@ -7,8 +7,10 @@ import (
 	"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) {
@@ -16,17 +18,7 @@ func TestRequirePasswordChange(t *testing.T) {
 	user := insertUser(t, gdb, "alice@example.com", "correct-horse")
 	p := &Plugin{app: app}
 	var _ pact.HasCommands = p
-	cmds := p.Commands()
-	var cmd bonfire.Command
-	for _, candidate := range cmds {
-		if candidate.Name == "user:require-password-change" {
-			cmd = candidate
-			break
-		}
-	}
-	if cmd.Name == "" {
-		t.Fatalf("commands = %#v", cmds)
-	}
+	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 {
@@ -47,6 +39,55 @@ func TestRequirePasswordChange(t *testing.T) {
 	}
 }
 
+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} }
@@ -61,3 +102,27 @@ func (e emailArg) Argument(name string) (string, bool) {
 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 }