mirror of
https://github.com/mxpv/podsync.git
synced 2024-05-11 05:55:04 +00:00
107 lines
2.3 KiB
Go
107 lines
2.3 KiB
Go
package storage
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/mxpv/podsync/pkg/api"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestPgStorage_CreateFeed(t *testing.T) {
|
|
feed := &api.Feed{
|
|
HashId: "xyz",
|
|
Provider: api.ProviderYoutube,
|
|
LinkType: api.LinkTypeChannel,
|
|
ItemId: "123",
|
|
}
|
|
|
|
client := createClient(t)
|
|
err := client.CreateFeed(feed)
|
|
require.NoError(t, err)
|
|
require.True(t, feed.Id > 0)
|
|
}
|
|
|
|
func TestPgStorage_CreateFeedWithDuplicate(t *testing.T) {
|
|
feed := &api.Feed{
|
|
HashId: "123",
|
|
Provider: api.ProviderYoutube,
|
|
LinkType: api.LinkTypeChannel,
|
|
ItemId: "123",
|
|
}
|
|
|
|
client := createClient(t)
|
|
err := client.CreateFeed(feed)
|
|
require.NoError(t, err)
|
|
|
|
// Ensure 1 record
|
|
count, err := client.db.Model(&api.Feed{}).Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, count)
|
|
|
|
// Insert duplicated feed
|
|
err = client.CreateFeed(feed)
|
|
require.NoError(t, err)
|
|
|
|
// Check no duplicates inserted
|
|
count, err = client.db.Model(&api.Feed{}).Count()
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, count)
|
|
}
|
|
|
|
func TestPgStorage_GetFeed(t *testing.T) {
|
|
feed := &api.Feed{
|
|
HashId: "xyz",
|
|
UserId: "123",
|
|
Provider: api.ProviderYoutube,
|
|
LinkType: api.LinkTypeChannel,
|
|
ItemId: "123",
|
|
}
|
|
|
|
client := createClient(t)
|
|
client.CreateFeed(feed)
|
|
|
|
out, err := client.GetFeed("xyz")
|
|
require.NoError(t, err)
|
|
require.Equal(t, feed.Id, out.Id)
|
|
}
|
|
|
|
func TestPgStorage_UpdateLastAccess(t *testing.T) {
|
|
feed := &api.Feed{
|
|
HashId: "xyz",
|
|
UserId: "123",
|
|
Provider: api.ProviderYoutube,
|
|
LinkType: api.LinkTypeChannel,
|
|
ItemId: "123",
|
|
}
|
|
|
|
client := createClient(t)
|
|
err := client.CreateFeed(feed)
|
|
require.NoError(t, err)
|
|
|
|
lastAccess := feed.LastAccess
|
|
require.True(t, lastAccess.Unix() > 0)
|
|
|
|
last, err := client.GetFeed("xyz")
|
|
require.NoError(t, err)
|
|
|
|
require.NotEmpty(t, last.HashId)
|
|
require.NotEmpty(t, last.UserId)
|
|
require.NotEmpty(t, last.Provider)
|
|
require.NotEmpty(t, last.LinkType)
|
|
require.NotEmpty(t, last.ItemId)
|
|
|
|
require.True(t, last.LastAccess.UnixNano() > lastAccess.UnixNano())
|
|
}
|
|
|
|
const TestDatabaseConnectionUrl = "postgres://postgres:@localhost/podsync?sslmode=disable"
|
|
|
|
func createClient(t *testing.T) *PgStorage {
|
|
pg, err := NewPgStorage(&PgConfig{ConnectionUrl: TestDatabaseConnectionUrl})
|
|
require.NoError(t, err)
|
|
|
|
_, err = pg.db.Model(&api.Feed{}).Where("1=1").Delete()
|
|
require.NoError(t, err)
|
|
|
|
return pg
|
|
}
|