terraform/builtin/providers/ignition/resource_ignition_config_te...

115 lines
2.9 KiB
Go

package ignition
import (
"encoding/json"
"fmt"
"regexp"
"testing"
"github.com/coreos/ignition/config/types"
"github.com/hashicorp/terraform/helper/resource"
"github.com/hashicorp/terraform/terraform"
)
func TestIngnitionFileReplace(t *testing.T) {
testIgnition(t, `
data "ignition_config" "test" {
replace {
source = "foo"
verification = "sha512-0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
}
}
`, func(c *types.Config) error {
r := c.Ignition.Config.Replace
if r == nil {
return fmt.Errorf("unable to find replace config")
}
if r.Source.String() != "foo" {
return fmt.Errorf("config.replace.source, found %q", r.Source)
}
if r.Verification.Hash.Sum != "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef" {
return fmt.Errorf("config.replace.verification, found %q", r.Verification.Hash)
}
return nil
})
}
func TestIngnitionFileAppend(t *testing.T) {
testIgnition(t, `
data "ignition_config" "test" {
append {
source = "foo"
verification = "sha512-0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
}
append {
source = "foo"
verification = "sha512-0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef"
}
}
`, func(c *types.Config) error {
a := c.Ignition.Config.Append
if len(a) != 2 {
return fmt.Errorf("unable to find append config, expected 2")
}
if a[0].Source.String() != "foo" {
return fmt.Errorf("config.replace.source, found %q", a[0].Source)
}
if a[0].Verification.Hash.Sum != "0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef0123456789abcdef" {
return fmt.Errorf("config.replace.verification, found %q", a[0].Verification.Hash)
}
return nil
})
}
func testIgnitionError(t *testing.T, input string, expectedErr *regexp.Regexp) {
resource.Test(t, resource.TestCase{
Providers: testProviders,
Steps: []resource.TestStep{
{
Config: fmt.Sprintf(testTemplate, input),
ExpectError: expectedErr,
},
},
})
}
func testIgnition(t *testing.T, input string, assert func(*types.Config) error) {
check := func(s *terraform.State) error {
got := s.RootModule().Outputs["rendered"].Value.(string)
c := &types.Config{}
err := json.Unmarshal([]byte(got), c)
if err != nil {
return err
}
return assert(c)
}
resource.Test(t, resource.TestCase{
Providers: testProviders,
Steps: []resource.TestStep{
{
Config: fmt.Sprintf(testTemplate, input),
Check: check,
},
},
})
}
var testTemplate = `
%s
output "rendered" {
value = "${data.ignition_config.test.rendered}"
}
`