mirror of
https://github.com/bpg/terraform-provider-proxmox.git
synced 2025-07-01 11:02:59 +00:00
95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
/*
|
|
* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
*/
|
|
|
|
package test
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"regexp"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/terraform-plugin-testing/helper/resource"
|
|
"github.com/hashicorp/terraform-plugin-testing/terraform"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
// ResourceAttributes is a helper function to test resource attributes.
|
|
func ResourceAttributes(res string, attrs map[string]string) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
for k, v := range attrs {
|
|
if v == "" {
|
|
if err := resource.TestCheckResourceAttr(res, k, "")(s); err != nil {
|
|
return fmt.Errorf("expected '%s' to be empty: %w", k, err)
|
|
}
|
|
|
|
continue
|
|
}
|
|
|
|
if err := resource.TestCheckResourceAttrWith(res, k, func(got string) error {
|
|
match, err := regexp.Match(v, []byte(got)) //nolint:mirror
|
|
if err != nil {
|
|
return fmt.Errorf("error matching '%s': %w", v, err)
|
|
}
|
|
if !match {
|
|
return fmt.Errorf("expected '%s' to match '%s'", got, v)
|
|
}
|
|
return nil
|
|
})(s); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// NoResourceAttributesSet is a helper function to test that no resource attributes are set.
|
|
func NoResourceAttributesSet(res string, attrs []string) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
for _, k := range attrs {
|
|
if err := resource.TestCheckNoResourceAttr(res, k)(s); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
// ResourceAttributesSet is a helper function to test that all resource attributes are set.
|
|
func ResourceAttributesSet(res string, attrs []string) resource.TestCheckFunc {
|
|
return func(s *terraform.State) error {
|
|
for _, k := range attrs {
|
|
if err := resource.TestCheckResourceAttrSet(res, k)(s); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
}
|
|
|
|
func CreateTempFile(t *testing.T, namePattern string, content string) *os.File {
|
|
t.Helper()
|
|
|
|
f, err := os.CreateTemp(t.TempDir(), namePattern)
|
|
require.NoError(t, err)
|
|
|
|
_, err = f.WriteString(content)
|
|
require.NoError(t, err)
|
|
|
|
defer func(f *os.File) {
|
|
_ = f.Close()
|
|
}(f)
|
|
|
|
t.Cleanup(func() {
|
|
_ = os.Remove(f.Name())
|
|
})
|
|
|
|
return f
|
|
}
|