360f81b030
This function can be used to check if a Provisioner has been marked for testing within the ACC_TEST_PROVISIONERS environment variable.
While testing I found that the shell acceptance test were also running when trying to run powershell tests.
Before change
```
⇶ ACC_TEST_BUILDERS=amazon-ebs ACC_TEST_PROVISIONERS=powershell go test -v ./provisioner/shell/... -timeout=1h
=== RUN
TestShellProvisioner/testing_amazon-ebs_builder_against_shell_provisioner
2020/04/06 15:18:12 ui: amazon-ebs: output will be in this color.
2020/04/06 15:18:12 ui:
2020/04/06 15:18:12 Build debug mode: false
2020/04/06 15:18:12 Force build: false
2020/04/06 15:18:12 On error:
2020/04/06 15:18:12 Preparing build: amazon-ebs
2020/04/06 15:18:12 Waiting on builds to complete...
2020/04/06 15:18:12 Starting build run: amazon-ebs
2020/04/06 15:18:12 Running builder: amazon-ebs
```
After changes
```
⇶ ACC_TEST_BUILDERS=amazon-ebs ACC_TEST_PROVISIONERS=powershell go test -v ./provisioner/shell/... -timeout=1h
--- SKIP: TestShellProvisioner (0.00s)
provisioners.go:88: Provisioner "shell" not defined in ACC_TEST_PROVISIONERS
```
84 lines
2.1 KiB
Go
84 lines
2.1 KiB
Go
package powershell_test
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"os"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/packer/provisioner/powershell"
|
|
|
|
"github.com/hashicorp/go-uuid"
|
|
"github.com/hashicorp/packer/command"
|
|
"github.com/hashicorp/packer/helper/tests/acc"
|
|
"github.com/hashicorp/packer/packer"
|
|
)
|
|
|
|
const TestProvisionerName = "powershell"
|
|
|
|
func TestPowershellProvisioner_Inline(t *testing.T) {
|
|
acc.TestProvisionersPreCheck(TestProvisionerName, t)
|
|
|
|
testProvisioner := PowershellProvisionerAccTest{"powershell-inline-provisioner.txt"}
|
|
acc.TestProvisionersAgainstBuilders(&testProvisioner, t)
|
|
}
|
|
|
|
func TestPowershellProvisioner_Script(t *testing.T) {
|
|
acc.TestProvisionersPreCheck(TestProvisionerName, t)
|
|
|
|
testProvisioner := PowershellProvisionerAccTest{"powershell-script-provisioner.txt"}
|
|
acc.TestProvisionersAgainstBuilders(&testProvisioner, t)
|
|
}
|
|
|
|
type PowershellProvisionerAccTest struct {
|
|
ConfigName string
|
|
}
|
|
|
|
func (s *PowershellProvisionerAccTest) GetName() string {
|
|
return TestProvisionerName
|
|
}
|
|
|
|
func (s *PowershellProvisionerAccTest) GetConfig() (string, error) {
|
|
filePath := filepath.Join("./test-fixtures", s.ConfigName)
|
|
config, err := os.Open(filePath)
|
|
if err != nil {
|
|
return "", fmt.Errorf("Expected to find %s", filePath)
|
|
}
|
|
defer config.Close()
|
|
|
|
file, err := ioutil.ReadAll(config)
|
|
return string(file), nil
|
|
}
|
|
|
|
func (s *PowershellProvisionerAccTest) GetProvisionerStore() packer.MapOfProvisioner {
|
|
return packer.MapOfProvisioner{
|
|
TestProvisionerName: func() (packer.Provisioner, error) { return &powershell.Provisioner{}, nil },
|
|
}
|
|
}
|
|
|
|
func (s *PowershellProvisionerAccTest) IsCompatible(builder string, vmOS string) bool {
|
|
return vmOS == "windows"
|
|
}
|
|
|
|
func (s *PowershellProvisionerAccTest) RunTest(c *command.BuildCommand, args []string) error {
|
|
UUID := os.Getenv("PACKER_RUN_UUID")
|
|
if UUID == "" {
|
|
UUID, _ = uuid.GenerateUUID()
|
|
os.Setenv("PACKER_RUN_UUID", UUID)
|
|
}
|
|
|
|
if code := c.Run(args); code != 0 {
|
|
ui := c.Meta.Ui.(*packer.BasicUi)
|
|
out := ui.Writer.(*bytes.Buffer)
|
|
err := ui.ErrorWriter.(*bytes.Buffer)
|
|
return fmt.Errorf(
|
|
"Bad exit code.\n\nStdout:\n\n%s\n\nStderr:\n\n%s",
|
|
out.String(),
|
|
err.String())
|
|
}
|
|
|
|
return nil
|
|
}
|