package main

import (
	"bytes"
	"strings"
	"testing"
)

func TestSeveralAccountsErr(t *testing.T) {
	err := severalAccountsErr([]apiAccountEntry{
		{Name: "tom@keymail.dev", Sqid: "bdk"}, {Name: "acme", Sqid: "bce"}})
	want := "you belong to several accounts — say which with --account: tom@keymail.dev (bdk), acme (bce)"
	if err.Error() != want {
		t.Fatalf("severalAccountsErr = %q\nwant %q", err, want)
	}
}

func TestChooseAccount(t *testing.T) {
	accts := []apiAccountEntry{
		{Name: "tom@keymail.dev", Sqid: "bdk"}, {Name: "acme", Sqid: "bce"}}
	refusal := severalAccountsErr(accts).Error()

	// A clean pick of the second entry, declining the save.
	var out bytes.Buffer
	picked, save, err := chooseAccount(strings.NewReader("2\nn\n"), &out, "https://console.example", accts, ".carlos/config")
	if err != nil || picked.Sqid != "bce" || save {
		t.Fatalf("pick 2 = %+v save=%v err=%v", picked, save, err)
	}
	for _, want := range []string{
		"you belong to several accounts on https://console.example",
		"1. tom@keymail.dev (bdk)", "2. acme (bce)", "account [1-2]:",
		"save acme as this directory's default account (.carlos/config)? [y/N]",
	} {
		if !strings.Contains(out.String(), want) {
			t.Fatalf("chooser output missing %q:\n%s", want, out.String())
		}
	}

	// Saying y to the save prompt.
	out.Reset()
	picked, save, err = chooseAccount(strings.NewReader("1\ny\n"), &out, "https://console.example", accts, ".carlos/config")
	if err != nil || picked.Sqid != "bdk" || !save {
		t.Fatalf("pick 1 + save = %+v save=%v err=%v", picked, save, err)
	}

	// An empty saveDest — the caller decided saving is off the table (from
	// $HOME the "project" file would be the global config) — asks nothing
	// and never saves, even with a y queued up.
	out.Reset()
	picked, save, err = chooseAccount(strings.NewReader("1\ny\n"), &out, "https://console.example", accts, "")
	if err != nil || picked.Sqid != "bdk" || save {
		t.Fatalf("no-dest pick = %+v save=%v err=%v, want pick without a save", picked, save, err)
	}
	if strings.Contains(out.String(), "save ") {
		t.Fatalf("no-dest chooser must not offer the save:\n%s", out.String())
	}

	// An empty save answer means no — N is the default.
	out.Reset()
	_, save, err = chooseAccount(strings.NewReader("1\n\n"), &out, "https://console.example", accts, ".carlos/config")
	if err != nil || save {
		t.Fatalf("empty save answer = save=%v err=%v, want no-save", save, err)
	}

	// EOF at the save prompt also means no; the pick itself stands.
	out.Reset()
	picked, save, err = chooseAccount(strings.NewReader("1\n"), &out, "https://console.example", accts, ".carlos/config")
	if err != nil || picked.Sqid != "bdk" || save {
		t.Fatalf("EOF at save = %+v save=%v err=%v", picked, save, err)
	}

	// Junk once re-prompts; a valid second answer lands.
	out.Reset()
	picked, _, err = chooseAccount(strings.NewReader("zzz\n1\nn\n"), &out, "https://console.example", accts, ".carlos/config")
	if err != nil || picked.Sqid != "bdk" {
		t.Fatalf("junk-then-1 = %+v err=%v", picked, err)
	}
	if strings.Count(out.String(), "account [1-2]:") != 2 {
		t.Fatalf("want exactly one re-prompt:\n%s", out.String())
	}

	// Junk twice — including out-of-range numbers — is the refusal.
	for _, input := range []string{"zzz\nzzz\n", "0\n3\n", ""} {
		out.Reset()
		if _, _, err = chooseAccount(strings.NewReader(input), &out, "https://console.example", accts, ".carlos/config"); err == nil || err.Error() != refusal {
			t.Fatalf("input %q = %v, want the several-accounts refusal", input, err)
		}
	}
}