package hostactors

import (
	"context"
	"crypto/ed25519"
	"crypto/rand"
	"github.com/carlosframework/platform/internal/release"
	"strings"
	"testing"
)

func fixture(t *testing.T) (Directory, release.Signer, Observed) {
	t.Helper()
	pub, key, err := ed25519.GenerateKey(rand.Reader)
	if err != nil {
		t.Fatal(err)
	}
	s := release.NewDirStore(t.TempDir())
	sg := release.RootSigner(key)
	r := Roster{V: 1, DeploymentBucket: "bucket", Revision: 1, Boxes: map[string]Box{"box": {Enrollment: strings.Repeat("a", 32), Namespace: "platform", BoxID: "box", Label: "label", Region: "ie", Pool: "default", Cloud: Cloud{Provider: "aws", Account: "123", Region: "eu-west-1", InstanceID: "i-box"}, EvidenceSHA256: strings.Repeat("b", 64), State: "active", LossIDs: []string{}}}, Consoles: map[string]Console{}, Holds: map[string]Reservation{}, Activations: map[string]Reservation{}}
	raw, err := release.SignRootEnvelope(sg, r)
	if err != nil {
		t.Fatal(err)
	}
	etag, err := s.Put(context.Background(), Key, raw, "")
	if err != nil {
		t.Fatal(err)
	}
	return Directory{s, pub, "bucket"}, sg, Observed{r, etag}
}
func TestDirectoryFreshQuarantineAndCAS(t *testing.T) {
	ctx := context.Background()
	d, sg, o := fixture(t)
	first, err := d.Lookup(ctx, "box")
	if err != nil || first.State != "active" {
		t.Fatal(first, err)
	}
	changed, err := d.Quarantine(ctx, sg, o, "box", first.Enrollment, "loss-1")
	if err != nil {
		t.Fatal(err)
	}
	got, err := d.Lookup(ctx, "box")
	if err != nil || got.State != "quarantined" || got.QuarantineRevision != 1 {
		t.Fatal(got, err)
	}
	if _, err = d.Quarantine(ctx, sg, o, "box", first.Enrollment, "loss-2"); err == nil {
		t.Fatal("stale CAS accepted")
	}
	next := Clone(changed.Roster)
	b := next.Boxes["box"]
	b.Cloud.InstanceID = "i-other"
	next.Boxes["box"] = b
	if _, err = d.Replace(ctx, sg, changed, next); err == nil {
		t.Fatal("cloud binding changed")
	}
}
func TestReservationsExcludeBothOrders(t *testing.T) {
	ctx := context.Background()
	for _, activationFirst := range []bool{true, false} {
		d, sg, o := fixture(t)
		en := o.Roster.Boxes["box"].Enrollment
		h := Reservation{ID: "hold", Host: "app.example", OpID: "op", ExpectedClaimETag: "etag", FloorSHA256: strings.Repeat("f", 64), Actors: map[string]string{"box:box": en}, Releases: map[string]Report{"box:box": {Enrollment: en}}}
		a := h
		a.ID = "activation"
		a.Hashes = map[string][]string{"box:box": {strings.Repeat("a", 64), strings.Repeat("b", 64), strings.Repeat("c", 64)}}
		first, second := h, a
		if activationFirst {
			first, second = a, h
		}
		next, err := d.Reserve(ctx, sg, o, first, activationFirst)
		if err != nil {
			t.Fatal(err)
		}
		if _, err = d.Reserve(ctx, sg, next, second, !activationFirst); err == nil {
			t.Fatal("overlap accepted")
		}
	}
}