cucumber

type Encoding struct {
	Name      string
	Marshal   func(any) ([]byte, error)
	Unmarshal func([]byte, any) error
}

type ScenarioSetupFunc func(ctx context.Context, s *TestScenario, sc *godog.Scenario) (func(context.Context) error, error)

type TaskRow struct {
	ID         string
	TaskType   string
	TaskBody   string
	RetryAt    time.Time
	RetryCount int
	LastError  *string
}

type TestDB interface {
	// ClearAll wipes all data (called before each scenario).
	ClearAll(ctx context.Context) error
	// ResolveGroupID returns the conversation_group_id for a given conversation ID.
	ResolveGroupID(ctx context.Context, conversationID string) (string, error)
	// SetConversationEntriesCreatedAt forces all entries in a conversation group to the same created_at timestamp.
	SetConversationEntriesCreatedAt(ctx context.Context, conversationID string, createdAt time.Time) error
	// ExecSQL runs a raw SQL query and returns rows as maps. Non-SQL backends return (nil, nil) to skip assertions.
	ExecSQL(ctx context.Context, query string) ([]map[string]interface{}, error)
	// ExecMongoQuery runs a MongoDB query spec (JSON payload) and returns rows as maps.
	// Non-Mongo backends return (nil, nil) to skip assertions.
	ExecMongoQuery(ctx context.Context, query string) ([]map[string]interface{}, error)
	// ArchiveConversation sets archived_at on the conversation and its group to `days` ago.
	ArchiveConversation(ctx context.Context, conversationID string, days int) error
	// ArchiveConversationOnly sets archived_at on the conversation only, without touching the group.
	ArchiveConversationOnly(ctx context.Context, conversationID string, days int) error
	// Task queue operations for task-queue BDD feature.
	DeleteAllTasks(ctx context.Context) error
	CreateTask(ctx context.Context, id, taskType, body string) error
	CreateFailedTask(ctx context.Context, id, taskType, body string) error
	ClaimReadyTasks(ctx context.Context, limit int) ([]TaskRow, error)
	DeleteTask(ctx context.Context, id string) error
	FailTask(ctx context.Context, id, errMsg string) error
	GetTask(ctx context.Context, id string) (*TaskRow, error)
	CountTasks(ctx context.Context) (int, error)
}

type TestScenario struct {
	Suite       *TestSuite
	ScenarioUID string
	CurrentUser string
	PathPrefix  string
	APIURL      string
	DB          TestDB
	Extra       map[string]interface{}

	Variables map[string]interface{}
	Users     map[string]*TestUser
	// contains filtered or unexported fields
}

type TestSession struct {
	TestUser  *TestUser
	Client    *http.Client
	Resp      *http.Response
	RespBytes []byte

	Header            http.Header
	EventStream       bool
	EventStreamEvents chan interface{}
	// contains filtered or unexported fields
}

type TestSuite struct {
	Context  interface{} // opaque application context
	APIURL   string
	Mu       sync.Mutex
	TestingT *testing.T
	Extra    map[string]interface{} // additional test-scoped objects (e.g. mock servers)
	DB       TestDB                 // injected by test runner for store-specific operations
	// ScenarioSetup optionally provisions per-scenario runtime state such as
	// dedicated servers, datastore handles, and extra values.
	ScenarioSetup ScenarioSetupFunc
}

type TestUser struct {
	Name    string
	Subject string // Bearer token value
	Mu      sync.Mutex
}