owl-blogs/infra/entry_repository_test.go

package infra_test

import (
	"owl-blogs/app/repository"
	"owl-blogs/infra"
	"owl-blogs/test"
	"testing"
	"time"

	"github.com/stretchr/testify/require"
)

func setupRepo() repository.EntryRepository {
	db := test.NewMockDb()
	repo := infra.NewEntryRepository(db)
	repo.RegisterEntryType(&test.MockEntry{})
	return repo
}

func TestRepoRegister(t *testing.T) {
	db := test.NewMockDb()
	repo := infra.NewEntryRepository(db)
	err := repo.RegisterEntryType(&test.MockEntry{})
	require.NoError(t, err)

	err = repo.RegisterEntryType(&test.MockEntry{})
	require.Error(t, err)
}

func TestRepoCreate(t *testing.T) {
	repo := setupRepo()

	entry := &test.MockEntry{}
	now := time.Now()
	entry.Create("id", "content", &now, &test.MockEntryMetaData{
		Str:    "str",
		Number: 1,
		Date:   now,
	})
	err := repo.Create(entry)
	require.NoError(t, err)

	entry2, err := repo.FindById("id")
	require.NoError(t, err)
	require.Equal(t, entry.ID(), entry2.ID())
	require.Equal(t, entry.Content(), entry2.Content())
	require.Equal(t, entry.PublishedAt().Unix(), entry2.PublishedAt().Unix())
	meta := entry.MetaData().(*test.MockEntryMetaData)
	meta2 := entry2.MetaData().(*test.MockEntryMetaData)
	require.Equal(t, meta.Str, meta2.Str)
	require.Equal(t, meta.Number, meta2.Number)
	require.Equal(t, meta.Date.Unix(), meta2.Date.Unix())
}

func TestRepoDelete(t *testing.T) {
	repo := setupRepo()

	entry := &test.MockEntry{}
	now := time.Now()
	entry.Create("id", "content", &now, &test.MockEntryMetaData{
		Str:    "str",
		Number: 1,
		Date:   now,
	})
	err := repo.Create(entry)
	require.NoError(t, err)

	err = repo.Delete(entry)
	require.NoError(t, err)

	_, err = repo.FindById("id")
	require.Error(t, err)
}

func TestRepoFindAll(t *testing.T) {
	repo := setupRepo()

	entry := &test.MockEntry{}
	now := time.Now()
	entry.Create("id", "content", &now, &test.MockEntryMetaData{
		Str:    "str",
		Number: 1,
		Date:   now,
	})
	err := repo.Create(entry)
	require.NoError(t, err)

	entry2 := &test.MockEntry{}
	now2 := time.Now()
	entry2.Create("id2", "content2", &now2, &test.MockEntryMetaData{
		Str:    "str2",
		Number: 2,
		Date:   now2,
	})
	err = repo.Create(entry2)
	require.NoError(t, err)

	entries, err := repo.FindAll(nil)
	require.NoError(t, err)
	require.Equal(t, 2, len(entries))

	entries, err = repo.FindAll(&[]string{"MockEntry"})
	require.NoError(t, err)
	require.Equal(t, 2, len(entries))

	entries, err = repo.FindAll(&[]string{"MockEntry2"})
	require.NoError(t, err)
	require.Equal(t, 0, len(entries))

}

func TestRepoUpdate(t *testing.T) {
	repo := setupRepo()

	entry := &test.MockEntry{}
	now := time.Now()
	entry.Create("id", "content", &now, &test.MockEntryMetaData{
		Str:    "str",
		Number: 1,
		Date:   now,
	})
	err := repo.Create(entry)
	require.NoError(t, err)

	entry2 := &test.MockEntry{}
	now2 := time.Now()
	entry2.Create("id", "content2", &now2, &test.MockEntryMetaData{
		Str:    "str2",
		Number: 2,
		Date:   now2,
	})
	err = repo.Update(entry2)
	require.NoError(t, err)

	entry3, err := repo.FindById("id")
	require.NoError(t, err)
	require.Equal(t, entry3.Content(), entry2.Content())
	require.Equal(t, entry3.PublishedAt().Unix(), entry2.PublishedAt().Unix())
	meta := entry3.MetaData().(*test.MockEntryMetaData)
	meta2 := entry2.MetaData().(*test.MockEntryMetaData)
	require.Equal(t, meta.Str, meta2.Str)
	require.Equal(t, meta.Number, meta2.Number)
	require.Equal(t, meta.Date.Unix(), meta2.Date.Unix())
}