75 lines
1.6 KiB
Go
75 lines
1.6 KiB
Go
package packer
|
|
|
|
import (
|
|
"bytes"
|
|
"io/ioutil"
|
|
"testing"
|
|
)
|
|
|
|
func TestCoreConfig(t *testing.T) *CoreConfig {
|
|
// Create some test components
|
|
components := ComponentFinder{
|
|
BuilderStore: MapOfBuilder{
|
|
"test": func() (Builder, error) { return &MockBuilder{}, nil },
|
|
},
|
|
}
|
|
|
|
return &CoreConfig{
|
|
Components: components,
|
|
}
|
|
}
|
|
|
|
func TestCore(t *testing.T, c *CoreConfig) *Core {
|
|
core, err := NewCore(c)
|
|
if err != nil {
|
|
t.Fatalf("err: %s", err)
|
|
}
|
|
|
|
return core
|
|
}
|
|
|
|
func TestUi(t *testing.T) Ui {
|
|
var buf bytes.Buffer
|
|
return &BasicUi{
|
|
Reader: &buf,
|
|
Writer: ioutil.Discard,
|
|
ErrorWriter: ioutil.Discard,
|
|
}
|
|
}
|
|
|
|
// TestBuilder sets the builder with the name n to the component finder
|
|
// and returns the mock.
|
|
func TestBuilder(t *testing.T, c *CoreConfig, n string) *MockBuilder {
|
|
var b MockBuilder
|
|
|
|
c.Components.BuilderStore = MapOfBuilder{
|
|
n: func() (Builder, error) { return &b, nil },
|
|
}
|
|
|
|
return &b
|
|
}
|
|
|
|
// TestProvisioner sets the prov. with the name n to the component finder
|
|
// and returns the mock.
|
|
func TestProvisioner(t *testing.T, c *CoreConfig, n string) *MockProvisioner {
|
|
var b MockProvisioner
|
|
|
|
c.Components.ProvisionerStore = MapOfProvisioner{
|
|
n: func() (Provisioner, error) { return &b, nil },
|
|
}
|
|
|
|
return &b
|
|
}
|
|
|
|
// TestPostProcessor sets the prov. with the name n to the component finder
|
|
// and returns the mock.
|
|
func TestPostProcessor(t *testing.T, c *CoreConfig, n string) *MockPostProcessor {
|
|
var b MockPostProcessor
|
|
|
|
c.Components.PostProcessorStore = MapOfPostProcessor{
|
|
n: func() (PostProcessor, error) { return &b, nil },
|
|
}
|
|
|
|
return &b
|
|
}
|