shad-go/dao/dao_test.go

58 lines
1.3 KiB
Go
Raw Normal View History

2022-03-26 17:41:17 +00:00
package dao
import (
"context"
"database/sql"
"sort"
"testing"
"github.com/stretchr/testify/require"
2022-04-17 15:28:33 +00:00
2022-03-26 17:41:17 +00:00
"gitlab.com/slon/shad-go/pgfixture"
)
func TestDao(t *testing.T) {
dsn := pgfixture.Start(t)
ctx := context.Background()
dao, err := CreateDao(ctx, dsn)
require.NoError(t, err)
2022-04-13 15:07:35 +00:00
defer func() { _ = dao.Close() }()
2022-03-26 17:41:17 +00:00
_, err = dao.Lookup(ctx, 42)
require.ErrorIs(t, err, sql.ErrNoRows)
aliceID, err := dao.Create(ctx, &User{Name: "Alice"})
require.NoError(t, err)
bobID, err := dao.Create(ctx, &User{Name: "Bob"})
require.NoError(t, err)
charlieID, err := dao.Create(ctx, &User{Name: "Charie"})
require.NoError(t, err)
require.Len(t, map[UserID]struct{}{aliceID: {}, bobID: {}, charlieID: {}}, 3)
alice, err := dao.Lookup(ctx, aliceID)
require.NoError(t, err)
require.Equal(t, alice, User{ID: aliceID, Name: "Alice"})
require.NoError(t, dao.Delete(ctx, bobID))
_, err = dao.Lookup(ctx, bobID)
require.ErrorIs(t, err, sql.ErrNoRows)
require.NoError(t, dao.Update(ctx, &User{ID: charlieID, Name: "Chaplin"}))
users, err := dao.List(ctx)
require.NoError(t, err)
sort.Slice(users, func(i, j int) bool {
return users[i].Name < users[j].Name
})
2022-04-11 11:54:37 +00:00
require.Equal(t, []User{
2022-03-26 17:41:17 +00:00
{ID: aliceID, Name: "Alice"},
{ID: charlieID, Name: "Chaplin"},
2022-04-11 11:54:37 +00:00
}, users)
2022-03-26 17:41:17 +00:00
}