Setup a test target

This commit is contained in:
sophia 2021-07-19 11:40:47 -05:00 committed by Paul Hinze
parent 5fc4cdcc98
commit 8ca1a29e93
No known key found for this signature in database
GPG Key ID: B69DEDF2D55501C0
2 changed files with 32 additions and 13 deletions

View File

@ -3,20 +3,11 @@ package core
import (
"testing"
"github.com/hashicorp/go-hclog"
"github.com/hashicorp/vagrant-plugin-sdk/core"
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
)
func targetInstance() *Target {
return &Target{
target: &vagrant_server.Target{},
logger: hclog.New(&hclog.LoggerOptions{Name: "test"}),
}
}
func TestTargetSpecializeMachine(t *testing.T) {
tt := targetInstance()
tt, _ := TestTarget(t)
specialized, err := tt.Specialize((*core.Machine)(nil))
if err != nil {
@ -29,7 +20,7 @@ func TestTargetSpecializeMachine(t *testing.T) {
}
func TestTargetSpecializeBad(t *testing.T) {
tt := targetInstance()
tt, _ := TestTarget(t)
specialized, err := tt.Specialize((*core.Project)(nil))
if err != nil {
@ -43,9 +34,8 @@ func TestTargetSpecializeBad(t *testing.T) {
func TestRun(t *testing.T) {
// TODO: needs
// - to be able to create a Target with a project
// - to be able to create a Task
// tt := targetInstance()
// tt, _ := TestTarget(t)
// ctx := context.Background()
// tk := &vagrant_server.Task{}

View File

@ -14,6 +14,7 @@ import (
"github.com/hashicorp/vagrant-plugin-sdk/datadir"
"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
"github.com/hashicorp/vagrant/internal/factory"
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
"github.com/hashicorp/vagrant/internal/server/singleprocess"
)
@ -31,6 +32,34 @@ var TestingTypeMap = map[component.Type]interface{}{
component.SyncedFolderType: (*component.SyncedFolder)(nil),
}
// 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...)
// vagrantServerTarget, err :=
tp.basis.client.UpsertTarget(
context.Background(),
&vagrant_server.UpsertTargetRequest{
Project: tp.Ref().(*vagrant_plugin_sdk.Ref_Project),
Target: &vagrant_server.Target{
Name: "test-target",
Project: tp.Ref().(*vagrant_plugin_sdk.Ref_Project),
},
},
)
target, err = tp.LoadTarget([]TargetOption{
WithTargetRef(&vagrant_plugin_sdk.Ref_Target{Project: tp.Ref().(*vagrant_plugin_sdk.Ref_Project), Name: "test-target"}),
}...)
// return &Target{
// target: &vagrant_server.Target{},
// logger: hclog.New(&hclog.LoggerOptions{Name: "test"}),
// }
return
}
// TestProject returns a fully in-memory and side-effect free Project that
// can be used for testing. Additional options can be given to provide your own
// factories, configuration, etc.