synapsecns/sanguine

package attestation

import (
    "context"
)

// MockCircleAPI is a mock CircleAPI for testing.
type MockCircleAPI struct {
    attestFunc func(context.Context, string) ([]byte, error)
}

// NewMockCircleAPI creates a new MockCircleAPI.
func NewMockCircleAPI() *MockCircleAPI {
    return &MockCircleAPI{
        attestFunc: func(context.Context, string) ([]byte, error) { return []byte{}, nil },
    }
}

// GetAttestation is a mock GetAttestation.
func (m *MockCircleAPI) GetAttestation(ctx context.Context, txHash string) (attestation []byte, err error) {
    return m.attestFunc(ctx, txHash)
}

// SetGetAttestation overrides the attestFunc.
func (m *MockCircleAPI) SetGetAttestation(attestFunc func(context.Context, string) ([]byte, error)) {
    m.attestFunc = attestFunc
}