beckermj/vaguerent
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Allow tests to set target config for test target

Browse Source
This commit is contained in:
sophia 2022-02-22 16:05:10 -06:00 committed by Paul Hinze
parent 06ad1b4565
commit 70d0d4dfb8
No known key found for this signature in database
GPG Key ID: B69DEDF2D55501C0
1 changed files with 20 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
internal/core/testing_project.go
View File

@ -6,6 +6,7 @@ import (
"os"
"github.com/hashicorp/go-hclog"
"github.com/hashicorp/go-multierror"
"github.com/mitchellh/go-testing-interface"
"github.com/stretchr/testify/mock"
"github.com/stretchr/testify/require"
@ -37,8 +38,8 @@ var TestingTypeMap = map[component.Type]interface{}{
// TestTarget returns a fully in-memory and side-effect free Target that
// can be used for testing. Additional options can be given to provide your own
// factories, configuration, etc.
func TestTarget(t testing.T, opts ...BasisOption) (target *Target, err error) {
tp := TestProject(t, opts...)
func TestTarget(t testing.T, opts ...TestTargetOption) (target *Target, err error) {
tp := TestProject(t)
tp.basis.client.UpsertTarget(
context.Background(),
&vagrant_server.UpsertTargetRequest{
@ -53,14 +54,20 @@ func TestTarget(t testing.T, opts ...BasisOption) (target *Target, err error) {
WithTargetRef(&vagrant_plugin_sdk.Ref_Target{Project: tp.Ref().(*vagrant_plugin_sdk.Ref_Project), Name: "test-target"}),
}...)
for _, opt := range opts {
if oerr := opt(target); oerr != nil {
err = multierror.Append(err, oerr)
}
}
return
}
// TestMachine returns a fully in-memory and side-effect free Machine that
// can be used for testing. Additional options can be given to provide your own
// factories, configuration, etc.
func TestMachine(t testing.T, opts ...BasisOption) (machine *Machine, err error) {
tt, _ := TestTarget(t)
func TestMachine(t testing.T, topts ...TestTargetOption) (machine *Machine, err error) {
tt, _ := TestTarget(t, topts...)
specialized, err := tt.Specialize((*core.Machine)(nil))
if err != nil {
return nil, err
@ -133,3 +140,12 @@ func TestFactory(t testing.T, typ component.Type) *factory.Factory {
func TestFactoryRegister(t testing.T, f *factory.Factory, n string, v interface{}) {
require.NoError(t, f.Register(n, func() interface{} { return v }))
}
type TestTargetOption func(*Target) error
func WithTestTargetConfig(config *vagrant_plugin_sdk.Vagrantfile_MachineConfig) TestTargetOption {
return func(t *Target) (err error) {
t.target.Configuration = config
return
}
}