1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
|
package singleprocess
import (
"context"
"testing"
"github.com/hashicorp/vagrant-plugin-sdk/proto/vagrant_plugin_sdk"
"github.com/hashicorp/vagrant/internal/server/proto/vagrant_server"
"github.com/stretchr/testify/require"
"google.golang.org/protobuf/types/known/emptypb"
)
func TestServiceTarget(t *testing.T) {
ctx := context.Background()
t.Run("set and get", func(t *testing.T) {
require := require.New(t)
client := TestServer(t)
// need a basis and a project to have a target
basisResp, err := client.UpsertBasis(ctx, &vagrant_server.UpsertBasisRequest{
Basis: &vagrant_server.Basis{
Name: "mybasis",
Path: "/dev/null",
},
})
require.NoError(err)
projectResp, err := client.UpsertProject(ctx, &vagrant_server.UpsertProjectRequest{
Project: &vagrant_server.Project{
Name: "myproject",
Path: "/dev/null/project",
Basis: &vagrant_plugin_sdk.Ref_Basis{ResourceId: basisResp.Basis.ResourceId},
},
})
require.NoError(err)
projectRef := &vagrant_plugin_sdk.Ref_Project{ResourceId: projectResp.Project.ResourceId}
resp, err := client.UpsertTarget(ctx, &vagrant_server.UpsertTargetRequest{
Target: &vagrant_server.Target{
Name: "mytarget",
Project: projectRef,
},
})
require.NoError(err)
require.NotNil(resp)
require.Equal("mytarget", resp.Target.Name)
})
t.Run("find and list and delete", func(t *testing.T) {
require := require.New(t)
client := TestServer(t)
// first insert
basisResp, err := client.UpsertBasis(ctx, &vagrant_server.UpsertBasisRequest{
Basis: &vagrant_server.Basis{
Name: "mybasis",
Path: "/dev/null",
},
})
require.NoError(err)
projectResp, err := client.UpsertProject(ctx, &vagrant_server.UpsertProjectRequest{
Project: &vagrant_server.Project{
Name: "myproject",
Path: "/dev/null/project",
Basis: &vagrant_plugin_sdk.Ref_Basis{ResourceId: basisResp.Basis.ResourceId},
},
})
projectRef := &vagrant_plugin_sdk.Ref_Project{ResourceId: projectResp.Project.ResourceId}
_, err = client.UpsertTarget(ctx, &vagrant_server.UpsertTargetRequest{
Target: &vagrant_server.Target{
Name: "mytarget",
Project: projectRef,
},
})
require.NoError(err)
// see if we can find it by name & project
findResp, err := client.FindTarget(ctx, &vagrant_server.FindTargetRequest{
Target: &vagrant_server.Target{Project: projectRef, Name: "mytarget"},
})
require.NoError(err)
require.NotNil(findResp)
require.Equal("mytarget", findResp.Target.Name)
// then ensure it shows up in a list
listResp, err := client.ListTargets(ctx, &emptypb.Empty{})
require.NoError(err)
require.NotNil(listResp)
require.Len(listResp.Targets, 1)
// then delete it and the list should be empty
_, err = client.DeleteTarget(ctx, &vagrant_server.DeleteTargetRequest{
Target: &vagrant_plugin_sdk.Ref_Target{
ResourceId: findResp.Target.ResourceId,
Project: projectRef,
},
})
require.NoError(err)
listResp, err = client.ListTargets(ctx, &emptypb.Empty{})
require.NoError(err)
require.NotNil(listResp)
require.Len(listResp.Targets, 0)
})
t.Run("reasonable errors: set without project", func(t *testing.T) {
require := require.New(t)
client := TestServer(t)
_, err := client.UpsertTarget(ctx, &vagrant_server.UpsertTargetRequest{
Target: &vagrant_server.Target{
Name: "ihavenoproject",
},
})
require.Error(err)
require.Contains(err.Error(), "not include parent")
})
}
|
