homebox/backend/internal/repo/repo_locations_test.go

package repo

import (
	"context"
	"testing"

	"github.com/hay-kot/content/backend/ent"
	"github.com/hay-kot/content/backend/internal/types"
	"github.com/hay-kot/content/backend/pkgs/faker"
	"github.com/stretchr/testify/assert"
)

var fk = faker.NewFaker()

func locationFactory() types.LocationCreate {
	return types.LocationCreate{
		Name:        fk.RandomString(10),
		Description: fk.RandomString(100),
	}
}

func Test_Locations_Get(t *testing.T) {
	loc, err := testRepos.Locations.Create(context.Background(), testGroup.ID, locationFactory())
	assert.NoError(t, err)

	// Get by ID
	foundLoc, err := testRepos.Locations.Get(context.Background(), loc.ID)
	assert.NoError(t, err)
	assert.Equal(t, loc.ID, foundLoc.ID)

	testRepos.Locations.Delete(context.Background(), loc.ID)
}

func Test_Locations_GetAll(t *testing.T) {
	created := make([]*ent.Location, 6)

	for i := 0; i < 6; i++ {
		result, err := testRepos.Locations.Create(context.Background(), testGroup.ID, types.LocationCreate{
			Name:        fk.RandomString(10),
			Description: fk.RandomString(100),
		})

		assert.NoError(t, err)
		created[i] = result
	}

	locations, err := testRepos.Locations.GetAll(context.Background(), testGroup.ID)
	assert.NoError(t, err)
	assert.Equal(t, 6, len(locations))

	for _, loc := range created {
		testRepos.Locations.Delete(context.Background(), loc.ID)
	}
}

func Test_Locations_Create(t *testing.T) {
	loc, err := testRepos.Locations.Create(context.Background(), testGroup.ID, locationFactory())
	assert.NoError(t, err)

	// Get by ID
	foundLoc, err := testRepos.Locations.Get(context.Background(), loc.ID)
	assert.NoError(t, err)
	assert.Equal(t, loc.ID, foundLoc.ID)

	testRepos.Locations.Delete(context.Background(), loc.ID)
}

func Test_Locations_Update(t *testing.T) {
	loc, err := testRepos.Locations.Create(context.Background(), testGroup.ID, locationFactory())
	assert.NoError(t, err)

	updateData := types.LocationUpdate{
		ID:          loc.ID,
		Name:        fk.RandomString(10),
		Description: fk.RandomString(100),
	}

	update, err := testRepos.Locations.Update(context.Background(), updateData)
	assert.NoError(t, err)

	foundLoc, err := testRepos.Locations.Get(context.Background(), loc.ID)
	assert.NoError(t, err)

	assert.Equal(t, update.ID, foundLoc.ID)
	assert.Equal(t, update.Name, foundLoc.Name)
	assert.Equal(t, update.Description, foundLoc.Description)

	testRepos.Locations.Delete(context.Background(), loc.ID)
}

func Test_Locations_Delete(t *testing.T) {
	loc, err := testRepos.Locations.Create(context.Background(), testGroup.ID, locationFactory())
	assert.NoError(t, err)

	err = testRepos.Locations.Delete(context.Background(), loc.ID)
	assert.NoError(t, err)

	_, err = testRepos.Locations.Get(context.Background(), loc.ID)
	assert.Error(t, err)
}