synapsecns/sanguine

package localmetrics_test

import (
    "fmt"
    "github.com/ory/dockertest/v3"
    "github.com/ory/dockertest/v3/docker"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core/metrics/localmetrics"
    "github.com/synapsecns/sanguine/core/testsuite"
    "testing"
)

// LocalServerSuite defines the basic test suite.
type LocalServerSuite struct {
    *testsuite.TestSuite
    client *docker.Client
}

func NewLocalServerSuite(tb testing.TB) *LocalServerSuite {
    tb.Helper()
    return &LocalServerSuite{
        TestSuite: testsuite.NewTestSuite(tb),
    }
}

func (l *LocalServerSuite) SetupSuite() {
    l.TestSuite.SetupSuite()
    pool, err := dockertest.NewPool("")
    l.Require().NoError(err)
    l.client = pool.Client
}

func TestCommonSuite(t *testing.T) {
    suite.Run(t, NewLocalServerSuite(t))
}

func (l *LocalServerSuite) hasContainerWithRunID(runID string) bool {
    containers := l.ContainersWithLabel(localmetrics.RunIDLabel, runID)
    return len(containers) > 0
}

func (l *LocalServerSuite) ContainersWithLabel(key, value string) []docker.APIContainers {
    containers, err := l.client.ListContainers(docker.ListContainersOptions{
        All: true,
        Filters: map[string][]string{
            "label": {fmt.Sprintf("%s=%s", key, value)},
        },
    })

    l.Require().NoError(err)
    return containers
}