forked from mirrors/homebox
31b34241e0
* change /content/ -> /homebox/ * add cache to code generators * update env variables to set data storage * update env variables * set env variables in prod container * implement attachment post route (WIP) * get attachment endpoint * attachment download * implement string utilities lib * implement generic drop zone * use explicit truncate * remove clean dir * drop strings composable for lib * update item types and add attachments * add attachment API * implement service context * consolidate API code * implement editing attachments * implement upload limit configuration * improve error handling * add docs for max upload size * fix test cases
62 lines
1.4 KiB
Go
62 lines
1.4 KiB
Go
package services
|
|
|
|
import (
|
|
"context"
|
|
"os"
|
|
"path"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/hay-kot/homebox/backend/internal/types"
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestItemService_AddAttachment(t *testing.T) {
|
|
temp := os.TempDir()
|
|
|
|
svc := &ItemService{
|
|
repo: tRepos,
|
|
filepath: temp,
|
|
}
|
|
|
|
loc, err := tSvc.Location.Create(context.Background(), tGroup.ID, types.LocationCreate{
|
|
Description: "test",
|
|
Name: "test",
|
|
})
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, loc)
|
|
|
|
itmC := types.ItemCreate{
|
|
Name: fk.Str(10),
|
|
Description: fk.Str(10),
|
|
LocationID: loc.ID,
|
|
}
|
|
|
|
itm, err := svc.Create(context.Background(), tGroup.ID, itmC)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, itm)
|
|
t.Cleanup(func() {
|
|
err := svc.repo.Items.Delete(context.Background(), itm.ID)
|
|
assert.NoError(t, err)
|
|
})
|
|
|
|
contents := fk.Str(1000)
|
|
reader := strings.NewReader(contents)
|
|
|
|
// Setup
|
|
afterAttachment, err := svc.AttachmentAdd(tCtx, itm.ID, "testfile.txt", "attachment", reader)
|
|
assert.NoError(t, err)
|
|
assert.NotNil(t, afterAttachment)
|
|
|
|
// Check that the file exists
|
|
storedPath := afterAttachment.Attachments[0].Document.Path
|
|
|
|
// {root}/{group}/{item}/{attachment}
|
|
assert.Equal(t, path.Join(temp, tGroup.ID.String(), itm.ID.String(), "testfile.txt"), storedPath)
|
|
|
|
// Check that the file contents are correct
|
|
bts, err := os.ReadFile(storedPath)
|
|
assert.NoError(t, err)
|
|
assert.Equal(t, contents, string(bts))
|
|
|
|
}
|