backend barebone done, big chunk of work done

server/internal/store/integration_test.go

@@ -0,0 +1,111 @@
+package store
+
+import (
+	"context"
+	"os"
+	"testing"
+	"time"
+
+	"github.com/jackc/pgx/v5/pgxpool"
+)
+
+// TestIntegration exercises basic CRUD operations against a real database.
+func TestIntegration(t *testing.T) {
+	url := os.Getenv("MESSENGER_TEST_DATABASE_URL")
+	if url == "" {
+		t.Skip("skipping integration test; MESSENGER_TEST_DATABASE_URL not set")
+	}
+	ctx := context.Background()
+	pool, err := pgxpool.New(ctx, url)
+	if err != nil {
+		t.Fatalf("unable to connect to db: %v", err)
+	}
+	defer pool.Close()
+
+	db := New(pool)
+	if err := db.Migrate(ctx); err != nil {
+		t.Fatalf("Migrate failed: %v", err)
+	}
+
+	// Users
+	u1, err := db.CreateUser(ctx, "user1", "User One", "hash1")
+	if err != nil {
+		t.Fatalf("CreateUser u1: %v", err)
+	}
+	u2, err := db.CreateUser(ctx, "user2", "User Two", "hash2")
+	if err != nil {
+		t.Fatalf("CreateUser u2: %v", err)
+	}
+	if _, err := db.GetUserByUsername(ctx, u1.Username); err != nil {
+		t.Fatalf("GetUserByUsername: %v", err)
+	}
+	if _, err := db.GetUserByID(ctx, u2.ID); err != nil {
+		t.Fatalf("GetUserByID: %v", err)
+	}
+
+	// Sessions
+	exp := time.Now().Add(time.Hour)
+	sess, err := db.CreateSession(ctx, u1.ID, "rh1", "ua", "127.0.0.1", exp)
+	if err != nil {
+		t.Fatalf("CreateSession: %v", err)
+	}
+	if _, err := db.GetSessionByRefreshHash(ctx, sess.RefreshHash); err != nil {
+		t.Fatalf("GetSessionByRefreshHash: %v", err)
+	}
+	if err := db.RevokeSession(ctx, sess.ID); err != nil {
+		t.Fatalf("RevokeSession: %v", err)
+	}
+	if err := db.RevokeAllSessions(ctx, u1.ID); err != nil {
+		t.Fatalf("RevokeAllSessions: %v", err)
+	}
+
+	// Contacts
+	if err := db.AddContact(ctx, u1.ID, u2.ID); err != nil {
+		t.Fatalf("AddContact: %v", err)
+	}
+	if ok, err := db.IsContact(ctx, u1.ID, u2.ID); err != nil || !ok {
+		t.Fatalf("IsContact: %v, %v", ok, err)
+	}
+	list, err := db.ListContacts(ctx, u1.ID)
+	if err != nil || len(list) != 1 {
+		t.Fatalf("ListContacts: got %v, err %v", list, err)
+	}
+	if err := db.RemoveContact(ctx, u1.ID, u2.ID); err != nil {
+		t.Fatalf("RemoveContact: %v", err)
+	}
+
+	// Conversations and messages
+	conv, err := db.CreateDM(ctx, u1.ID, u2.ID)
+	if err != nil {
+		t.Fatalf("CreateDM: %v", err)
+	}
+	if conv2, err := db.CreateDM(ctx, u1.ID, u2.ID); err != nil || conv2.ID != conv.ID {
+		t.Fatalf("CreateDM idempotent: %v, %v", err, conv2)
+	}
+	if ok, err := db.IsMember(ctx, conv.ID, u1.ID); err != nil || !ok {
+		t.Fatalf("IsMember: %v, %v", ok, err)
+	}
+	ids, err := db.ConversationIDsForUser(ctx, u1.ID)
+	if err != nil || len(ids) == 0 {
+		t.Fatalf("ConversationIDsForUser: %v, %v", ids, err)
+	}
+	sums, err := db.ListConversationsForUser(ctx, u1.ID)
+	if err != nil || len(sums) != 1 {
+		t.Fatalf("ListConversationsForUser: %v, %v", sums, err)
+	}
+
+	msg, err := db.InsertMessage(ctx, conv.ID, u1.ID, "hello")
+	if err != nil {
+		t.Fatalf("InsertMessage: %v", err)
+	}
+	if _, err := db.GetMessageByID(ctx, msg.ID); err != nil {
+		t.Fatalf("GetMessageByID: %v", err)
+	}
+	msgs, err := db.ListMessages(ctx, conv.ID, 0, 10)
+	if err != nil || len(msgs) != 1 {
+		t.Fatalf("ListMessages: %v, %v", msgs, err)
+	}
+	if err := db.UpdateLastRead(ctx, conv.ID, u2.ID); err != nil {
+		t.Fatalf("UpdateLastRead: %v", err)
+	}
+}