package lagoon

import (
	"database/sql"
	"errors"
	"testing"

	"git.golem15.com/golem15/summercms/modules/backpack"
	"gorm.io/gorm"
)

// TestOnDatabaseAfterActivate covers the production boot order: plugins
// register database hooks at Boot, before serve publishes the database.
func TestOnDatabaseAfterActivate(t *testing.T) {
	db, _ := dedicatedDB(t, "lagoon_ondatabase")
	gdb, err := Use(t.Context(), db)
	if err != nil {
		t.Fatal(err)
	}

	t.Run("before_publish_runs_on_publish", func(t *testing.T) {
		app := backpack.New(nil)
		calls := 0
		var gotSQL *sql.DB
		var gotGDB *gorm.DB
		if err := OnDatabase(app, func(s *sql.DB, g *gorm.DB) error {
			calls++
			gotSQL, gotGDB = s, g
			return nil
		}); err != nil {
			t.Fatal(err)
		}
		if calls != 0 {
			t.Fatalf("hook ran %d times before Publish", calls)
		}
		if err := Publish(app, db, gdb); err != nil {
			t.Fatal(err)
		}
		if calls != 1 || gotSQL != db || gotGDB != gdb {
			t.Fatalf("after Publish: calls=%d sql=%v gdb=%v", calls, gotSQL == db, gotGDB == gdb)
		}
	})

	t.Run("after_publish_runs_now", func(t *testing.T) {
		app := backpack.New(nil)
		if err := Publish(app, db, gdb); err != nil {
			t.Fatal(err)
		}
		calls := 0
		if err := OnDatabase(app, func(*sql.DB, *gorm.DB) error { calls++; return nil }); err != nil {
			t.Fatal(err)
		}
		if calls != 1 {
			t.Fatalf("hook ran %d times, want 1", calls)
		}
	})

	t.Run("gorm_only_published_runs_now", func(t *testing.T) {
		app := backpack.New(nil)
		if err := app.Publish(gdb); err != nil {
			t.Fatal(err)
		}
		var gotSQL *sql.DB
		if err := OnDatabase(app, func(s *sql.DB, _ *gorm.DB) error { gotSQL = s; return nil }); err != nil {
			t.Fatal(err)
		}
		if gotSQL != db {
			t.Fatal("hook did not receive the pool behind the published *gorm.DB")
		}
	})

	t.Run("errors_surface", func(t *testing.T) {
		boom := errors.New("boom")
		app := backpack.New(nil)
		if err := OnDatabase(app, func(*sql.DB, *gorm.DB) error { return boom }); err != nil {
			t.Fatal(err)
		}
		if err := Publish(app, db, gdb); !errors.Is(err, boom) {
			t.Fatalf("Publish err = %v, want the hook error", err)
		}
		if err := OnDatabase(app, func(*sql.DB, *gorm.DB) error { return boom }); !errors.Is(err, boom) {
			t.Fatalf("OnDatabase err = %v, want the hook error", err)
		}
	})
}