Netpulse_SasS/agent/internal/ncmx/apply_test.go

package ncmx

import (
	"context"
	"strings"
	"testing"

	npv1 "github.com/netpulse/netpulse/gen/go/netpulse/v1"
)

// Перевірки заливки конфігу.
//
// Головне, що тут перевіряється, — поводження з ВІДМОВОЮ пристрою.
// Заливка, яка не помітила відмови й пішла далі, лишає по собі
// напівзастосований конфіг, який виглядає застосованим: жодна помилка
// нікуди не записалась, звіт зелений, а вузол не той. Це найдорожчий
// різновид збою в усій системі, і ловиться він рівно тут.

func applyJob(commands []string, commit string) *npv1.ConfigApplyJob {
	return &npv1.ConfigApplyJob{
		RollbackId:    "rb-1",
		Commands:      commands,
		CommitCommand: commit,
	}
}

func TestApplyRunsAllLinesAndCommits(t *testing.T) {
	d := newDevice("\nsw1(config)#", "", map[string]string{
		"configure terminal": "",
		"vlan 10":            "",
		"end":                "",
		"write memory":       "Building configuration...\r\nOK\r\n",
	})
	cli := cliFor(d, `[>#]\s*$`)
	ctx := context.Background()
	_ = cli.WaitPrompt(ctx)

	outcomes, committed, err := applyLoop(ctx, cli,
		applyJob([]string{"configure terminal", "vlan 10", "end"}, "write memory"))

	if err != nil {
		t.Fatalf("несподівана помилка: %v", err)
	}
	if !committed {
		t.Error("команда збереження відпрацювала, а committed лишився false")
	}
	if len(outcomes) != 4 {
		t.Fatalf("очікували 4 результати (3 рядки + збереження), отримали %d", len(outcomes))
	}
	for _, oc := range outcomes {
		if !oc.GetSuccess() {
			t.Errorf("рядок %q позначено невдалим: %s", oc.GetCommand(), oc.GetErrorLine())
		}
	}
}

// Відмову пристрою видно лише у виводі: коду помилки в CLI немає.
// Пропустити її означає долити решту конфігу поверх того, що не
// застосувалось.
func TestApplyStopsOnDeviceRejection(t *testing.T) {
	d := newDevice("\nsw1(config)#", "", map[string]string{
		"configure terminal": "",
		"vlan 10":            "% Invalid input detected at '^' marker.\r\n",
		"name office":        "",
	})
	cli := cliFor(d, `[>#]\s*$`)
	ctx := context.Background()
	_ = cli.WaitPrompt(ctx)

	outcomes, committed, err := applyLoop(ctx, cli,
		applyJob([]string{"configure terminal", "vlan 10", "name office"}, "write memory"))

	if err == nil {
		t.Fatal("відмову пристрою не помічено")
	}
	if committed {
		t.Error("напівзалитий конфіг не має зберігатись")
	}
	if len(outcomes) != 2 {
		t.Fatalf("послідовність мала спинитись на другому рядку, отримали %d", len(outcomes))
	}
	if outcomes[1].GetSuccess() {
		t.Error("відхилений рядок позначено успішним")
	}
	if !strings.Contains(outcomes[1].GetErrorLine(), "Invalid input") {
		t.Errorf("у звіті немає слів самого пристрою: %q", outcomes[1].GetErrorLine())
	}
}

// Невдале збереження — це НЕ провал заливки: конфіг уже на пристрої й
// працює. Але й не успіх: перезавантаження його прибере. Стан має бути
// видно окремо, тому committed лишається false, а помилка — непорожня.
func TestApplyReportsFailedCommitSeparately(t *testing.T) {
	d := newDevice("\nsw1(config)#", "", map[string]string{
		"configure terminal": "",
		"vlan 10":            "",
		"write memory":       "%% Error: startup-config is locked\r\n",
	})
	cli := cliFor(d, `[>#]\s*$`)
	ctx := context.Background()
	_ = cli.WaitPrompt(ctx)

	outcomes, committed, err := applyLoop(ctx, cli,
		applyJob([]string{"configure terminal", "vlan 10"}, "write memory"))

	if err == nil {
		t.Fatal("невдале збереження мало дати помилку")
	}
	if committed {
		t.Error("committed має лишитись false")
	}
	// Рядки конфігу при цьому успішні: вони справді виконались.
	if !outcomes[0].GetSuccess() || !outcomes[1].GetSuccess() {
		t.Error("рядки конфігу помилково позначено невдалими")
	}
	if outcomes[len(outcomes)-1].GetSuccess() {
		t.Error("невдале збереження позначено успішним")
	}
}

// Порожній вивід — нормальна відповідь на конфігураційний рядок:
// більшість платформ на успішний `vlan 10` не пише нічого.
func TestApplyAcceptsSilentSuccess(t *testing.T) {
	d := newDevice("\nsw1(config)#", "", map[string]string{"vlan 10": ""})
	cli := cliFor(d, `[>#]\s*$`)
	ctx := context.Background()
	_ = cli.WaitPrompt(ctx)

	outcomes, _, err := applyLoop(ctx, cli, applyJob([]string{"vlan 10"}, ""))
	if err != nil {
		t.Fatalf("мовчазний успіх прийнято за помилку: %v", err)
	}
	if len(outcomes) != 1 || !outcomes[0].GetSuccess() {
		t.Fatalf("отримали %+v", outcomes)
	}
}

// Слово «invalid» усередині законного рядка не має зупиняти заливку:
// зразки відмов прив'язані до маркерів, а не шукаються будь-де в тексті.
func TestApplyDoesNotMistakeConfigTextForRejection(t *testing.T) {
	if line := rejectLine("ip access-list extended INVALID-SRC"); line != "" {
		t.Errorf("законний рядок прийнято за відмову: %q", line)
	}
	if line := rejectLine("% Invalid input detected"); line == "" {
		t.Error("справжню відмову не помічено")
	}
}