Function TestForkSession

pkg/session/branch_test.go:374–458

(t *testing.T)

}

func TestForkSession(t *testing.T) {
	t.Parallel()
	t.Run("nil parent returns error", func(t *testing.T) {
		t.Parallel()
		_, err := ForkSession(nil, 0)
		require.Error(t, err)
		assert.Contains(t, err.Error(), "parent session is nil")
	})

	t.Run("out-of-range position returns error", func(t *testing.T) {
		t.Parallel()
		parent := &Session{Messages: []Item{NewMessageItem(UserMessage("test"))}}
		_, err := ForkSession(parent, 2)
		require.Error(t, err)
		assert.Contains(t, err.Error(), "out of range")
	})

	t.Run("copies messages up to position and uses fork-numbered title", func(t *testing.T) {
		t.Parallel()
		parent := &Session{
			ID:    "parent-id",
			Title: "Parent Title",
			Messages: []Item{
				NewMessageItem(UserMessage("msg1")),
				NewMessageItem(UserMessage("msg2")),
				NewMessageItem(UserMessage("msg3")),
			},
		}

		forked, err := ForkSession(parent, 2)
		require.NoError(t, err)
		assert.NotEqual(t, parent.ID, forked.ID)
		assert.Equal(t, "Parent Title (fork 1)", forked.Title)
		assert.Len(t, forked.Messages, 2)
		assert.Equal(t, "msg1", forked.Messages[0].Message.Message.Content)
		assert.Equal(t, "msg2", forked.Messages[1].Message.Message.Content)
	})

	t.Run("fork of a fork increments the counter", func(t *testing.T) {
		t.Parallel()
		parent := &Session{
			Title:    "Parent Title (fork 2)",
			Messages: []Item{NewMessageItem(UserMessage("hi"))},
		}

		forked, err := ForkSession(parent, 1)
		require.NoError(t, err)
		assert.Equal(t, "Parent Title (fork 3)", forked.Title)
	})

	t.Run("position zero produces an empty fork", func(t *testing.T) {
		t.Parallel()
		parent := &Session{
			Title:    "Parent Title",
			Messages: []Item{NewMessageItem(UserMessage("only"))},
		}

		forked, err := ForkSession(parent, 0)