synapsecns/sanguine

package screener

import (
    "context"
    "fmt"

    "github.com/synapsecns/sanguine/contrib/screener-api/chainalysis"
    "github.com/synapsecns/sanguine/contrib/screener-api/config"
    "github.com/synapsecns/sanguine/core/metrics"
)

type TestScreener interface {
    Screener
    SetClient(client chainalysis.Client)
}

func NewTestScreener(ctx context.Context, cfg config.Config, metricHandler metrics.Handler) (_ TestScreener, err error) {
    screener, err := NewScreener(ctx, cfg, metricHandler)
    if err != nil {
        return nil, fmt.Errorf("could not create screener: %w", err)
    }

    ts, ok := screener.(TestScreener)
    if !ok {
        return nil, fmt.Errorf("could not cast screener to test screener")
    }

    return ts, nil
}

func (s *screenerImpl) SetClient(client chainalysis.Client) {
    s.client = client
}