temporal/chasm/node_pure_task_mock.go

package chasm

import (
	"context"
	"sync"
)

// MockNodePureTask is a lightweight manual mock for the NodePureTask interface.
// Methods may be stubbed by assigning the corresponding Handle fields. Call history
// is recorded in the struct fields (thread-safe).
type MockNodePureTask struct {
	HandleExecutePureTask func(baseCtx context.Context, taskAttributes TaskAttributes, taskInstance any) (bool, error)

	mu           sync.Mutex
	ExecuteCalls []struct {
		BaseCtx    context.Context
		Attributes TaskAttributes
		Task       any
	}
}

func (m *MockNodePureTask) ExecutePureTask(
	baseCtx context.Context,
	taskAttributes TaskAttributes,
	taskInstance any,
) (bool, error) {
	if m.HandleExecutePureTask != nil {
		ok, err := m.HandleExecutePureTask(baseCtx, taskAttributes, taskInstance)

		m.mu.Lock()
		m.ExecuteCalls = append(m.ExecuteCalls, struct {
			BaseCtx    context.Context
			Attributes TaskAttributes
			Task       any
		}{BaseCtx: baseCtx, Attributes: taskAttributes, Task: taskInstance})
		m.mu.Unlock()

		return ok, err
	}

	m.mu.Lock()
	m.ExecuteCalls = append(m.ExecuteCalls, struct {
		BaseCtx    context.Context
		Attributes TaskAttributes
		Task       any
	}{BaseCtx: baseCtx, Attributes: taskAttributes, Task: taskInstance})
	m.mu.Unlock()

	return false, nil
}