mirror of https://github.com/hashicorp/boundary
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
304 lines
9.3 KiB
304 lines
9.3 KiB
package hostsets_test
|
|
|
|
import (
|
|
"fmt"
|
|
"net/http"
|
|
"testing"
|
|
|
|
"github.com/hashicorp/boundary/api"
|
|
"github.com/hashicorp/boundary/api/hostcatalogs"
|
|
"github.com/hashicorp/boundary/api/hosts"
|
|
"github.com/hashicorp/boundary/api/hostsets"
|
|
"github.com/hashicorp/boundary/internal/host/static"
|
|
"github.com/hashicorp/boundary/internal/iam"
|
|
"github.com/hashicorp/boundary/internal/servers/controller"
|
|
"github.com/kr/pretty"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCustom(t *testing.T) {
|
|
for _, newStyle := range []bool{false, true} {
|
|
t.Run(fmt.Sprintf("custom_%t", newStyle), func(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
tc := controller.NewTestController(t, nil)
|
|
defer tc.Shutdown()
|
|
|
|
token := tc.Token()
|
|
_, proj := iam.TestScopes(t, tc.IamRepo(), iam.WithUserId(token.UserId))
|
|
client := tc.Client().Clone()
|
|
client.SetScopeId(proj.GetPublicId())
|
|
|
|
hc, apiErr, err := hostcatalogs.NewClient(client).Create(tc.Context(), "static")
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
|
|
hClient := hosts.NewClient(client)
|
|
h1, apiErr, err := hClient.Create(tc.Context(), hc.Id, hosts.WithStaticHostAddress("someaddress"))
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
h2, apiErr, err := hClient.Create(tc.Context(), hc.Id, hosts.WithStaticHostAddress("someaddress"))
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
|
|
hSetClient := hostsets.NewClient(client)
|
|
var hSet *hostsets.HostSet
|
|
if newStyle {
|
|
hSet, apiErr, err = hSetClient.Create2(tc.Context(), hc.Id)
|
|
} else {
|
|
hSet, apiErr, err = hSetClient.Create(tc.Context(), hc.Id)
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
|
|
if newStyle {
|
|
hSet, apiErr, err = hSetClient.AddHosts2(tc.Context(), hSet.Id, hSet.Version, []string{h1.Id, h2.Id})
|
|
} else {
|
|
hSet, apiErr, err = hSetClient.AddHosts(tc.Context(), hc.Id, hSet.Id, hSet.Version, []string{h1.Id, h2.Id})
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.Contains(hSet.HostIds, h1.Id, h2.Id)
|
|
|
|
if newStyle {
|
|
hSet, apiErr, err = hSetClient.SetHosts2(tc.Context(), hSet.Id, hSet.Version, []string{h1.Id})
|
|
} else {
|
|
hSet, apiErr, err = hSetClient.SetHosts(tc.Context(), hc.Id, hSet.Id, hSet.Version, []string{h1.Id})
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr, pretty.Sprint(apiErr))
|
|
assert.ElementsMatch([]string{h1.Id}, hSet.HostIds)
|
|
|
|
if newStyle {
|
|
hSet, apiErr, err = hSetClient.RemoveHosts2(tc.Context(), hSet.Id, hSet.Version, []string{h1.Id})
|
|
} else {
|
|
hSet, apiErr, err = hSetClient.RemoveHosts(tc.Context(), hc.Id, hSet.Id, hSet.Version, []string{h1.Id})
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.Empty(hSet.HostIds)
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestSet_List(t *testing.T) {
|
|
for _, newStyle := range []bool{false, true} {
|
|
assert, require := assert.New(t), require.New(t)
|
|
tc := controller.NewTestController(t, nil)
|
|
defer tc.Shutdown()
|
|
|
|
client := tc.Client()
|
|
token := tc.Token()
|
|
_, proj := iam.TestScopes(t, tc.IamRepo(), iam.WithUserId(token.UserId))
|
|
client.SetScopeId(proj.GetPublicId())
|
|
|
|
hc, apiErr, err := hostcatalogs.NewClient(client).Create(tc.Context(), "static")
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
require.NotNil(hc)
|
|
|
|
hClient := hostsets.NewClient(client)
|
|
var ul []*hostsets.HostSet
|
|
|
|
if newStyle {
|
|
ul, apiErr, err = hClient.List2(tc.Context(), hc.Id)
|
|
} else {
|
|
ul, apiErr, err = hClient.List(tc.Context(), hc.Id)
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.Empty(ul)
|
|
|
|
var expected []*hostsets.HostSet
|
|
for i := 0; i < 10; i++ {
|
|
expected = append(expected, &hostsets.HostSet{Name: fmt.Sprint(i)})
|
|
}
|
|
|
|
expected[0], apiErr, err = hClient.Create(tc.Context(), hc.Id, hostsets.WithName(expected[0].Name))
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
|
|
if newStyle {
|
|
ul, apiErr, err = hClient.List2(tc.Context(), hc.Id)
|
|
} else {
|
|
ul, apiErr, err = hClient.List(tc.Context(), hc.Id)
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.ElementsMatch(comparableSetSlice(expected[:1]), comparableSetSlice(ul))
|
|
|
|
for i := 1; i < 10; i++ {
|
|
expected[i], apiErr, err = hClient.Create(tc.Context(), hc.Id, hostsets.WithName(expected[i].Name))
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
}
|
|
if newStyle {
|
|
ul, apiErr, err = hClient.List2(tc.Context(), hc.Id)
|
|
} else {
|
|
ul, apiErr, err = hClient.List(tc.Context(), hc.Id)
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.ElementsMatch(comparableSetSlice(expected), comparableSetSlice(ul))
|
|
}
|
|
}
|
|
|
|
func comparableSetSlice(in []*hostsets.HostSet) []hostsets.HostSet {
|
|
var filtered []hostsets.HostSet
|
|
for _, i := range in {
|
|
p := hostsets.HostSet{
|
|
Id: i.Id,
|
|
Name: i.Name,
|
|
Description: i.Description,
|
|
CreatedTime: i.CreatedTime,
|
|
UpdatedTime: i.UpdatedTime,
|
|
}
|
|
filtered = append(filtered, p)
|
|
}
|
|
return filtered
|
|
}
|
|
|
|
func TestSet_Crud(t *testing.T) {
|
|
for _, newStyle := range []bool{false, true} {
|
|
t.Run(fmt.Sprintf("crud_%t", newStyle), func(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
tc := controller.NewTestController(t, nil)
|
|
defer tc.Shutdown()
|
|
|
|
client := tc.Client()
|
|
token := tc.Token()
|
|
org, proj := iam.TestScopes(t, tc.IamRepo(), iam.WithUserId(token.UserId))
|
|
client.SetScopeId(org.GetPublicId())
|
|
projClient := client.Clone()
|
|
projClient.SetScopeId(proj.GetPublicId())
|
|
|
|
hc, apiErr, err := hostcatalogs.NewClient(projClient).Create(tc.Context(), "static")
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
require.NotNil(hc)
|
|
|
|
checkHost := func(t *testing.T, step string, h *hostsets.HostSet, apiErr *api.Error, err error, wantedName string, wantVersion uint32) {
|
|
t.Helper()
|
|
require.NoError(err, step)
|
|
if !assert.Nil(apiErr, step) && apiErr.Message != "" {
|
|
t.Errorf("ApiError message: %q", apiErr.Message)
|
|
}
|
|
assert.NotNil(h, "returned no resource", step)
|
|
gotName := ""
|
|
if h.Name != "" {
|
|
gotName = h.Name
|
|
}
|
|
assert.Equal(wantedName, gotName, step)
|
|
assert.Equal(wantVersion, h.Version)
|
|
}
|
|
|
|
hClient := hostsets.NewClient(projClient)
|
|
|
|
var h *hostsets.HostSet
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Create2(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
} else {
|
|
h, apiErr, err = hClient.Create(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
}
|
|
checkHost(t, "create", h, apiErr, err, "foo", 1)
|
|
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Read2(tc.Context(), h.Id)
|
|
} else {
|
|
h, apiErr, err = hClient.Read(tc.Context(), hc.Id, h.Id)
|
|
}
|
|
checkHost(t, "read", h, apiErr, err, "foo", 1)
|
|
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Update2(tc.Context(), h.Id, h.Version, hostsets.WithName("bar"))
|
|
} else {
|
|
h, apiErr, err = hClient.Update(tc.Context(), hc.Id, h.Id, h.Version, hostsets.WithName("bar"))
|
|
}
|
|
checkHost(t, "update", h, apiErr, err, "bar", 2)
|
|
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Update2(tc.Context(), h.Id, h.Version, hostsets.DefaultName())
|
|
} else {
|
|
h, apiErr, err = hClient.Update(tc.Context(), hc.Id, h.Id, h.Version, hostsets.DefaultName())
|
|
}
|
|
checkHost(t, "update", h, apiErr, err, "", 3)
|
|
|
|
var existed bool
|
|
if newStyle {
|
|
existed, apiErr, err = hClient.Delete2(tc.Context(), h.Id)
|
|
} else {
|
|
existed, apiErr, err = hClient.Delete(tc.Context(), hc.Id, h.Id)
|
|
}
|
|
assert.NoError(err)
|
|
assert.True(existed, "Expected existing catalog when deleted, but it wasn't.")
|
|
|
|
if newStyle {
|
|
existed, apiErr, err = hClient.Delete2(tc.Context(), h.Id)
|
|
} else {
|
|
existed, apiErr, err = hClient.Delete(tc.Context(), hc.Id, h.Id)
|
|
}
|
|
require.NoError(err)
|
|
assert.NotNil(apiErr)
|
|
assert.EqualValues(http.StatusForbidden, apiErr.Status)
|
|
})
|
|
}
|
|
}
|
|
|
|
// TODO: Get better coverage for expected errors and error formats.
|
|
func TestSet_Errors(t *testing.T) {
|
|
for _, newStyle := range []bool{false, true} {
|
|
assert, require := assert.New(t), require.New(t)
|
|
tc := controller.NewTestController(t, nil)
|
|
defer tc.Shutdown()
|
|
|
|
client := tc.Client()
|
|
token := tc.Token()
|
|
_, proj := iam.TestScopes(t, tc.IamRepo(), iam.WithUserId(token.UserId))
|
|
client.SetScopeId(proj.GetPublicId())
|
|
|
|
hc, apiErr, err := hostcatalogs.NewClient(client).Create(tc.Context(), "static")
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
require.NotNil(hc)
|
|
|
|
hClient := hostsets.NewClient(client)
|
|
|
|
var h *hostsets.HostSet
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Create2(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
} else {
|
|
h, apiErr, err = hClient.Create(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
}
|
|
require.NoError(err)
|
|
require.Nil(apiErr)
|
|
assert.NotNil(h)
|
|
|
|
if newStyle {
|
|
h, apiErr, err = hClient.Create2(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
} else {
|
|
h, apiErr, err = hClient.Create(tc.Context(), hc.Id, hostsets.WithName("foo"))
|
|
}
|
|
require.NoError(err)
|
|
assert.NotNil(apiErr)
|
|
assert.Nil(h)
|
|
|
|
if newStyle {
|
|
_, apiErr, err = hClient.Read2(tc.Context(), static.HostSetPrefix+"_doesntexis")
|
|
} else {
|
|
_, apiErr, err = hClient.Read(tc.Context(), hc.Id, static.HostSetPrefix+"_doesntexis")
|
|
}
|
|
require.NoError(err)
|
|
assert.NotNil(apiErr)
|
|
assert.EqualValues(http.StatusForbidden, apiErr.Status)
|
|
|
|
if newStyle {
|
|
_, apiErr, err = hClient.Read2(tc.Context(), "invalid id")
|
|
} else {
|
|
_, apiErr, err = hClient.Read(tc.Context(), hc.Id, "invalid id")
|
|
}
|
|
require.NoError(err)
|
|
assert.NotNil(apiErr)
|
|
assert.EqualValues(http.StatusBadRequest, apiErr.Status)
|
|
}
|
|
}
|