main_test.go

package main

import (
	"context"
	"os"
	"testing"

	"github.com/civo/civo-csi/pkg/driver"
	"github.com/civo/civogo"
	"github.com/kubernetes-csi/csi-test/v4/pkg/sanity"
	"github.com/rs/zerolog"
	"github.com/rs/zerolog/log"
	"golang.org/x/sync/errgroup"
)

// TestCivoCSI runs the Sanity test suite
func TestCivoCSI(t *testing.T) {
	zerolog.TimeFieldFormat = zerolog.TimeFormatUnix
	log.Logger = log.Output(zerolog.ConsoleWriter{Out: os.Stderr})

	fc, _ := civogo.NewFakeClient()
	d, _ := driver.NewTestDriver(fc)

	ctx, cancel := context.WithCancel(context.Background())

	os.Setenv("REGION", "TESTING1")
	os.Setenv("NAMESPACE", "default")
	cluster, _ := d.CivoClient.NewKubernetesClusters(&civogo.KubernetesClusterConfig{
		Name:           "test",
		NumTargetNodes: 1,
	})
	os.Setenv("NODE_ID", cluster.Pools[0].Instances[0].ID)
	d.ClusterID = cluster.ID
	os.Setenv("CLUSTER_ID", cluster.ID)

	var eg errgroup.Group
	eg.Go(func() error {
		return d.Run(ctx)
	})

	config := sanity.NewTestConfig()
	if err := os.RemoveAll(config.TargetPath); err != nil {
		t.Fatalf("failed to delete target path %s: %s", config.TargetPath, err)
	}
	if err := os.RemoveAll(config.StagingPath); err != nil {
		t.Fatalf("failed to delete staging path %s: %s", config.StagingPath, err)
	}

	config.Address = d.SocketFilename

	sanity.Test(t, config)

	cancel()
	if err := eg.Wait(); err != nil {
		t.Errorf("driver run failed: %s", err)
	}
}