Rework project structure

pkg/storage/pg_test.go

@@ -0,0 +1,106 @@
+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.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
+}