mirror of
https://github.com/usememos/memos.git
synced 2024-11-15 11:17:58 +08:00
5bcbbd4c52
It's just a matter of explicitly closing the database, so that TempDir.removeAll doesn't fail.
54 lines
1.3 KiB
Go
54 lines
1.3 KiB
Go
package teststore
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
|
|
storepb "github.com/usememos/memos/proto/gen/store"
|
|
"github.com/usememos/memos/store"
|
|
)
|
|
|
|
func TestInboxStore(t *testing.T) {
|
|
ctx := context.Background()
|
|
ts := NewTestingStore(ctx, t)
|
|
user, err := createTestingHostUser(ctx, ts)
|
|
require.NoError(t, err)
|
|
const systemBotID int32 = 0
|
|
create := &store.Inbox{
|
|
SenderID: systemBotID,
|
|
ReceiverID: user.ID,
|
|
Status: store.UNREAD,
|
|
Message: &storepb.InboxMessage{
|
|
Type: storepb.InboxMessage_TYPE_MEMO_COMMENT,
|
|
},
|
|
}
|
|
inbox, err := ts.CreateInbox(ctx, create)
|
|
require.NoError(t, err)
|
|
require.NotNil(t, inbox)
|
|
require.Equal(t, create.Message, inbox.Message)
|
|
inboxes, err := ts.ListInboxes(ctx, &store.FindInbox{
|
|
ReceiverID: &user.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, len(inboxes))
|
|
require.Equal(t, inbox, inboxes[0])
|
|
updatedInbox, err := ts.UpdateInbox(ctx, &store.UpdateInbox{
|
|
ID: inbox.ID,
|
|
Status: store.ARCHIVED,
|
|
})
|
|
require.NoError(t, err)
|
|
require.NotNil(t, updatedInbox)
|
|
require.Equal(t, store.ARCHIVED, updatedInbox.Status)
|
|
err = ts.DeleteInbox(ctx, &store.DeleteInbox{
|
|
ID: inbox.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
inboxes, err = ts.ListInboxes(ctx, &store.FindInbox{
|
|
ReceiverID: &user.ID,
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 0, len(inboxes))
|
|
ts.Close()
|
|
}
|