integration_azure_test.go

package rockset_test

import (
	"testing"

	"github.com/stretchr/testify/suite"

	"github.com/rockset/rockset-go-client"
	"github.com/rockset/rockset-go-client/internal/test"
)

type AzureIntegrationsSuite struct {
	suite.Suite
	rc               *rockset.RockClient
	integrationName  string
	connectionString string
}

func (s *AzureIntegrationsSuite) TestAzureBlob() {
	ctx := test.Context()

	_, err := s.rc.CreateAzureBlobStorageIntegration(ctx, s.integrationName, s.connectionString)
	s.Require().NoError(err)

	integration, err := s.rc.GetIntegration(ctx, s.integrationName)
	s.Require().NoError(err)
	s.NotNil(integration)

	err = s.rc.DeleteIntegration(ctx, s.integrationName)
	s.Require().NoError(err)
}

func (s *AzureIntegrationsSuite) TearDownSuite() {
	ctx := test.Context()

	// cleanup the integration if it is still around
	_, err := s.rc.GetIntegration(ctx, s.integrationName)
	if err == nil {
		s.T().Logf("deleting lingering azure integration: %s", s.integrationName)
		err = s.rc.DeleteIntegration(ctx, s.integrationName)
		s.Require().NoError(err)
	}
}

func TestAzureIntegrations(t *testing.T) {
	connectionString := test.SkipUnlessEnvSet(t, "AZURE_CONNECTION_STRING")
	rc, randomName := vcrTestClient(t, t.Name())

	s := AzureIntegrationsSuite{
		rc:               rc,
		integrationName:  randomName("azure"),
		connectionString: connectionString,
	}
	suite.Run(t, &s)
}