mxpv-podsync/web/pkg/storage/pg_test.go

package storage

import (
	"testing"

	"github.com/mxpv/podsync/web/pkg/api"
	"github.com/stretchr/testify/require"
)

func TestPgStorage_CreateFeed(t *testing.T) {
	feed := &api.Feed{
		HashId:   "xyz",
		Provider: api.Youtube,
		LinkType: api.Channel,
		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.Youtube,
		LinkType: api.Channel,
		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.Youtube,
		LinkType: api.Channel,
		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.Youtube,
		LinkType: api.Channel,
		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
}