callbacks_test.go - main - githubmirror/pop

callbacks_test.go
89 строк2 KB
aeneasr
Add `AfterEagerFind()`
02 янв 2023, 18:39
02 янв 2023, 18:393596719
package pop

import (
	"testing"

	"github.com/stretchr/testify/require"
)

func Test_Callbacks(t *testing.T) {
	if PDB == nil {
		t.Skip("skipping integration tests")
	}
	transaction(func(tx *Connection) {
		r := require.New(t)

		user := &CallbacksUser{
			BeforeS: "BS",
			BeforeC: "BC",
			BeforeU: "BU",
			BeforeD: "BD",
			BeforeV: "BV",
			AfterS:  "AS",
			AfterC:  "AC",
			AfterU:  "AU",
			AfterD:  "AD",
			AfterF:  "AF",
		}

		r.NoError(tx.Save(user))

		r.Equal("BeforeSave", user.BeforeS)
		r.Equal("BeforeCreate", user.BeforeC)
		r.Equal("AfterSave", user.AfterS)
		r.Equal("AfterCreate", user.AfterC)
		r.Equal("BU", user.BeforeU)
		r.Equal("AU", user.AfterU)

		r.NoError(tx.Update(user))

		r.Equal("BeforeUpdate", user.BeforeU)
		r.Equal("AfterUpdate", user.AfterU)
		r.Equal("BD", user.BeforeD)
		r.Equal("AD", user.AfterD)

		r.Equal("AF", user.AfterF)
		r.NoError(tx.Find(user, user.ID))
		r.Equal("AfterFind", user.AfterF)
		r.Empty(user.AfterEF)

		r.NoError(tx.Eager().Find(user, user.ID))
		r.Equal("AfterEagerFind", user.AfterEF)

		r.NoError(tx.Destroy(user))

		r.Equal("BeforeDestroy", user.BeforeD)
		r.Equal("AfterDestroy", user.AfterD)

		verrs, err := tx.ValidateAndSave(user)
		r.False(verrs.HasAny())
		r.NoError(err)
		r.Equal("BeforeValidate", user.BeforeV)
	})
}

func Test_Callbacks_on_Slice(t *testing.T) {
	if PDB == nil {
		t.Skip("skipping integration tests")
	}
	transaction(func(tx *Connection) {
		r := require.New(t)
		for i := 0; i < 2; i++ {
			r.NoError(tx.Create(&CallbacksUser{}))
		}

		users := CallbacksUsers{}
		r.NoError(tx.All(&users))
		r.Len(users, 2)
		for _, u := range users {
			r.Equal("AfterFind", u.AfterF)
			r.Empty(u.AfterEF)
		}

		r.NoError(tx.Eager().All(&users))
		r.Len(users, 2)
		for _, u := range users {
			r.Equal("AfterEagerFind", u.AfterEF)
		}
	})
}