package dbi

import (
	"testing"
)

func genActivity(username string) Activity {
	return Activity{
		Title:       randString(20),
		Description: randString(60),
		Hyperlink:   "https://" + randString(12) + ".example.com",
		Status:      "reported",
		Username:    username,
	}
}

func TestCreateActivity(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("CreateActivity: %v", err)
	}
	if a.UID == "" {
		t.Error("expected UID to be set after create")
	}
	if a.Created == "" {
		t.Error("expected Created to be set after create")
	}
	if a.Modified == "" {
		t.Error("expected Modified to be set after create")
	}

	got, err := GetActivity(conn, a.UID)
	if err != nil {
		t.Fatalf("GetActivity after create: %v", err)
	}
	if got != a {
		t.Errorf("got %+v, want %+v", got, a)
	}
}

func TestCreateActivityPresetUID(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))
	a.UID = "preset" + randString(10)
	presetUID := a.UID

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("CreateActivity: %v", err)
	}
	if a.UID != presetUID {
		t.Error("CreateActivity should not overwrite a non-empty UID")
	}
}

func TestCreateActivityDuplicateUID(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("first CreateActivity: %v", err)
	}
	a2 := a
	if err := CreateActivity(conn, &a2); err == nil {
		t.Error("expected error on duplicate UID, got nil")
	}
}

func TestGetActivity(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("CreateActivity: %v", err)
	}

	got, err := GetActivity(conn, a.UID)
	if err != nil {
		t.Fatalf("GetActivity: %v", err)
	}
	if got != a {
		t.Errorf("got %+v, want %+v", got, a)
	}
}

func TestGetActivityNotFound(t *testing.T) {
	conn := testConn(t, testDB(t))

	_, err := GetActivity(conn, randString(32))
	if err == nil {
		t.Error("expected error for missing activity, got nil")
	}
}

func TestUpdateActivity(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("CreateActivity: %v", err)
	}

	originalUID := a.UID
	originalCreated := a.Created
	a.Title = randString(20)
	a.Description = randString(60)
	a.Hyperlink = "https://" + randString(12) + ".example.com"
	a.Status = "posted"

	if err := UpdateActivity(conn, &a); err != nil {
		t.Fatalf("UpdateActivity: %v", err)
	}

	got, err := GetActivity(conn, a.UID)
	if err != nil {
		t.Fatalf("GetActivity after update: %v", err)
	}
	if got.UID != originalUID {
		t.Errorf("UID changed: got %q, want %q", got.UID, originalUID)
	}
	if got.Created != originalCreated {
		t.Errorf("Created changed: got %q, want %q", got.Created, originalCreated)
	}
	if got.Title != a.Title {
		t.Errorf("Title not updated: got %q, want %q", got.Title, a.Title)
	}
	if got.Description != a.Description {
		t.Errorf("Description not updated: got %q, want %q", got.Description, a.Description)
	}
	if got.Status != "posted" {
		t.Errorf("Status not updated: got %q", got.Status)
	}
	if got.Modified == "" {
		t.Error("Modified should be set after update")
	}
}

func TestDeleteActivity(t *testing.T) {
	conn := testConn(t, testDB(t))
	a := genActivity(randString(10))

	if err := CreateActivity(conn, &a); err != nil {
		t.Fatalf("CreateActivity: %v", err)
	}

	if err := DeleteActivity(conn, a.UID); err != nil {
		t.Fatalf("DeleteActivity: %v", err)
	}

	_, err := GetActivity(conn, a.UID)
	if err == nil {
		t.Error("expected error after delete, got nil")
	}
}

func TestGetActivitiesForUsername(t *testing.T) {
	conn := testConn(t, testDB(t))
	username := randString(10)

	items := make([]Activity, 3)
	for i := range items {
		items[i] = genActivity(username)
		if err := CreateActivity(conn, &items[i]); err != nil {
			t.Fatalf("CreateActivity %d: %v", i, err)
		}
	}
	other := genActivity(randString(10))
	if err := CreateActivity(conn, &other); err != nil {
		t.Fatalf("CreateActivity other: %v", err)
	}

	got, err := GetActivitiesForUsername(conn, username)
	if err != nil {
		t.Fatalf("GetActivitiesForUsername: %v", err)
	}
	if len(got) != 3 {
		t.Errorf("got %d activities, want 3", len(got))
	}
	byUID := make(map[string]Activity)
	for _, a := range got {
		byUID[a.UID] = a
	}
	for _, want := range items {
		a, ok := byUID[want.UID]
		if !ok {
			t.Errorf("activity %q missing from results", want.UID)
			continue
		}
		if a != want {
			t.Errorf("activity %q: got %+v, want %+v", want.UID, a, want)
		}
	}
}

func TestGetActivitiesForUsernameEmpty(t *testing.T) {
	conn := testConn(t, testDB(t))

	got, err := GetActivitiesForUsername(conn, randString(10))
	if err != nil {
		t.Fatalf("GetActivitiesForUsername: %v", err)
	}
	if len(got) != 0 {
		t.Errorf("expected empty slice, got %d items", len(got))
	}
}