synology-csi/test/sanity/sanity_test.go

package sanitytest

import (
	"fmt"
	"io/ioutil"
	"os"
	"testing"

	sanity "github.com/kubernetes-csi/csi-test/v4/pkg/sanity"

	"github.com/SynologyOpenSource/synology-csi/pkg/driver"
	"github.com/SynologyOpenSource/synology-csi/pkg/dsm/common"
	"github.com/SynologyOpenSource/synology-csi/pkg/dsm/service"
)

const (
	ConfigPath = "./../../config/client-info.yml"
)

func TestSanity(t *testing.T) {
	nodeID := "CSINode"

	endpointFile, err := ioutil.TempFile("", "csi-gcs.*.sock")
	if err != nil {
		t.Fatal(err)
	}
	defer os.Remove(endpointFile.Name())

	stagingPath, err := ioutil.TempDir("", "csi-gcs-staging")
	if err != nil {
		t.Fatal(err)
	}
	os.Remove(stagingPath)

	targetPath, err := ioutil.TempDir("", "csi-gcs-target")
	if err != nil {
		t.Fatal(err)
	}
	os.Remove(targetPath)

	info, err := common.LoadConfig(ConfigPath)
	if err != nil {
		t.Fatal(fmt.Sprintf("Failed to read config: %v", err))
	}

	dsmService := service.NewDsmService()

	for _, client := range info.Clients {
		err := dsmService.AddDsm(client)
		if err != nil {
			fmt.Printf("Failed to add DSM: %s, error: %v\n", client.Host, err)
		}
	}

	if dsmService.GetDsmsCount() == 0 {
		t.Fatal("No available DSM.")
	}
	defer dsmService.RemoveAllDsms()

	endpoint := "unix://" + endpointFile.Name()
	drv, err := driver.NewControllerAndNodeDriver(nodeID, endpoint, dsmService)
	if err != nil {
		t.Fatal(fmt.Sprintf("Failed to create driver: %v\n", err))
	}
	drv.Activate()

	// Set configuration options as needed
	testConfig := sanity.NewTestConfig()
	testConfig.TargetPath = targetPath
	testConfig.StagingPath = stagingPath
	testConfig.Address = endpoint
	// Run test
	sanity.Test(t, testConfig)
}