package main

import "testing"

func TestNewManifestDistributesUsersAndEndpoints(t *testing.T) {
	cfg := &Config{Tenant: "006", UsersPerMachine: 3, Endpoints: map[string][]string{
		"a": {"ws://a1", "ws://a2"}, "b": {"ws://b1"},
	}}
	manifest := newManifest(cfg)
	if len(manifest.Accounts) != 6 {
		t.Fatalf("accounts=%d", len(manifest.Accounts))
	}
	if got := accountsForMachine(manifest, "a"); got[0].Username != "lta0001" || got[1].Endpoint != "ws://a2" || got[2].Endpoint != "ws://a1" {
		t.Fatalf("unexpected A accounts: %+v", got)
	}
	if err := manifest.validate(); err != nil {
		t.Fatal(err)
	}
}

func TestManifestRejectsDuplicateAccount(t *testing.T) {
	manifest := &Manifest{Version: manifestVersion, Accounts: []AccountRecord{
		{Machine: "a", Index: 1, Username: "lta0001"},
		{Machine: "a", Index: 1, Username: "duplicate"},
	}}
	if err := manifest.validate(); err == nil {
		t.Fatal("expected duplicate account validation error")
	}
}

func TestApplyConfigEndpointsRefreshesManifest(t *testing.T) {
	cfg := &Config{Tenant: "006", UsersPerMachine: 1, Endpoints: map[string][]string{
		"a": {"ws://new-a"}, "b": {"ws://new-b"},
	}}
	manifest := &Manifest{Tenant: "006", Accounts: []AccountRecord{
		{Machine: "a", Index: 1, Username: "lta0001", Endpoint: "ws://old"},
		{Machine: "b", Index: 1, Username: "ltb0001", Endpoint: "ws://old"},
	}}
	if err := applyConfigEndpoints(cfg, manifest); err != nil {
		t.Fatal(err)
	}
	if manifest.Accounts[0].Endpoint != "ws://new-a" || manifest.Accounts[1].Endpoint != "ws://new-b" {
		t.Fatalf("endpoints were not refreshed: %+v", manifest.Accounts)
	}
}