117 lines
2.9 KiB
Go
117 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{
|
|
IsUnitTest: true,
|
|
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{
|
|
IsUnitTest: true,
|
|
Providers: testProviders,
|
|
Steps: []resource.TestStep{
|
|
{
|
|
Config: fmt.Sprintf(testTemplate, input),
|
|
Check: check,
|
|
},
|
|
},
|
|
})
|
|
}
|
|
|
|
var testTemplate = `
|
|
%s
|
|
|
|
output "rendered" {
|
|
value = "${data.ignition_config.test.rendered}"
|
|
}
|
|
|
|
`
|