package transport_test

import (
	"testing"

	"iop/apps/node/internal/transport"
)

func TestSession_RegisterCancel_CancelRun(t *testing.T) {
	sess := &transport.Session{}
	called := false

	sess.RegisterCancel("run-1", func() { called = true })
	sess.CancelRun("run-1")

	if !called {
		t.Fatal("cancel function was not called")
	}
}

func TestSession_DeregisterCancel(t *testing.T) {
	sess := &transport.Session{}
	called := false

	sess.RegisterCancel("run-1", func() { called = true })
	sess.DeregisterCancel("run-1")
	sess.CancelRun("run-1")

	if called {
		t.Fatal("cancel function should not have been called after deregister")
	}
}

func TestSession_CancelRun_UnknownID(t *testing.T) {
	sess := &transport.Session{}

	sess.CancelRun("missing-run")
}