diff --git a/server/datastore/mysql/invites_test.go b/server/datastore/mysql/invites_test.go index b1f6a0e587..384131977b 100644 --- a/server/datastore/mysql/invites_test.go +++ b/server/datastore/mysql/invites_test.go @@ -45,19 +45,29 @@ func testInvitesCreate(t *testing.T, ds *Datastore) { Name: "user", Token: "some_user", Teams: []fleet.UserTeam{ - {Role: "observer", Team: fleet.Team{ID: 1}}, - {Role: "maintainer", Team: fleet.Team{ID: 3}}, + {Role: fleet.RoleObserver, Team: fleet.Team{ID: 1}}, + {Role: fleet.RoleMaintainer, Team: fleet.Team{ID: 3}}, }, } invite, err := ds.NewInvite(context.Background(), invite) - require.Nil(t, err) + require.NoError(t, err) verify, err := ds.InviteByEmail(context.Background(), invite.Email) - require.Nil(t, err) + require.NoError(t, err) assert.Equal(t, invite.ID, verify.ID) assert.Equal(t, invite.Email, verify.Email) assert.Len(t, invite.Teams, 2) + + _, err = ds.NewInvite(context.Background(), &fleet.Invite{ + Email: "anotheruser@foo.com", + Name: "anotheruser", + Token: "anothersome_user", + Teams: []fleet.UserTeam{ + {Role: fleet.RoleAdmin, Team: fleet.Team{ID: 3}}, + }, + }) + require.NoError(t, err) } func setupTestInvites(t *testing.T, ds fleet.Datastore) { diff --git a/server/fleet/teams.go b/server/fleet/teams.go index 5ee3cd376f..f7c62ef209 100644 --- a/server/fleet/teams.go +++ b/server/fleet/teams.go @@ -61,6 +61,7 @@ type TeamUser struct { } var teamRoles = map[string]bool{ + RoleAdmin: true, RoleObserver: true, RoleMaintainer: true, }