velox-bot/internal/db/repos/settingsrepo/meeting_integration_test.go

//go:build integration

// Runs against a real Postgres with the dashboard API's migrations applied
// (the lobby limit is a trigger from migration 000020):
//
//	TEST_DB=postgres://... go test -tags integration ./internal/db/repos/settingsrepo
package settingsrepo

import (
	"context"
	"database/sql"
	"errors"
	"os"
	"sync"
	"testing"

	_ "github.com/jackc/pgx/v5/stdlib"
)

func testRepo(t *testing.T) (*Repo, int64) {
	t.Helper()
	dsn := os.Getenv("TEST_DB")
	if dsn == "" {
		t.Skip("TEST_DB not set")
	}
	db, err := sql.Open("pgx", dsn)
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { db.Close() })
	const guild = 4242
	if _, err := db.Exec(`DELETE FROM meeting_lobbies WHERE guild_id IN ($1, $2)`, guild, guild+1); err != nil {
		t.Fatal(err)
	}
	return NewRepo(db), guild
}

func TestMeetingLobbies(t *testing.T) {
	r, g := testRepo(t)
	ctx := context.Background()

	if ids, err := r.ListMeetingLobbies(ctx, g); err != nil || len(ids) != 0 {
		t.Fatalf("a fresh server has no lobbies: %v %v", ids, err)
	}

	for ch := int64(1); ch <= MaxMeetingLobbies; ch++ {
		added, err := r.AddMeetingLobby(ctx, g, ch)
		if err != nil || !added {
			t.Fatalf("adding lobby %d: added=%v err=%v", ch, added, err)
		}
	}
	if _, err := r.AddMeetingLobby(ctx, g, 99); !errors.Is(err, ErrMeetingLobbyLimit) {
		t.Errorf("an 11th lobby must hit the limit, got %v", err)
	}
	if added, err := r.AddMeetingLobby(ctx, g, 3); err != nil || added {
		t.Errorf("adding an existing lobby is a no-op even at the limit: added=%v err=%v", added, err)
	}
	if added, err := r.AddMeetingLobby(ctx, g+1, 99); err != nil || !added {
		t.Errorf("the limit is per server: added=%v err=%v", added, err)
	}

	if ok, _ := r.IsMeetingLobby(ctx, g, 7); !ok {
		t.Error("7 is a lobby")
	}
	if ok, _ := r.IsMeetingLobby(ctx, g, 99); ok {
		t.Error("99 isn't a lobby of this server (it is one of the other server's)")
	}

	if removed, err := r.RemoveMeetingLobby(ctx, g, 7); err != nil || !removed {
		t.Errorf("removing 7: removed=%v err=%v", removed, err)
	}
	if removed, _ := r.RemoveMeetingLobby(ctx, g, 7); removed {
		t.Error("7 was already removed")
	}
	if ok, _ := r.IsMeetingLobby(ctx, g, 7); ok {
		t.Error("7 is no longer a lobby")
	}
	if added, err := r.AddMeetingLobby(ctx, g, 99); err != nil || !added {
		t.Errorf("removing one makes room: added=%v err=%v", added, err)
	}

	ids, err := r.ListMeetingLobbies(ctx, g)
	if err != nil || len(ids) != MaxMeetingLobbies || ids[len(ids)-1] != 99 {
		t.Errorf("oldest first, newest last: %v %v", ids, err)
	}
}

// Two people adding a lobby at the same moment must not both fit under the limit.
func TestMeetingLobbiesConcurrent(t *testing.T) {
	r, g := testRepo(t)
	ctx := context.Background()
	for ch := int64(1); ch < MaxMeetingLobbies; ch++ {
		if _, err := r.AddMeetingLobby(ctx, g, ch); err != nil {
			t.Fatal(err)
		}
	}

	var wg sync.WaitGroup
	var mu sync.Mutex
	succeeded := 0
	for i := 0; i < 8; i++ {
		wg.Add(1)
		go func(ch int64) {
			defer wg.Done()
			if added, err := r.AddMeetingLobby(ctx, g, ch); err == nil && added {
				mu.Lock()
				succeeded++
				mu.Unlock()
			}
		}(int64(100 + i))
	}
	wg.Wait()
	if succeeded != 1 {
		t.Errorf("exactly one of the concurrent adds fits, %d did", succeeded)
	}
	if ids, _ := r.ListMeetingLobbies(ctx, g); len(ids) != MaxMeetingLobbies {
		t.Errorf("have %d lobbies, want %d", len(ids), MaxMeetingLobbies)
	}
}