Function TestLinearNode_Next

go/engine/node_test.go:41–82

(t *testing.T)

		require.NoError(t, n.AddTransition(PortFalse, Transition{TargetID: "f-target"}))
	})
}

func TestLinearNode_Next(t *testing.T) {
	t.Run("returns StateIdle when no transition is wired", func(t *testing.T) {
		n := NewLinearNode("n1")
		s, err := NewMainScope(nil)
		require.NoError(t, err)
		next, err := n.Next(PortCtrl, s)
		require.NoError(t, err)
		assert.Equal(t, StateIdle, next)
	})

	t.Run("returns target id and applies transition", func(t *testing.T) {
		n := NewLinearNode("n1")
		require.NoError(t, n.AddTransition(PortCtrl, Transition{
			TargetID: "next",
			EdgeType: workflowapi.AgentChoice, // clears conversation as a side effect
		}))
		s, err := NewMainScope(nil)
		require.NoError(t, err)
		s.SetConversation(llmproxy.InputString("will be cleared"))

		next, err := n.Next(PortCtrl, s)
		require.NoError(t, err)
		assert.Equal(t, "next", next)
		assert.Empty(t, s.GetConversation())
	})

	t.Run("propagates Apply errors", func(t *testing.T) {
		n := NewLinearNode("n1")
		require.NoError(t, n.AddTransition(PortCtrl, Transition{
			TargetID: "next",
			EdgeType: workflowapi.AgentTask,
			Prompt:   nil, // AgentTask requires prompt → error
		}))
		s, err := NewMainScope(nil)
		require.NoError(t, err)

		_, err = n.Next(PortCtrl, s)
		require.Error(t, err)
		assert.Contains(t, err.Error(), "n1")
		assert.Contains(t, err.Error(), PortCtrl)
	})
}