package message_test

import (
	"fmt"
	"net/url"
	"os"
	"testing"

	"github.com/stretchr/testify/require"
	"heckel.io/ntfy/v2/message"
	"heckel.io/ntfy/v2/postgres"
	"heckel.io/ntfy/v2/util"
)

func newTestPostgresStore(t *testing.T) message.Store {
	dsn := os.Getenv("NTFY_TEST_DATABASE_URL")
	if dsn == "" {
		t.Skip("NTFY_TEST_DATABASE_URL not set, skipping PostgreSQL tests")
	}
	schema := fmt.Sprintf("test_%s", util.RandomString(10))
	u, err := url.Parse(dsn)
	require.Nil(t, err)
	q := u.Query()
	q.Set("pool_max_conns", "2")
	u.RawQuery = q.Encode()
	dsn = u.String()
	q.Set("search_path", schema)
	u.RawQuery = q.Encode()
	schemaDSN := u.String()
	setupDB, err := postgres.OpenDB(dsn)
	require.Nil(t, err)
	_, err = setupDB.Exec(fmt.Sprintf("CREATE SCHEMA %s", schema))
	require.Nil(t, err)
	require.Nil(t, setupDB.Close())
	db, err := postgres.OpenDB(schemaDSN)
	require.Nil(t, err)
	store, err := message.NewPostgresStore(db, 0, 0)
	require.Nil(t, err)
	t.Cleanup(func() {
		store.Close()
		cleanDB, err := postgres.OpenDB(dsn)
		if err == nil {
			cleanDB.Exec(fmt.Sprintf("DROP SCHEMA %s CASCADE", schema))
			cleanDB.Close()
		}
	})
	return store
}

func TestPostgresStore_Messages(t *testing.T) {
	testCacheMessages(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessagesLock(t *testing.T) {
	testCacheMessagesLock(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessagesScheduled(t *testing.T) {
	testCacheMessagesScheduled(t, newTestPostgresStore(t))
}

func TestPostgresStore_Topics(t *testing.T) {
	testCacheTopics(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessagesTagsPrioAndTitle(t *testing.T) {
	testCacheMessagesTagsPrioAndTitle(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessagesSinceID(t *testing.T) {
	testCacheMessagesSinceID(t, newTestPostgresStore(t))
}

func TestPostgresStore_Prune(t *testing.T) {
	testCachePrune(t, newTestPostgresStore(t))
}

func TestPostgresStore_Attachments(t *testing.T) {
	testCacheAttachments(t, newTestPostgresStore(t))
}

func TestPostgresStore_AttachmentsExpired(t *testing.T) {
	testCacheAttachmentsExpired(t, newTestPostgresStore(t))
}

func TestPostgresStore_Sender(t *testing.T) {
	testSender(t, newTestPostgresStore(t))
}

func TestPostgresStore_DeleteScheduledBySequenceID(t *testing.T) {
	testDeleteScheduledBySequenceID(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessageByID(t *testing.T) {
	testMessageByID(t, newTestPostgresStore(t))
}

func TestPostgresStore_MarkPublished(t *testing.T) {
	testMarkPublished(t, newTestPostgresStore(t))
}

func TestPostgresStore_ExpireMessages(t *testing.T) {
	testExpireMessages(t, newTestPostgresStore(t))
}

func TestPostgresStore_MarkAttachmentsDeleted(t *testing.T) {
	testMarkAttachmentsDeleted(t, newTestPostgresStore(t))
}

func TestPostgresStore_Stats(t *testing.T) {
	testStats(t, newTestPostgresStore(t))
}

func TestPostgresStore_AddMessages(t *testing.T) {
	testAddMessages(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessagesDue(t *testing.T) {
	testMessagesDue(t, newTestPostgresStore(t))
}

func TestPostgresStore_MessageFieldRoundTrip(t *testing.T) {
	testMessageFieldRoundTrip(t, newTestPostgresStore(t))
}