e7164c0460
* add DNS record * revamp tests * lint * cleanup * feat dns test * chore: add DNS Record tests * linting * f
95 lines
2.5 KiB
Go
95 lines
2.5 KiB
Go
package acctest
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/filipowm/go-unifi/unifi"
|
|
"github.com/filipowm/terraform-provider-unifi/internal/provider"
|
|
pt "github.com/filipowm/terraform-provider-unifi/internal/provider/testing"
|
|
"github.com/hashicorp/go-version"
|
|
"github.com/hashicorp/terraform-plugin-framework/providerserver"
|
|
"github.com/hashicorp/terraform-plugin-go/tfprotov5"
|
|
"github.com/hashicorp/terraform-plugin-go/tfprotov6"
|
|
"github.com/hashicorp/terraform-plugin-mux/tf5to6server"
|
|
"github.com/hashicorp/terraform-plugin-mux/tf6muxserver"
|
|
"github.com/hashicorp/terraform-plugin-sdk/v2/helper/schema"
|
|
"github.com/hashicorp/terraform-plugin-testing/helper/resource"
|
|
"os"
|
|
"testing"
|
|
)
|
|
|
|
type providersMap map[string]func() (tfprotov6.ProviderServer, error)
|
|
|
|
var (
|
|
providers = createProviders()
|
|
testClient unifi.Client
|
|
)
|
|
|
|
type Steps []resource.TestStep
|
|
|
|
type AcceptanceTestCase struct {
|
|
CheckDestroy resource.TestCheckFunc
|
|
VersionConstraint string
|
|
MinVersion *version.Version
|
|
PreCheck func()
|
|
Steps Steps
|
|
}
|
|
|
|
func AcceptanceTest(t *testing.T, testCase AcceptanceTestCase) {
|
|
t.Helper()
|
|
if len(testCase.Steps) == 0 {
|
|
t.Fatal("missing test steps")
|
|
}
|
|
resource.ParallelTest(t, resource.TestCase{
|
|
PreCheck: func() {
|
|
pt.PreCheck(t)
|
|
if testCase.VersionConstraint != "" {
|
|
PreCheckVersionConstraint(t, testCase.VersionConstraint)
|
|
}
|
|
if testCase.MinVersion != nil {
|
|
PreCheckMinVersion(t, testCase.MinVersion)
|
|
}
|
|
if testCase.PreCheck != nil {
|
|
testCase.PreCheck()
|
|
}
|
|
},
|
|
ProtoV6ProviderFactories: providers,
|
|
CheckDestroy: testCase.CheckDestroy,
|
|
Steps: testCase.Steps,
|
|
})
|
|
}
|
|
|
|
func TestMain(m *testing.M) {
|
|
providers = createProviders()
|
|
os.Exit(pt.Run(m, func(env *pt.TestEnvironment) {
|
|
testClient = env.Client
|
|
}))
|
|
}
|
|
|
|
func createProviders() providersMap {
|
|
ctx := context.Background()
|
|
// Init mux servers
|
|
return map[string]func() (tfprotov6.ProviderServer, error){
|
|
"unifi": func() (tfprotov6.ProviderServer, error) {
|
|
return tf6muxserver.NewMuxServer(ctx,
|
|
providerserver.NewProtocol6(provider.NewV2("acctestv2")()),
|
|
func() tfprotov6.ProviderServer {
|
|
sdkV2Provider, err := tf5to6server.UpgradeServer(
|
|
ctx,
|
|
func() tfprotov5.ProviderServer {
|
|
return schema.NewGRPCProviderServer(
|
|
provider.New("acctestv1")(),
|
|
)
|
|
},
|
|
)
|
|
if err != nil {
|
|
panic(fmt.Errorf("failed to create test providers: %w", err))
|
|
}
|
|
|
|
return sdkV2Provider
|
|
},
|
|
)
|
|
},
|
|
}
|
|
}
|