synapsecns/sanguine

package localmetrics

import (
    "context"
    "github.com/Flaque/filet"
    "github.com/brianvoe/gofakeit/v6"
    "github.com/ory/dockertest/v3"
    "github.com/stretchr/testify/assert"
    "testing"
)

// UIResource exports ui resource for testing.
type UIResource interface {
    GetResource() *dockertest.Resource
    GetUIURL() string
}

func (u uiResource) GetResource() *dockertest.Resource {
    return u.Resource
}

func (u uiResource) GetUIURL() string {
    return u.uiURL
}

var _ UIResource = &uiResource{}

// TestJaeger exports jaeger for testing.
type TestJaeger interface {
    StartJaegerServer(ctx context.Context) UIResource
    StartPyroscopeServer(ctx context.Context) UIResource
    GetRunID() string
    GetPool() *dockertest.Pool
}

// StartTestServer starts a local jaeger server for testing.
// this will return a TestJaeger that can be used to start other servers.
func StartTestServer(ctx context.Context, tb testing.TB) TestJaeger {
    tb.Helper()

    tj := startServer(ctx, tb, WithPyroscopeJaeger(true))
    return &exportedJaeger{
        tj: tj,
    }
}

type exportedJaeger struct {
    tj *testJaeger
}

func NewTestJaeger(tb testing.TB, opts ...Option) TestJaeger {
    tb.Helper()

    logDir := filet.TmpDir(tb, "")
    pool, err := dockertest.NewPool("")
    assert.NoError(tb, err)

    cfg := makeConfig(opts)

    return &exportedJaeger{
        tj: &testJaeger{
            tb:     tb,
            logDir: logDir,
            pool:   pool,
            runID:  gofakeit.UUID(),
            cfg:    cfg,
        },
    }
}

func (e *exportedJaeger) GetPool() *dockertest.Pool {
    return e.tj.pool
}

func (e *exportedJaeger) GetRunID() string {
    return e.tj.runID
}

func (e *exportedJaeger) StartJaegerServer(ctx context.Context) UIResource {
    return e.tj.StartJaegerServer(ctx)
}

func (e *exportedJaeger) StartPyroscopeServer(ctx context.Context) UIResource {
    return e.tj.StartPyroscopeServer(ctx)
}

const RunIDLabel = runIDLabel
const AppLabel = appLabel