hooks.go

package testcase

import (
	"runtime"
	"sync"
	"testing"

	"go.llib.dev/testcase/internal/caller"
)

const hookWarning = `you cannot create spec hooks after you used describe/when/and/then,
unless you create a new spec with the previously mentioned calls`

type hook struct {
	Block func(*T) func()
	Frame runtime.Frame
}

type hookOnce struct {
	Block  func(testing.TB)
	Frame  runtime.Frame
	DoOnce func(testing.TB)
}

// Before give you the ability to run a block before each test case.
// This is ideal for doing clean ahead before each test case.
// The received *testing.T object is the same as the Test block *testing.T object
// This hook applied to this scope and anything that is nested from here.
// All setup block is stackable.
func (spec *Spec) Before(beforeBlock func(t *T)) {
	spec.H().Helper()
	spec.Around(func(t *T) func() {
		spec.H().Helper()
		beforeBlock(t)
		return func() {}
	})
}

// After give you the ability to run a block after each test case.
// This is ideal for running cleanups.
// The received *testing.T object is the same as the Then block *testing.T object
// This hook applied to this scope and anything that is nested from here.
// All setup block is stackable.
func (spec *Spec) After(afterBlock func(t *T)) {
	spec.H().Helper()
	spec.Around(func(t *T) func() {
		spec.H().Helper()
		return func() {
			spec.H().Helper()
			afterBlock(t)
		}
	})
}

// Around give you the ability to create "Before" setup for each test case,
// with the additional ability that the returned function will be deferred to run after the Then block is done.
// This is ideal for setting up mocks, and then return the assertion request calls in the return func.
// This hook applied to this scope and anything that is nested from here.
// All setup block is stackable.
//
// Deprecated: use Spec.Before with T.Cleanup or Spec.Before with T.Defer instead
func (spec *Spec) Around(block func(*T) func()) {
	spec.H().Helper()
	frame, _ := caller.GetFrame()
	spec.modify(func(spec *Spec) {
		spec.H().Helper()
		if spec.immutable {
			spec.testingTB.Fatal(hookWarning)
		}
		spec.hooks.Around = append(spec.hooks.Around, hook{
			Block: block,
			Frame: frame,
		})
	})
}

// BeforeAll give you the ability to create a hook
// that runs only once before the test cases.
func (spec *Spec) BeforeAll(blk func(tb testing.TB)) {
	spec.H().Helper()
	frame, _ := caller.GetFrame()
	spec.modify(func(spec *Spec) {
		spec.H().Helper()

		if spec.immutable {
			spec.testingTB.Fatal(hookWarning)
		}

		var onCall sync.Once
		var beforeAll = func(tb testing.TB) {
			onCall.Do(func() { blk(tb) })
		}

		h := hookOnce{
			DoOnce: beforeAll,
			Block:  blk,
			Frame:  frame,
		}

		spec.hooks.BeforeAll = append(spec.hooks.BeforeAll, h)
	})
}

func (spec *Spec) AfterAll(blk func(tb testing.TB)) {
	spec.H().Helper()
	frame, _ := caller.GetFrame()
	spec.modify(func(spec *Spec) {
		spec.H().Helper()

		if spec.immutable {
			spec.testingTB.Fatal(hookWarning)
		}

		var onCall sync.Once
		var beforeAll = func(tb testing.TB) {
			onCall.Do(func() { blk(tb) })
		}

		h := hookOnce{
			DoOnce: beforeAll,
			Block:  blk,
			Frame:  frame,
		}

		spec.hooks.AfterAll = append(spec.hooks.AfterAll, h)
	})
}