Add Enterprise tests

This commit is contained in:
Gabriel Adrian Samfira 2022-12-04 17:30:27 +00:00
parent 0869073906
commit cb5baeb547
9 changed files with 640 additions and 141 deletions

View file

@ -51,28 +51,6 @@ type OrgTestSuite struct {
Fixtures *OrgTestFixtures
}
func (s *OrgTestSuite) equalOrgsByName(expected, actual []params.Organization) {
s.Require().Equal(len(expected), len(actual))
sort.Slice(expected, func(i, j int) bool { return expected[i].Name > expected[j].Name })
sort.Slice(actual, func(i, j int) bool { return actual[i].Name > actual[j].Name })
for i := 0; i < len(expected); i++ {
s.Require().Equal(expected[i].Name, actual[i].Name)
}
}
func (s *OrgTestSuite) equalPoolsByID(expected, actual []params.Pool) {
s.Require().Equal(len(expected), len(actual))
sort.Slice(expected, func(i, j int) bool { return expected[i].ID > expected[j].ID })
sort.Slice(actual, func(i, j int) bool { return actual[i].ID > actual[j].ID })
for i := 0; i < len(expected); i++ {
s.Require().Equal(expected[i].ID, actual[i].ID)
}
}
func (s *OrgTestSuite) equalInstancesByName(expected, actual []params.Instance) {
s.Require().Equal(len(expected), len(actual))
@ -276,7 +254,7 @@ func (s *OrgTestSuite) TestListOrganizations() {
orgs, err := s.Store.ListOrganizations(context.Background())
s.Require().Nil(err)
s.equalOrgsByName(s.Fixtures.Orgs, orgs)
garmTesting.EqualDBEntityByName(s.T(), s.Fixtures.Orgs, orgs)
}
func (s *OrgTestSuite) TestListOrganizationsDBFetchErr() {
@ -689,7 +667,7 @@ func (s *OrgTestSuite) TestListOrgPools() {
pools, err := s.Store.ListOrgPools(context.Background(), s.Fixtures.Orgs[0].ID)
s.Require().Nil(err)
s.equalPoolsByID(orgPools, pools)
garmTesting.EqualDBEntityID(s.T(), orgPools, pools)
}
func (s *OrgTestSuite) TestListOrgPoolsInvalidOrgID() {

View file

@ -22,7 +22,6 @@ import (
garmTesting "garm/internal/testing"
"garm/params"
"regexp"
"sort"
"testing"
"github.com/stretchr/testify/suite"
@ -45,17 +44,6 @@ type PoolsTestSuite struct {
Fixtures *PoolsTestFixtures
}
func (s *PoolsTestSuite) equalPoolsByID(expected, actual []params.Pool) {
s.Require().Equal(len(expected), len(actual))
sort.Slice(expected, func(i, j int) bool { return expected[i].ID > expected[j].ID })
sort.Slice(actual, func(i, j int) bool { return actual[i].ID > actual[j].ID })
for i := 0; i < len(expected); i++ {
s.Require().Equal(expected[i].ID, actual[i].ID)
}
}
func (s *PoolsTestSuite) assertSQLMockExpectations() {
err := s.Fixtures.SQLMock.ExpectationsWereMet()
if err != nil {
@ -134,7 +122,7 @@ func (s *PoolsTestSuite) TestListAllPools() {
pools, err := s.Store.ListAllPools(context.Background())
s.Require().Nil(err)
s.equalPoolsByID(s.Fixtures.Pools, pools)
garmTesting.EqualDBEntityID(s.T(), s.Fixtures.Pools, pools)
}
func (s *PoolsTestSuite) TestListAllPoolsDBFetchErr() {

View file

@ -60,17 +60,6 @@ func (s *RepoTestSuite) equalReposByName(expected, actual []params.Repository) {
}
}
func (s *RepoTestSuite) equalPoolsByID(expected, actual []params.Pool) {
s.Require().Equal(len(expected), len(actual))
sort.Slice(expected, func(i, j int) bool { return expected[i].ID > expected[j].ID })
sort.Slice(actual, func(i, j int) bool { return actual[i].ID > actual[j].ID })
for i := 0; i < len(expected); i++ {
s.Require().Equal(expected[i].ID, actual[i].ID)
}
}
func (s *RepoTestSuite) equalInstancesByID(expected, actual []params.Instance) {
s.Require().Equal(len(expected), len(actual))
@ -714,7 +703,7 @@ func (s *RepoTestSuite) TestListRepoPools() {
pools, err := s.Store.ListRepoPools(context.Background(), s.Fixtures.Repos[0].ID)
s.Require().Nil(err)
s.equalPoolsByID(repoPools, pools)
garmTesting.EqualDBEntityID(s.T(), repoPools, pools)
}
func (s *RepoTestSuite) TestListRepoPoolsInvalidRepoID() {