Netpulse_SasS/server/internal/store/devices_restore_test.go

package store

import "testing"

// Повернення хоста з архіву впирається у дві часткові унікальні умови,
// які стосуються ЛИШЕ живих рядків: ім'я (devices_tenant_name_uniq) і
// прив'язка «хост = машина зонда» (devices_self_agent_uniq). Поки хост
// лежав прибраним, обидві його не бачили, і зіткнення виникає рівно в
// мить повернення.
//
// Перевіряти це на живій базі означало б стенд із двома хостами, зондом
// і майданчиком на кожен випадок; ціна помилки — помилка унікального
// індексу посеред UPDATE, тобто відмова ВСЬОМУ набору замість одного
// хоста. Тому рішення про набір винесене чистою функцією, а тут — усі
// випадки, які вона мусить розрізняти.

func TestPlanRestoreKeepsNamesUnique(t *testing.T) {
	// Живий хост уже носить це ім'я — повертати нікуди.
	p := planRestore([]restoreCandidate{
		{id: "1", name: "core-sw", nameTaken: true},
		{id: "2", name: "edge-1"},
	})
	if len(p.back) != 1 || p.back[0] != "2" {
		t.Fatalf("повертати треба лише edge-1, а повертається %v", p.back)
	}
	if len(p.failed) != 1 || p.failed[0].DeviceID != "1" {
		t.Fatalf("відмова мала дістатись core-sw, а дісталась %+v", p.failed)
	}
	if p.failed[0].Error == "" {
		t.Fatal("відмова без причини: людина не дізнається, що робити далі")
	}
}

func TestPlanRestoreCatchesDuplicateInsideBatch(t *testing.T) {
	// Два ПРИБРАНІ хости з однаковим іменем: у базі вони не зіткнулись,
	// бо часткова унікальна умова не бачить прибраних. Зіткнуться, щойно
	// повернуться обидва — і зловити це має набір, а не індекс.
	//
	// Регістр різний навмисно: індекс іде по lower(name).
	p := planRestore([]restoreCandidate{
		{id: "1", name: "Core-SW"},
		{id: "2", name: "core-sw"},
	})
	if len(p.back) != 1 || p.back[0] != "1" {
		t.Fatalf("перший претендент мав лишити ім'я собі, повертається %v", p.back)
	}
	if len(p.failed) != 1 || p.failed[0].DeviceID != "2" {
		t.Fatalf("другому мала бути відмова, а маємо %+v", p.failed)
	}
}

func TestPlanRestoreDropsTakenSelfAgent(t *testing.T) {
	// Прив'язку до машини зонда за час в архіві забрав живий хост.
	// Хост усе одно повертається — інакше він лишився б в архіві через
	// налаштування, яке й так треба переглянути, — але вже без неї.
	p := planRestore([]restoreCandidate{
		{id: "1", name: "srv-1", selfAgent: "agent-a", selfTaken: true},
	})
	if len(p.back) != 1 {
		t.Fatalf("хост мав повернутись, повертається %v", p.back)
	}
	if len(p.unbind) != 1 || p.unbind[0] != "1" {
		t.Fatalf("прив'язку мали зняти, знімається з %v", p.unbind)
	}
	if !p.restored[0].SelfAgentDropped {
		t.Fatal("зняту прив'язку не позначено — людина про неї не дізнається")
	}
	if len(p.failed) != 0 {
		t.Fatalf("відмовляти тут нема за що: %+v", p.failed)
	}
}

func TestPlanRestoreSplitsSelfAgentInsideBatch(t *testing.T) {
	// Двоє прибраних претендують на ту саму машину зонда. Один може
	// лишити прив'язку собі, другий — ні; вирішує порядок набору.
	p := planRestore([]restoreCandidate{
		{id: "1", name: "srv-1", selfAgent: "agent-a"},
		{id: "2", name: "srv-2", selfAgent: "agent-a"},
	})
	if len(p.back) != 2 {
		t.Fatalf("обидва мали повернутись, повертається %v", p.back)
	}
	if len(p.unbind) != 1 || p.unbind[0] != "2" {
		t.Fatalf("прив'язку мали зняти з другого, знімається з %v", p.unbind)
	}
	if p.restored[0].SelfAgentDropped || !p.restored[1].SelfAgentDropped {
		t.Fatalf("позначки зняття стоять не там: %+v", p.restored)
	}
}

func TestPlanRestoreKeepsFreeSelfAgent(t *testing.T) {
	// Прив'язка вільна — знімати її нема причини. Окремий випадок, бо
	// саме він найчастіший, і зайве зняття тут означало б, що збір
	// конфігів сервера мовчки не починається після повернення.
	p := planRestore([]restoreCandidate{
		{id: "1", name: "srv-1", selfAgent: "agent-a", enabled: true},
		// Хост без прив'язки взагалі: порожній ключ не має вважатись
		// зайнятим першим же таким хостом.
		{id: "2", name: "srv-2"},
		{id: "3", name: "srv-3"},
	})
	if len(p.unbind) != 0 {
		t.Fatalf("нічого знімати не треба, а знімається %v", p.unbind)
	}
	if len(p.back) != 3 || len(p.failed) != 0 {
		t.Fatalf("мали повернутись усі три: %v, %+v", p.back, p.failed)
	}
	// enabled їде з archived_enabled і має дійти до відповіді як є:
	// «повернувся вимкненим» — це те, чого людина не побачить ніде,
	// крім цього поля.
	if !p.restored[0].Enabled || p.restored[1].Enabled {
		t.Fatalf("стан enabled перекручено: %+v", p.restored)
	}
}

func TestPlanRestoreEmptyBatch(t *testing.T) {
	p := planRestore(nil)
	if len(p.back) != 0 || len(p.failed) != 0 {
		t.Fatalf("з порожнього набору не береться нічого: %+v", p)
	}
	// Порожній зріз, а не nil: він їде в JSON відповіді, і `null` там
	// означав би для клієнта «поля немає», а не «нікого не повернуто».
	if p.restored == nil {
		t.Fatal("restored має бути порожнім зрізом, а не nil")
	}
}