chore: import upstream snapshot with attribution

flow/registry_test.go

@@ -0,0 +1,65 @@
+package flow
+
+import (
+	"context"
+	"testing"
+
+	"go-micro.dev/v6/broker"
+	"go-micro.dev/v6/client"
+	"go-micro.dev/v6/registry"
+)
+
+// A trigger-bound flow announces itself in the registry as type=flow
+// while running, and deregisters on Stop — liveness, like a service.
+func TestFlowRegistersAndDeregisters(t *testing.T) {
+	reg := registry.NewMemoryRegistry()
+	br := broker.NewMemoryBroker()
+	if err := br.Connect(); err != nil {
+		t.Fatalf("broker connect: %v", err)
+	}
+
+	f := New("onboard",
+		Trigger("events.user.created"),
+		Steps(appendStep("a")),
+	)
+	if err := f.Register(reg, br, client.DefaultClient); err != nil {
+		t.Fatalf("Register: %v", err)
+	}
+
+	svcs, err := reg.GetService("onboard")
+	if err != nil || len(svcs) == 0 {
+		t.Fatalf("flow not registered: %v", err)
+	}
+	if svcs[0].Metadata["type"] != "flow" {
+		t.Errorf("registry metadata type = %q, want flow", svcs[0].Metadata["type"])
+	}
+	if svcs[0].Metadata["trigger"] != "events.user.created" {
+		t.Errorf("registry metadata trigger = %q", svcs[0].Metadata["trigger"])
+	}
+
+	if err := f.Stop(); err != nil {
+		t.Fatalf("Stop: %v", err)
+	}
+	if svcs, _ := reg.GetService("onboard"); len(svcs) != 0 {
+		t.Errorf("flow should be deregistered after Stop, got %d services", len(svcs))
+	}
+}
+
+// A flow without a trigger is not a running listener and isn't registered.
+func TestFlowWithoutTriggerNotRegistered(t *testing.T) {
+	reg := registry.NewMemoryRegistry()
+	br := broker.NewMemoryBroker()
+	_ = br.Connect()
+
+	f := New("oneshot", Steps(appendStep("a")))
+	if err := f.Register(reg, br, client.DefaultClient); err != nil {
+		t.Fatalf("Register: %v", err)
+	}
+	if svcs, _ := reg.GetService("oneshot"); len(svcs) != 0 {
+		t.Errorf("triggerless flow should not register, got %d", len(svcs))
+	}
+	// It still runs on demand.
+	if err := f.Execute(context.Background(), ""); err != nil {
+		t.Fatalf("Execute: %v", err)
+	}
+}