diff --git a/invite/invite_test.go b/invite/invite_test.go index c414005..d217874 100644 --- a/invite/invite_test.go +++ b/invite/invite_test.go @@ -44,6 +44,20 @@ func TestInvite(t *testing.T) { } }) + t.Run("generate invite successfully when one already exists", func(t *testing.T) { + st.EXPECT().GetServer(server.Id).Return(server, nil) + st.EXPECT().GetInvite(gomock.Any()).Return(store.Invite{}, nil) + st.EXPECT().SaveInvite(gomock.Any()).Return(nil) + + token, err := inm.Create(inv, user) + if err != nil { + t.Fatal(err) + } + if token == "" { + t.Errorf("expected non empty string, but got empty string") + } + }) + t.Run("no logged uses of invite", func(t *testing.T) { st.EXPECT().InviteLog(inv).Return([]store.InviteLog{}, nil) uses, err := inm.RemainingUses(inv)