diff options
Diffstat (limited to 'models/org_team_test.go')
| -rw-r--r-- | models/org_team_test.go | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/models/org_team_test.go b/models/org_team_test.go index 87bfbb4841..a81f9c0749 100644 --- a/models/org_team_test.go +++ b/models/org_team_test.go @@ -285,8 +285,22 @@ func TestGetTeamMembers(t *testing.T) { func TestGetUserTeams(t *testing.T) { assert.NoError(t, PrepareTestDatabase()) + test := func(userID int64) { + teams, err := GetUserTeams(userID) + assert.NoError(t, err) + for _, team := range teams { + AssertExistsAndLoadBean(t, &TeamUser{TeamID: team.ID, UID: userID}) + } + } + test(2) + test(5) + test(NonexistentID) +} + +func TestGetUserOrgTeams(t *testing.T) { + assert.NoError(t, PrepareTestDatabase()) test := func(orgID, userID int64) { - teams, err := GetUserTeams(orgID, userID) + teams, err := GetUserOrgTeams(orgID, userID) assert.NoError(t, err) for _, team := range teams { assert.EqualValues(t, orgID, team.OrgID) |