package store

import "testing"

func TestAnEngineBuildIsChosenPerNodeAndCanBeLeftToItAgain(t *testing.T) {
	db := capacityStore(t)
	ctx := t.Context()
	if err := db.SetNodeEngine(ctx, "gpu07", "vllm", "cu129", "ilker"); err != nil {
		t.Fatal(err)
	}
	if err := db.SetNodeEngine(ctx, "gpu07", "vllm", "none", "ilker"); err != nil {
		t.Fatal(err)
	}
	got, err := db.NodeEngines(ctx)
	if err != nil || got["gpu07"]["vllm"] != "none" {
		t.Fatalf("got %v, %v; the second choice replaces the first", got, err)
	}
	if err := db.SetNodeEngine(ctx, "gpu07", "vllm", "", "ilker"); err != nil {
		t.Fatal(err)
	}
	if got, _ := db.NodeEngines(ctx); len(got["gpu07"]) != 0 {
		t.Errorf("left to the node again, still %v", got)
	}
}