synapsecns/sanguine

package client_test

import (
    "fmt"
    "testing"

    "github.com/Flaque/filet"
    . "github.com/stretchr/testify/assert"
    "github.com/stretchr/testify/suite"
    "github.com/synapsecns/sanguine/core"
    "github.com/synapsecns/sanguine/core/metrics"
    "github.com/synapsecns/sanguine/core/metrics/localmetrics"
    "github.com/synapsecns/sanguine/core/testsuite"
    "github.com/synapsecns/sanguine/services/scribe/client"
    pbscribe "github.com/synapsecns/sanguine/services/scribe/grpc/types/types/v1"
    "github.com/synapsecns/sanguine/services/scribe/metadata"
    "google.golang.org/grpc"
    "google.golang.org/grpc/credentials/insecure"
)

// ClientSuite defines the basic test suite.
type ClientSuite struct {
    *testsuite.TestSuite
    db      string
    dbPath  string
    metrics metrics.Handler
}

// NewTestSuite creates a new test suite and performs some basic checks afterward.
// Every test suite in the synapse library should inherit from this suite and override where necessary.
func NewTestSuite(tb testing.TB) *ClientSuite {
    tb.Helper()
    return &ClientSuite{
        TestSuite: testsuite.NewTestSuite(tb),
    }
}

func (g *ClientSuite) SetupTest() {
    g.TestSuite.SetupTest()
    g.dbPath = filet.TmpDir(g.T(), "")
    g.db = "sqlite"
}

func (g *ClientSuite) SetupSuite() {
    g.TestSuite.SetupSuite()

    // don't use metrics on ci for integration tests
    isCI := core.GetEnvBool("CI", false)
    useMetrics := !isCI
    metricsHandler := metrics.Null

    if useMetrics {
        localmetrics.SetupTestJaeger(g.GetSuiteContext(), g.T())
        metricsHandler = metrics.Jaeger
    }

    var err error
    g.metrics, err = metrics.NewByType(g.GetSuiteContext(), metadata.BuildInfo(), metricsHandler)
    g.Require().Nil(err)
}

func (g *ClientSuite) TestEmbeddedScribe() {
    embeddedClient := client.NewEmbeddedScribe(g.db, g.dbPath, g.metrics)

    go func() {
        Nil(g.T(), embeddedClient.Start(g.GetSuiteContext()))
    }()

    g.Eventually(func() bool {
        conn, err := grpc.DialContext(g.GetSuiteContext(), fmt.Sprintf("%s:%d", embeddedClient.URL, embeddedClient.Port), grpc.WithTransportCredentials(insecure.NewCredentials()))
        if err != nil {
            return false
        }

        grpcClient := pbscribe.NewScribeServiceClient(conn)

        healthCheck, err := grpcClient.Check(g.GetSuiteContext(), &pbscribe.HealthCheckRequest{})
        if err != nil {
            return false
        }

        return healthCheck.GetStatus() == pbscribe.HealthCheckResponse_SERVING
    })
}

func TestClientSuite(t *testing.T) {
    suite.Run(t, NewTestSuite(t))
}