mirror of
https://github.com/mxpv/podsync.git
synced 2024-05-11 05:55:04 +00:00
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
package support
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-pg/pg"
|
|
"github.com/mxpv/patreon-go"
|
|
"github.com/mxpv/podsync/pkg/api"
|
|
"github.com/mxpv/podsync/pkg/model"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestCreate(t *testing.T) {
|
|
pledge := createPledge()
|
|
|
|
hook := createHandler(t)
|
|
err := hook.Hook(pledge, patreon.EventCreatePledge)
|
|
require.NoError(t, err)
|
|
|
|
model := &model.Pledge{PledgeID: 12345}
|
|
err = hook.db.Select(model)
|
|
require.NoError(t, err)
|
|
require.Equal(t, pledge.Attributes.AmountCents, model.AmountCents)
|
|
}
|
|
|
|
func TestUpdate(t *testing.T) {
|
|
pledge := createPledge()
|
|
|
|
hook := createHandler(t)
|
|
err := hook.Hook(pledge, patreon.EventCreatePledge)
|
|
require.NoError(t, err)
|
|
|
|
pledge.Attributes.AmountCents = 999
|
|
|
|
err = hook.Hook(pledge, patreon.EventUpdatePledge)
|
|
require.NoError(t, err)
|
|
|
|
model := &model.Pledge{PledgeID: 12345}
|
|
err = hook.db.Select(model)
|
|
require.NoError(t, err)
|
|
require.Equal(t, 999, model.AmountCents)
|
|
}
|
|
|
|
func TestDelete(t *testing.T) {
|
|
pledge := createPledge()
|
|
hook := createHandler(t)
|
|
|
|
err := hook.Hook(pledge, patreon.EventCreatePledge)
|
|
require.NoError(t, err)
|
|
|
|
err = hook.Hook(pledge, patreon.EventDeletePledge)
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
func TestFindPledge(t *testing.T) {
|
|
pledge := createPledge()
|
|
hook := createHandler(t)
|
|
|
|
err := hook.Hook(pledge, patreon.EventCreatePledge)
|
|
require.NoError(t, err)
|
|
|
|
res, err := hook.FindPledge("67890")
|
|
require.NoError(t, err)
|
|
require.Equal(t, res.AmountCents, pledge.Attributes.AmountCents)
|
|
}
|
|
|
|
func TestGetFeatureLevel(t *testing.T) {
|
|
pledge := createPledge()
|
|
hook := createHandler(t)
|
|
|
|
err := hook.Hook(pledge, patreon.EventCreatePledge)
|
|
require.NoError(t, err)
|
|
|
|
require.Equal(t, api.PodcasterFeature, hook.GetFeatureLevel(creatorID))
|
|
require.Equal(t, api.DefaultFeatures, hook.GetFeatureLevel("xyz"))
|
|
require.Equal(t, api.ExtendedPagination, hook.GetFeatureLevel(pledge.Relationships.Patron.Data.ID))
|
|
}
|
|
|
|
func createHandler(t *testing.T) *Patreon {
|
|
opts, err := pg.ParseURL("postgres://postgres:@localhost/podsync?sslmode=disable")
|
|
if err != nil {
|
|
require.NoError(t, err)
|
|
}
|
|
|
|
db := pg.Connect(opts)
|
|
|
|
_, err = db.Model(&model.Pledge{}).Where("1=1").Delete()
|
|
require.NoError(t, err)
|
|
|
|
return NewPatreon(db)
|
|
}
|
|
|
|
func createPledge() *patreon.Pledge {
|
|
pledge := &patreon.Pledge{
|
|
ID: "12345",
|
|
Type: "pledge",
|
|
}
|
|
|
|
pledge.Attributes.AmountCents = 400
|
|
pledge.Attributes.CreatedAt = patreon.NullTime{Valid: true, Time: time.Now().UTC()}
|
|
|
|
pledge.Relationships.Patron = &patreon.PatronRelationship{}
|
|
pledge.Relationships.Patron.Data.ID = "67890"
|
|
|
|
return pledge
|
|
}
|