package engine

import (
	"testing"

	"git.cloudinit.dev/coreci/orca/internal/store"
)

func TestPickNodeBestFit(t *testing.T) {
	caps := []*store.NodeCapacity{
		{NodeID: "node-b", CPUMillicores: 1000, MemoryMiB: 1024, DiskMiB: 1024},
		{NodeID: "node-a", CPUMillicores: 4000, MemoryMiB: 4096, DiskMiB: 4096},
		{NodeID: "node-c", CPUMillicores: 500, MemoryMiB: 512, DiskMiB: 512},
	}
	spec := JobSpec{CPUMillicores: 1000, MemoryMiB: 1024, DiskMiB: 1024}
	got, idx, err := PickNode(spec, caps)
	if err != nil {
		t.Fatalf("PickNode: %v", err)
	}
	if got.NodeID != "node-a" {
		t.Errorf("PickNode: got %s, want node-a (most free capacity)", got.NodeID)
	}
	if idx != 1 {
		t.Errorf("PickNode: got idx %d, want 1", idx)
	}
}

func TestPickNodeNoFit(t *testing.T) {
	caps := []*store.NodeCapacity{
		{NodeID: "node-a", CPUMillicores: 100, MemoryMiB: 100, DiskMiB: 100},
	}
	spec := JobSpec{CPUMillicores: 1000, MemoryMiB: 1024, DiskMiB: 1024}
	_, _, err := PickNode(spec, caps)
	if err == nil {
		t.Fatal("expected PickNode to fail when no node can fit")
	}
}

func TestPickNodeTieDeterministic(t *testing.T) {
	// Two nodes with identical free capacity. Tie broken by NodeID
	// (lexicographic) for determinism.
	caps := []*store.NodeCapacity{
		{NodeID: "node-z", CPUMillicores: 4000, MemoryMiB: 4096, DiskMiB: 4096},
		{NodeID: "node-a", CPUMillicores: 4000, MemoryMiB: 4096, DiskMiB: 4096},
	}
	spec := JobSpec{CPUMillicores: 1000, MemoryMiB: 1024, DiskMiB: 1024}
	got, _, err := PickNode(spec, caps)
	if err != nil {
		t.Fatalf("PickNode: %v", err)
	}
	if got.NodeID != "node-a" {
		t.Errorf("PickNode tie-break: got %s, want node-a (lexicographic)", got.NodeID)
	}
}

func TestJobSpecFits(t *testing.T) {
	spec := JobSpec{CPUMillicores: 1000, MemoryMiB: 1024, DiskMiB: 1024}
	c := &store.NodeCapacity{CPUMillicores: 2000, MemoryMiB: 2048, DiskMiB: 2048}
	if !spec.Fits(c) {
		t.Error("Fits: should fit")
	}
	c.CPUMillicores = 500
	if spec.Fits(c) {
		t.Error("Fits: should not fit (CPU too low)")
	}
}