Mainflux.mainflux/readers/influxdb/setup_test.go

package influxdb_test

import (
	"context"
	"fmt"
	"log"
	"os"
	"os/signal"
	"syscall"
	"testing"
	"time"

	influxdata "github.com/influxdata/influxdb-client-go/v2"
	mainflux_log "github.com/mainflux/mainflux/logger"
	dockertest "github.com/ory/dockertest/v3"
)

var (
	testLog, _ = mainflux_log.New(os.Stdout, mainflux_log.Info.String())
	address    string
)

const (
	dbToken       = "test-token"
	dbOrg         = "test-org"
	dbAdmin       = "test-admin"
	dbPass        = "test-password"
	dbBucket      = "test-bucket"
	dbInitMode    = "setup"
	dbFluxEnabled = "true"
	dbBindAddress = ":8088"
	port          = "8086/tcp"
	broker        = "influxdb"
	brokerVersion = "2.2-alpine"
	poolMaxWait   = 120 * time.Second
)

func TestMain(m *testing.M) {
	pool, err := dockertest.NewPool("")
	if err != nil {
		testLog.Error(fmt.Sprintf("Could not connect to docker: %s", err))
	}

	cfg := []string{
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_MODE=%s", dbInitMode),
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_USERNAME=%s", dbAdmin),
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_PASSWORD=%s", dbPass),
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_ORG=%s", dbOrg),
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_BUCKET=%s", dbBucket),
		fmt.Sprintf("DOCKER_INFLUXDB_INIT_ADMIN_TOKEN=%s", dbToken),
		fmt.Sprintf("INFLUXDB_HTTP_FLUX_ENABLED=%s", dbFluxEnabled),
		fmt.Sprintf("INFLUXDB_BIND_ADDRESS=%s", dbBindAddress),
	}
	container, err := pool.Run(broker, brokerVersion, cfg)
	if err != nil {
		testLog.Error(fmt.Sprintf("Could not start container: %s", err))
	}

	handleInterrupt(m, pool, container)

	address = fmt.Sprintf("%s:%s", "http://localhost", container.GetPort(port))
	pool.MaxWait = poolMaxWait

	if err := pool.Retry(func() error {
		client = influxdata.NewClient(address, dbToken)
		_, err = client.Ready(context.Background())
		return err
	}); err != nil {
		testLog.Error(fmt.Sprintf("Could not connect to docker: %s", err))
	}
	code := m.Run()

	if err := pool.Purge(container); err != nil {
		testLog.Error(fmt.Sprintf("Could not purge container: %s", err))
	}

	os.Exit(code)
}

func handleInterrupt(m *testing.M, pool *dockertest.Pool, container *dockertest.Resource) {
	c := make(chan os.Signal, 2)
	signal.Notify(c, os.Interrupt, syscall.SIGTERM)
	go func() {
		<-c
		if err := pool.Purge(container); err != nil {
			log.Fatalf("Could not purge container: %s", err)
		}
		os.Exit(0)
	}()
}