package deploy

import (
	"context"
	"testing"
)

func TestParseOrganizationalUnit(t *testing.T) {
	if got := parseOrganizationalUnit("organizational_unit=ou/Batch\n"); got != "ou/Batch" {
		t.Fatalf("got=%q", got)
	}
	if got := parseOrganizationalUnit("AETHERFORGE_OU=ou/GPU\n"); got != "ou/GPU" {
		t.Fatalf("got=%q", got)
	}
}

func TestReadCloudVenueProbeMockIMDS(t *testing.T) {
	prev := ec2IMDSReadVenue
	t.Cleanup(func() { ec2IMDSReadVenue = prev })
	ec2IMDSReadVenue = func(ctx context.Context) (*CloudVenueProbe, error) {
		return &CloudVenueProbe{
			CloudProvider:      "aws",
			Environment:        "prod",
			Workload:           "batch",
			InstanceType:       "m5.large",
			InstanceLifecycle:  "spot",
			AvailabilityZone:   "us-east-1a",
			OrganizationalUnit: "ou/Batch",
			EC2Tags:            map[string]string{"Environment": "prod", "Workload": "batch"},
		}, nil
	}
	probe := ReadCloudVenueProbe()
	if probe == nil || probe.InstanceType != "m5.large" || probe.InstanceLifecycle != "spot" {
		t.Fatalf("probe=%+v", probe)
	}
	if probe.Environment != "prod" || probe.Workload != "batch" {
		t.Fatalf("env=%q workload=%q", probe.Environment, probe.Workload)
	}
}

func TestReadCloudVenueProbeNonAWSReturnsNil(t *testing.T) {
	prev := ec2IMDSReadVenue
	t.Cleanup(func() { ec2IMDSReadVenue = prev })
	ec2IMDSReadVenue = func(ctx context.Context) (*CloudVenueProbe, error) {
		return nil, context.DeadlineExceeded
	}
	if ReadCloudVenueProbe() != nil {
		t.Fatal("expected nil probe off AWS")
	}
}