Publish ITWorx Pulse source

internal/eventapi/postgres_integration_test.go

@@ -0,0 +1,61 @@
+package eventapi
+
+import (
+	"context"
+	"crypto/rand"
+	"fmt"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/itworx/pulse/internal/database"
+)
+
+func integrationID(t *testing.T) string {
+	t.Helper()
+	var value [16]byte
+	if _, err := rand.Read(value[:]); err != nil {
+		t.Fatal(err)
+	}
+	value[6] = (value[6] & 0x0f) | 0x40
+	value[8] = (value[8] & 0x3f) | 0x80
+	return fmt.Sprintf("%x-%x-%x-%x-%x", value[0:4], value[4:6], value[6:8], value[8:10], value[10:16])
+}
+
+func TestPostgresStoreListsNewestEventsDeterministically(t *testing.T) {
+	dsn := os.Getenv("PULSE_TEST_DATABASE_URL")
+	if dsn == "" {
+		t.Skip("PULSE_TEST_DATABASE_URL is not set")
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), 60*time.Second)
+	defer cancel()
+	pool, err := database.NewPool(ctx, database.Config{URL: dsn, MaxConns: 2, MinConns: 1})
+	if err != nil {
+		t.Fatal(err)
+	}
+	defer pool.Close()
+	if err := database.Migrate(ctx, pool); err != nil {
+		t.Fatal(err)
+	}
+	sourceID, olderID, newerID := integrationID(t), integrationID(t), integrationID(t)
+	if _, err := pool.Exec(ctx, `INSERT INTO data_sources(id,type,name,configuration_ref) VALUES($1,'fixture','event-api-test','test')`, sourceID); err != nil {
+		t.Fatal(err)
+	}
+	t.Cleanup(func() {
+		cleanup, stop := context.WithTimeout(context.Background(), 10*time.Second)
+		defer stop()
+		_, _ = pool.Exec(cleanup, `DELETE FROM events WHERE source_id=$1`, sourceID)
+		_, _ = pool.Exec(cleanup, `DELETE FROM data_sources WHERE id=$1`, sourceID)
+	})
+	at := time.Date(2099, 8, 12, 2, 0, 0, 0, time.UTC)
+	if _, err := pool.Exec(ctx, `INSERT INTO events(id,event_type,severity,source_id,occurred_at,dedup_key,summary) VALUES($1,'older','info',$3,$4,'older','Ouder'),($2,'newer','warning',$3,$5,'newer','Nieuwer')`, olderID, newerID, sourceID, at, at.Add(time.Minute)); err != nil {
+		t.Fatal(err)
+	}
+	items, err := (PostgresStore{Pool: pool}).List(ctx, 2)
+	if err != nil {
+		t.Fatal(err)
+	}
+	if len(items) < 2 || items[0].ID != newerID || items[1].ID != olderID {
+		t.Fatalf("events not deterministically newest first: %#v", items)
+	}
+}