server/database/drivers/mysqldriver/items_test.go

package mysqldriver

import (
	"context"
	"git.aiterp.net/stufflog/server/internal/slerrors"
	"git.aiterp.net/stufflog/server/models"
	"github.com/stretchr/testify/assert"
	"testing"
	"time"
)

var item1ImageURL = "data:image/gif;base64,R0lGODlhAQABAIAAAAAAAP///yH5BAEAAAAALAAAAAABAAEAAAIBRAA7"

var item1 = models.Item{
	Name:        "Salmon Fillet - 500g",
	Description: "Best fish",
	Tags:        []string{"Groceries"},
	ImageURL:    &item1ImageURL,
}
var item1Updated = models.Item{
	Name:        "Salmon Fillet - 450g",
	Description: "Do not handle under suspicious circumstances.",
	Tags:        []string{"Groceries"},
	ImageURL:    nil,
}
var item2 = models.Item{
	Name:        "Tape - Basic",
	Description: "",
	Tags:        []string{"Groceries", "Hardware", "Office Supplies"},
	ImageURL:    nil,
}
var item3 = models.Item{
	Name:        "Flour - Wheat - 1kg",
	Description: "For bread and stuff",
	Tags:        []string{"Groceries"},
	ImageURL:    nil,
}
var item4 = models.Item{
	Name:        "Flour - Wheat - 2kg",
	Description: "For more bread and stuff",
	Tags:        []string{"Groceries"},
	ImageURL:    nil,
}

func TestItemRepository(t *testing.T) {
	items := testDB.items
	ctx, cancel := context.WithTimeout(context.Background(), time.Second*10)
	defer cancel()

	assert.NoError(t, clearTable("item"))
	assert.NoError(t, clearTable("item_tag"))

	// INSERT
	// IDs are random, so test data needs to be changed before comparison.
	result, err := items.Insert(ctx, item1)
	assert.NoError(t, err)
	if result != nil {
		item1.ID = result.ID
		item1Updated.ID = result.ID
	}
	assert.Equal(t, item1, *result)
	result, err = items.Insert(ctx, item2)
	assert.NoError(t, err)
	if result != nil {
		item2.ID = result.ID
	}
	assert.Equal(t, item2, *result)
	result, err = items.Insert(ctx, item3)
	assert.NoError(t, err)
	if result != nil {
		item3.ID = result.ID
	}
	assert.Equal(t, item3, *result)
	result, err = items.Insert(ctx, item4)
	assert.NoError(t, err)
	if result != nil {
		item4.ID = result.ID
	}
	assert.Equal(t, item4, *result)

	if t.Failed() {
		return
	}

	// FIND
	result, err = items.Find(ctx, item1.ID)
	assert.NoError(t, err)
	assert.Equal(t, &item1, result)
	result, err = items.Find(ctx, item2.ID)
	assert.NoError(t, err)
	assert.Equal(t, &item2, result)

	// FIND't
	result, err = items.Find(ctx, "Iobviouslyinvalidid")
	assert.Error(t, err)
	assert.True(t, slerrors.IsNotFound(err))
	assert.Nil(t, result)

	// LIST
	results, err := items.List(ctx, models.ItemFilter{})
	assert.NoError(t, err)
	assert.Equal(t, []*models.Item{&item3, &item4, &item1, &item2}, results)
	results, err = items.List(ctx, models.ItemFilter{
		Tags: []string{"Hardware"},
	})
	assert.NoError(t, err)
	assert.Equal(t, []*models.Item{&item2}, results)
	results, err = items.List(ctx, models.ItemFilter{
		ItemIDs: []string{item1.ID, item2.ID},
	})
	assert.NoError(t, err)
	assert.Equal(t, []*models.Item{&item1, &item2}, results)

	// UPDATE
	err = items.Save(ctx, item1Updated)
	assert.NoError(t, err)

	// FIND after UPDATE
	result, err = items.Find(ctx, item1.ID)
	assert.NoError(t, err)
	assert.NotEqual(t, &item1, result)
	assert.Equal(t, &item1Updated, result)

	// TAGS
	allTags, err := items.GetTags(ctx)
	assert.NoError(t, err)
	assert.Equal(t, item2.Tags, allTags)

	// DELETE
	err = items.Delete(ctx, item2)
	assert.NoError(t, err)

	// LIST after DELETE
	results, err = items.List(ctx, models.ItemFilter{})
	assert.NoError(t, err)
	assert.Equal(t, []*models.Item{&item3, &item4, &item1Updated}, results)

	// TAGS after DELETE
	allTags, err = items.GetTags(ctx)
	assert.NoError(t, err)
	assert.Equal(t, item1.Tags, allTags)
}