mirror of https://github.com/hay-kot/homebox.git
134 lines
3.0 KiB
Go
134 lines
3.0 KiB
Go
package repo
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/google/uuid"
|
|
"github.com/hay-kot/content/backend/ent"
|
|
"github.com/hay-kot/content/backend/ent/attachment"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestAttachmentRepo_Create(t *testing.T) {
|
|
doc := useDocs(t, 1)[0]
|
|
item := useItems(t, 1)[0]
|
|
|
|
ids := []uuid.UUID{doc.ID, item.ID}
|
|
t.Cleanup(func() {
|
|
for _, id := range ids {
|
|
_ = tRepos.Attachments.Delete(context.Background(), id)
|
|
}
|
|
})
|
|
|
|
type args struct {
|
|
ctx context.Context
|
|
itemId uuid.UUID
|
|
docId uuid.UUID
|
|
typ attachment.Type
|
|
}
|
|
tests := []struct {
|
|
name string
|
|
args args
|
|
want *ent.Attachment
|
|
wantErr bool
|
|
}{
|
|
{
|
|
name: "create attachment",
|
|
args: args{
|
|
ctx: context.Background(),
|
|
itemId: item.ID,
|
|
docId: doc.ID,
|
|
typ: attachment.TypePhoto,
|
|
},
|
|
want: &ent.Attachment{
|
|
Type: attachment.TypePhoto,
|
|
},
|
|
},
|
|
{
|
|
name: "create attachment with invalid item id",
|
|
args: args{
|
|
ctx: context.Background(),
|
|
itemId: uuid.New(),
|
|
docId: doc.ID,
|
|
typ: "blarg",
|
|
},
|
|
wantErr: true,
|
|
},
|
|
}
|
|
for _, tt := range tests {
|
|
t.Run(tt.name, func(t *testing.T) {
|
|
|
|
got, err := tRepos.Attachments.Create(tt.args.ctx, tt.args.itemId, tt.args.docId, tt.args.typ)
|
|
if (err != nil) != tt.wantErr {
|
|
t.Errorf("AttachmentRepo.Create() error = %v, wantErr %v", err, tt.wantErr)
|
|
return
|
|
}
|
|
|
|
if tt.wantErr {
|
|
return
|
|
}
|
|
|
|
assert.Equal(t, tt.want.Type, got.Type)
|
|
|
|
withItems, err := tRepos.Attachments.Get(tt.args.ctx, got.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, tt.args.itemId, withItems.Edges.Item.ID)
|
|
assert.Equal(t, tt.args.docId, withItems.Edges.Document.ID)
|
|
|
|
ids = append(ids, got.ID)
|
|
})
|
|
}
|
|
}
|
|
|
|
func useAttachments(t *testing.T, n int) []*ent.Attachment {
|
|
t.Helper()
|
|
|
|
doc := useDocs(t, 1)[0]
|
|
item := useItems(t, 1)[0]
|
|
|
|
ids := make([]uuid.UUID, 0, n)
|
|
t.Cleanup(func() {
|
|
for _, id := range ids {
|
|
_ = tRepos.Attachments.Delete(context.Background(), id)
|
|
}
|
|
})
|
|
|
|
attachments := make([]*ent.Attachment, n)
|
|
for i := 0; i < n; i++ {
|
|
attachment, err := tRepos.Attachments.Create(context.Background(), item.ID, doc.ID, attachment.TypePhoto)
|
|
assert.NoError(t, err)
|
|
attachments[i] = attachment
|
|
|
|
ids = append(ids, attachment.ID)
|
|
}
|
|
|
|
return attachments
|
|
}
|
|
|
|
func TestAttachmentRepo_Update(t *testing.T) {
|
|
entity := useAttachments(t, 1)[0]
|
|
|
|
for _, typ := range []attachment.Type{"photo", "manual", "warranty", "attachment"} {
|
|
t.Run(string(typ), func(t *testing.T) {
|
|
_, err := tRepos.Attachments.Update(context.Background(), entity.ID, typ)
|
|
assert.NoError(t, err)
|
|
|
|
updated, err := tRepos.Attachments.Get(context.Background(), entity.ID)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, typ, updated.Type)
|
|
})
|
|
}
|
|
|
|
}
|
|
|
|
func TestAttachmentRepo_Delete(t *testing.T) {
|
|
entity := useAttachments(t, 1)[0]
|
|
|
|
err := tRepos.Attachments.Delete(context.Background(), entity.ID)
|
|
assert.NoError(t, err)
|
|
|
|
_, err = tRepos.Attachments.Get(context.Background(), entity.ID)
|
|
assert.Error(t, err)
|
|
}
|