package tui

import (
	"testing"
)

func TestTUIInitialization(t *testing.T) {
	ui := New()
	if ui == nil {
		t.Fatal("Failed to initialize TUI")
	}

	if ui.CurrentScreen != ScreenHostList {
		t.Errorf("expected CurrentScreen ScreenHostList, got %d", ui.CurrentScreen)
	}

	if ui.Quit {
		t.Error("expected Quit to be false")
	}
}

func TestTUILoadHosts(t *testing.T) {
	ui := New()
	if ui == nil {
		t.Fatal("Failed to initialize TUI")
	}

	ui.LoadHosts(nil)
	if ui.Hosts != nil {
		t.Error("expected Hosts to be nil")
	}
}