owl-blogs/infra/entry_repository_test.go

134 lines
3.1 KiB
Go
Raw Normal View History

2023-06-25 18:04:06 +00:00
package infra_test
import (
2023-06-25 19:54:04 +00:00
"owl-blogs/app"
2023-06-25 18:04:06 +00:00
"owl-blogs/app/repository"
"owl-blogs/infra"
"owl-blogs/test"
"testing"
"time"
"github.com/stretchr/testify/require"
)
func setupRepo() repository.EntryRepository {
db := test.NewMockDb()
2023-06-25 19:54:04 +00:00
register := app.NewEntryTypeRegistry()
register.Register(&test.MockEntry{})
repo := infra.NewEntryRepository(db, register)
2023-06-25 18:04:06 +00:00
return repo
}
func TestRepoCreate(t *testing.T) {
repo := setupRepo()
entry := &test.MockEntry{}
now := time.Now()
2023-07-06 17:36:24 +00:00
err := repo.Create(entry, &now, &test.MockEntryMetaData{
2023-06-25 18:04:06 +00:00
Str: "str",
Number: 1,
Date: now,
})
require.NoError(t, err)
2023-07-06 17:36:24 +00:00
entry2, err := repo.FindById(entry.ID())
2023-06-25 18:04:06 +00:00
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()
2023-07-06 17:36:24 +00:00
err := repo.Create(entry, &now, &test.MockEntryMetaData{
2023-06-25 18:04:06 +00:00
Str: "str",
Number: 1,
Date: now,
})
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()
2023-07-06 17:36:24 +00:00
err := repo.Create(entry, &now, &test.MockEntryMetaData{
2023-06-25 18:04:06 +00:00
Str: "str",
Number: 1,
Date: now,
})
2023-07-06 17:36:24 +00:00
2023-06-25 18:04:06 +00:00
require.NoError(t, err)
entry2 := &test.MockEntry{}
now2 := time.Now()
2023-07-06 17:36:24 +00:00
err = repo.Create(entry2, &now2, &test.MockEntryMetaData{
Str: "str",
Number: 1,
Date: now,
2023-06-25 18:04:06 +00:00
})
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()
2023-07-06 17:36:24 +00:00
err := repo.Create(entry, &now, &test.MockEntryMetaData{
2023-06-25 18:04:06 +00:00
Str: "str",
Number: 1,
Date: now,
})
require.NoError(t, err)
entry2 := &test.MockEntry{}
now2 := time.Now()
2023-07-06 17:36:24 +00:00
err = repo.Create(entry2, &now2, &test.MockEntryMetaData{
2023-06-25 18:04:06 +00:00
Str: "str2",
Number: 2,
Date: now2,
})
2023-07-06 17:36:24 +00:00
require.NoError(t, err)
2023-06-25 18:04:06 +00:00
err = repo.Update(entry2)
require.NoError(t, err)
2023-07-06 17:36:24 +00:00
entry3, err := repo.FindById(entry2.ID())
2023-06-25 18:04:06 +00:00
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())
}