package atlas_test

import (
	"reflect"
	"testing"

	"git.d-ma.be/mathias/cad-atlas/internal/atlas"
)

func TestBuild_OverlaysCIStageNodesFromWorkflow(t *testing.T) {
	atlasJSON := []byte(`{
	  "substrate": [{"n":"koala","k":"gpu"}],
	  "stages": [
	    {"no":"STAGE 06","title":"PR → CI","generate":"ci-jobs","nodes":[]},
	    {"no":"STAGE 01","title":"TELOS","nodes":[{"t":"Intention substrate"}]}
	  ]}`)
	workflow := []byte("jobs:\n  check:\n    a: 1\n  build:\n    b: 2\n  deploy:\n    c: 3\n")

	a, err := atlas.Build(atlasJSON, workflow)
	if err != nil {
		t.Fatalf("Build: %v", err)
	}

	// CI stage nodes generated from the workflow's jobs, in order.
	var titles []string
	for _, n := range a.Stages[0].Nodes {
		titles = append(titles, n.Title)
	}
	if want := []string{"check", "build", "deploy"}; !reflect.DeepEqual(titles, want) {
		t.Fatalf("CI stage nodes = %v, want %v", titles, want)
	}

	// Non-generated stage is untouched.
	if a.Stages[1].Nodes[0].Title != "Intention substrate" {
		t.Fatalf("authored stage was altered: %+v", a.Stages[1])
	}

	// Substrate preserved.
	if len(a.Substrate) != 1 || a.Substrate[0].Name != "koala" {
		t.Fatalf("substrate not preserved: %+v", a.Substrate)
	}
}

func TestBuild_ErrorsOnBadAtlasJSON(t *testing.T) {
	if _, err := atlas.Build([]byte("{not json"), []byte("jobs:\n  x:\n    a: 1\n")); err == nil {
		t.Fatal("expected error on bad atlas JSON, got nil")
	}
}