123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172 |
- package teststore
- import (
- "context"
- "testing"
- "github.com/stretchr/testify/require"
- "github.com/usememos/memos/store"
- )
- func TestResourceStore(t *testing.T) {
- ctx := context.Background()
- ts := NewTestingStore(ctx, t)
- _, err := ts.CreateResource(ctx, &store.Resource{
- CreatorID: 101,
- Filename: "test.epub",
- Blob: []byte("test"),
- InternalPath: "",
- ExternalLink: "",
- Type: "application/epub+zip",
- Size: 637607,
- })
- require.NoError(t, err)
- correctFilename := "test.epub"
- incorrectFilename := "test.png"
- resource, err := ts.GetResource(ctx, &store.FindResource{
- Filename: &correctFilename,
- })
- require.NoError(t, err)
- require.Equal(t, correctFilename, resource.Filename)
- require.Equal(t, int32(1), resource.ID)
- _, err = ts.UpsertMemoResource(ctx, &store.UpsertMemoResource{
- MemoID: 1,
- ResourceID: resource.ID,
- })
- require.NoError(t, err)
- resource, err = ts.GetResource(ctx, &store.FindResource{
- ID: &resource.ID,
- })
- require.NoError(t, err)
- require.Equal(t, *resource.RelatedMemoID, int32(1))
- notFoundResource, err := ts.GetResource(ctx, &store.FindResource{
- Filename: &incorrectFilename,
- })
- require.NoError(t, err)
- require.Nil(t, notFoundResource)
- var correctCreatorID int32 = 101
- var incorrectCreatorID int32 = 102
- _, err = ts.GetResource(ctx, &store.FindResource{
- CreatorID: &correctCreatorID,
- })
- require.NoError(t, err)
- notFoundResource, err = ts.GetResource(ctx, &store.FindResource{
- CreatorID: &incorrectCreatorID,
- })
- require.NoError(t, err)
- require.Nil(t, notFoundResource)
- err = ts.DeleteResource(ctx, &store.DeleteResource{
- ID: 1,
- })
- require.NoError(t, err)
- err = ts.DeleteResource(ctx, &store.DeleteResource{
- ID: 2,
- })
- require.NoError(t, err)
- }
|