mirror of
https://github.com/hay-kot/homebox.git
synced 2024-12-18 21:16:31 +00:00
114 lines
2.5 KiB
Go
114 lines
2.5 KiB
Go
package repo
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"fmt"
|
|
"io"
|
|
"path/filepath"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/hay-kot/homebox/backend/internal/core/blobstore"
|
|
"github.com/hay-kot/homebox/backend/internal/data/ent"
|
|
"github.com/stretchr/testify/assert"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func useDocs(t *testing.T, num int) []DocumentOut {
|
|
t.Helper()
|
|
|
|
results := make([]DocumentOut, 0, num)
|
|
ids := make([]uuid.UUID, 0, num)
|
|
|
|
for i := 0; i < num; i++ {
|
|
doc, err := tRepos.Docs.Create(context.Background(), tGroup.ID, DocumentCreate{
|
|
Title: fk.Str(10) + ".md",
|
|
Content: bytes.NewReader([]byte(fk.Str(10))),
|
|
})
|
|
|
|
require.NoError(t, err)
|
|
assert.NotNil(t, doc)
|
|
results = append(results, doc)
|
|
ids = append(ids, doc.ID)
|
|
}
|
|
|
|
t.Cleanup(func() {
|
|
for _, id := range ids {
|
|
err := tRepos.Docs.Delete(context.Background(), id)
|
|
if err != nil {
|
|
assert.True(t, ent.IsNotFound(err))
|
|
}
|
|
}
|
|
})
|
|
|
|
return results
|
|
}
|
|
|
|
func TestDocumentRepository_CreateUpdateDelete(t *testing.T) {
|
|
temp := t.TempDir()
|
|
r := DocumentRepository{
|
|
db: tClient,
|
|
bs: blobstore.NewLocalBlobStore(temp),
|
|
}
|
|
|
|
type args struct {
|
|
ctx context.Context
|
|
gid uuid.UUID
|
|
doc DocumentCreate
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
content string
|
|
args args
|
|
title string
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "basic create",
|
|
title: "test.md",
|
|
content: "Lorem ipsum dolor sit amet, consectetur adipiscing elit.",
|
|
args: args{
|
|
ctx: context.Background(),
|
|
gid: tGroup.ID,
|
|
doc: DocumentCreate{
|
|
Title: "test.md",
|
|
Content: bytes.NewReader([]byte("Lorem ipsum dolor sit amet, consectetur adipiscing elit.")),
|
|
},
|
|
},
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
// Create Document
|
|
got, err := r.Create(tt.args.ctx, tt.args.gid, tt.args.doc)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.title, got.Title)
|
|
assert.Equal(t, fmt.Sprintf("%s/documents", tt.args.gid), filepath.Dir(got.Path))
|
|
|
|
ensureRead := func() {
|
|
// Read Document
|
|
bts, err := r.bs.Get(tt.args.ctx, got.Path)
|
|
require.NoError(t, err)
|
|
buf, err := io.ReadAll(bts)
|
|
require.NoError(t, err)
|
|
assert.Equal(t, tt.content, string(buf))
|
|
}
|
|
ensureRead()
|
|
|
|
// Update Document
|
|
got, err = r.Rename(tt.args.ctx, got.ID, "__"+tt.title+"__")
|
|
require.NoError(t, err)
|
|
assert.Equal(t, "__"+tt.title+"__", got.Title)
|
|
|
|
ensureRead()
|
|
|
|
// Delete Document
|
|
err = r.Delete(tt.args.ctx, got.ID)
|
|
require.NoError(t, err)
|
|
|
|
_, err = r.bs.Get(tt.args.ctx, got.Path)
|
|
require.Error(t, err)
|
|
})
|
|
}
|
|
}
|