Michael Gaffney
9ca8a4ec20
* Update host set member test * Add ListSetMembers method signature and doc * Move repository methods for host set members to new file * Add TestSetMembers * Implement ListSetMembers * Return hosts from ListSetMembers * Implement AddSetMembers * Implement DeleteSetMembers * Use sql.QueryContext not sql.Query * Implement SetSetMembers * Refactor: extract getHosts method * Refactor: extract createMembers method * Refactor: extract common functions * Return correct count of changed In SetSetMembers, retrieving the current list of hosts in the set happens after the transaction to modify the set has completed. This retrieval can result in an error but the number of modified rows is not effected. * Make list set members an unexported method * Get hosts within the same transaction as mutations of the set * Return hosts of set from LookupSet method * Return hosts of set from UpdateSet method * Remove listSetMembers * Fix merge conflict * Add NOTE and TODOs about using new pattern in "SetMembers" methods
90 lines
2.2 KiB
Go
90 lines
2.2 KiB
Go
package static
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/hashicorp/boundary/internal/db"
|
|
"github.com/hashicorp/boundary/internal/iam"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func Test_TestCatalogs(t *testing.T) {
|
|
assert, require := assert.New(t), require.New(t)
|
|
conn, _ := db.TestSetup(t, "postgres")
|
|
wrapper := db.TestWrapper(t)
|
|
_, proj := iam.TestScopes(t, iam.TestRepo(t, conn, wrapper))
|
|
require.NotNil(proj)
|
|
assert.NotEmpty(proj.GetPublicId())
|
|
|
|
count := 4
|
|
cs := TestCatalogs(t, conn, proj.GetPublicId(), count)
|
|
assert.Len(cs, count)
|
|
for _, c := range cs {
|
|
assert.NotEmpty(c.GetPublicId())
|
|
}
|
|
}
|
|
|
|
func Test_TestHosts(t *testing.T) {
|
|
t.Helper()
|
|
assert, require := assert.New(t), require.New(t)
|
|
conn, _ := db.TestSetup(t, "postgres")
|
|
wrapper := db.TestWrapper(t)
|
|
org, _ := iam.TestScopes(t, iam.TestRepo(t, conn, wrapper))
|
|
|
|
require.NotNil(org)
|
|
assert.NotEmpty(org.GetPublicId())
|
|
|
|
c := TestCatalogs(t, conn, org.GetPublicId(), 1)[0]
|
|
|
|
count := 4
|
|
hosts := TestHosts(t, conn, c.GetPublicId(), count)
|
|
assert.Len(hosts, count)
|
|
for _, h := range hosts {
|
|
assert.NotEmpty(h.GetPublicId())
|
|
}
|
|
}
|
|
|
|
func Test_TestSets(t *testing.T) {
|
|
t.Helper()
|
|
assert, require := assert.New(t), require.New(t)
|
|
conn, _ := db.TestSetup(t, "postgres")
|
|
wrapper := db.TestWrapper(t)
|
|
_, prj := iam.TestScopes(t, iam.TestRepo(t, conn, wrapper))
|
|
|
|
require.NotNil(prj)
|
|
assert.NotEmpty(prj.GetPublicId())
|
|
|
|
c := TestCatalogs(t, conn, prj.GetPublicId(), 1)[0]
|
|
|
|
count := 4
|
|
sets := TestSets(t, conn, c.GetPublicId(), count)
|
|
assert.Len(sets, count)
|
|
for _, s := range sets {
|
|
assert.NotEmpty(s.GetPublicId())
|
|
}
|
|
}
|
|
|
|
func Test_TestSetMembers(t *testing.T) {
|
|
t.Helper()
|
|
assert, require := assert.New(t), require.New(t)
|
|
conn, _ := db.TestSetup(t, "postgres")
|
|
wrapper := db.TestWrapper(t)
|
|
_, prj := iam.TestScopes(t, iam.TestRepo(t, conn, wrapper))
|
|
|
|
require.NotNil(prj)
|
|
assert.NotEmpty(prj.GetPublicId())
|
|
|
|
c := TestCatalogs(t, conn, prj.GetPublicId(), 1)[0]
|
|
s := TestSets(t, conn, c.GetPublicId(), 1)[0]
|
|
|
|
count := 4
|
|
hosts := TestHosts(t, conn, c.GetPublicId(), count)
|
|
members := TestSetMembers(t, conn, s.PublicId, hosts)
|
|
assert.Len(members, count)
|
|
for _, m := range members {
|
|
assert.NotEmpty(m.GetSetId())
|
|
assert.NotEmpty(m.GetHostId())
|
|
}
|
|
}
|