package state

import (
	"testing"
)

func TestUnreadTable(t *testing.T) {
	db, close := connectToDB(t)
	defer close()
	table := NewUnreadTable(db)
	userID := "@alice:localhost"
	roomA := "!TestUnreadTableA:localhost"
	roomB := "!TestUnreadTableB:localhost"
	roomC := "!TestUnreadTableC:localhost"

	two := 2
	one := 1
	zero := 0

	// try all kinds of insertions
	assertNoError(t, table.UpdateUnreadCounters(userID, roomA, &two, &one)) // both
	assertNoError(t, table.UpdateUnreadCounters(userID, roomB, &two, nil))  // one
	assertNoError(t, table.UpdateUnreadCounters(userID, roomC, nil, &two))  // one
	assertUnread(t, table, userID, roomA, 2, 1)
	assertUnread(t, table, userID, roomB, 2, 0)
	assertUnread(t, table, userID, roomC, 0, 2)

	// try all kinds of updates
	assertNoError(t, table.UpdateUnreadCounters(userID, roomA, &zero, nil))   // one
	assertNoError(t, table.UpdateUnreadCounters(userID, roomB, nil, &two))    // one
	assertNoError(t, table.UpdateUnreadCounters(userID, roomC, &zero, &zero)) // both
	assertUnread(t, table, userID, roomA, 0, 1)
	assertUnread(t, table, userID, roomB, 2, 2)
	assertUnread(t, table, userID, roomC, 0, 0)

	wantHighlights := map[string]int{
		roomB: 2,
	}
	wantNotifs := map[string]int{
		roomA: 1,
		roomB: 2,
	}
	assertNoError(t, table.SelectAllNonZeroCountsForUser(userID, func(gotRoomID string, gotHighlight int, gotNotif int) {
		wantHighlight := wantHighlights[gotRoomID]
		if wantHighlight != gotHighlight {
			t.Errorf("SelectAllNonZeroCountsForUser for %v got %d highlights, want %d", gotRoomID, gotHighlight, wantHighlight)
		}
		wantNotif := wantNotifs[gotRoomID]
		if wantNotif != gotNotif {
			t.Errorf("SelectAllNonZeroCountsForUser for %v got %d notifs, want %d", gotRoomID, gotNotif, wantNotif)
		}
		delete(wantHighlights, gotRoomID)
		delete(wantNotifs, gotRoomID)
	}))
	if len(wantHighlights) != 0 {
		t.Errorf("SelectAllNonZeroCountsForUser missed highlight rooms: %+v", wantHighlights)
	}
	if len(wantNotifs) != 0 {
		t.Errorf("SelectAllNonZeroCountsForUser missed notif rooms: %+v", wantNotifs)
	}
}

func assertUnread(t *testing.T, table *UnreadTable, userID, roomID string, wantHighight, wantNotif int) {
	t.Helper()
	gotHighlight, gotNotif, err := table.SelectUnreadCounters(userID, roomID)
	if err != nil {
		t.Fatalf("SelectUnreadCounters %s %s: %s", userID, roomID, err)
	}
	if gotHighlight != wantHighight {
		t.Errorf("SelectUnreadCounters: got %d highlights, want %d", gotHighlight, wantHighight)
	}
	if gotNotif != wantNotif {
		t.Errorf("SelectUnreadCounters: got %d notifs, want %d", gotNotif, wantNotif)
	}
}

func assertNoError(t *testing.T, err error) {
	t.Helper()
	if err != nil {
		t.Fatalf("got error: %s", err)
	}
}