145 lines
3.4 KiB
Go
145 lines
3.4 KiB
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())
|
|
}
|