package store

import (
	"context"
	"path/filepath"
	"testing"
	"time"
)

// checkUsageNamesWhoAndWhere is run against both dialects: a report that
// named people by id and projects by id was a table nobody could read.
func checkUsageNamesWhoAndWhere(t *testing.T, s *DB) {
	t.Helper()
	ctx := context.Background()
	must := func(err error) {
		t.Helper()
		if err != nil {
			t.Fatal(err)
		}
	}
	must(s.PutOrg(ctx, Org{ID: "o", Name: "Acme"}))
	must(s.PutTeam(ctx, Team{ID: "t", OrgID: "o", Name: "ml"}))
	must(s.PutProject(ctx, Project{ID: "p", TeamID: "t", Name: "training"}))
	must(s.CreateUser(ctx, User{ID: "usr-1", Email: "ada@example.com", CreatedAt: time.Now()}))
	now := time.Now()
	must(s.WriteUsage(ctx, []Usage{
		{At: now, KeyID: "session:usr-1", UserID: "usr-1", ProjectID: "p", Model: "m", ServedModel: "m", Status: 200},
		{At: now, KeyID: "session:usr-1", UserID: "usr-1", Model: "m", ServedModel: "m", Status: 200},
	}))

	rows, err := s.UsageSince(ctx, now.Add(-time.Hour))
	if err != nil {
		t.Fatal(err)
	}
	if len(rows) != 2 {
		t.Fatalf("expected two rows, got %+v", rows)
	}
	places := map[string]bool{}
	for _, r := range rows {
		if r.Email != "ada@example.com" || !r.Chat() {
			t.Errorf("row %+v should name ada and read as the chat page", r)
		}
		places[r.ProjectPath] = true
	}
	if !places["Acme / ml / training"] || !places[""] {
		t.Errorf("places = %v, expected the project's path and personal", places)
	}
}

func TestUsageNamesWhoAndWhere(t *testing.T) {
	s, err := OpenSQLite(context.Background(), filepath.Join(t.TempDir(), "u.db"))
	if err != nil {
		t.Fatal(err)
	}
	t.Cleanup(func() { _ = s.Close() })
	checkUsageNamesWhoAndWhere(t, s)
}

func TestPostgresUsageNamesWhoAndWhere(t *testing.T) {
	checkUsageNamesWhoAndWhere(t, openPG(t))
}