mcp_session_test.go

//   /    ctx:                         https://ctx.ist
// ,'`./    do you remember?
// `.,'\
//   \    Copyright 2026-present Context contributors.
//                 SPDX-License-Identifier: Apache-2.0

package entity

import (
	"testing"
	"time"
)

func TestNewMCPSession(t *testing.T) {
	s := NewMCPSession()
	if s.ToolCalls != 0 {
		t.Errorf("ToolCalls = %d, want 0", s.ToolCalls)
	}
	if s.AddsPerformed == nil {
		t.Fatal("AddsPerformed should be initialized")
	}
	if len(s.AddsPerformed) != 0 {
		t.Errorf(
			"AddsPerformed length = %d, want 0",
			len(s.AddsPerformed),
		)
	}
	if s.SessionStartedAt.IsZero() {
		t.Error("SessionStartedAt should be set")
	}
	if len(s.PendingFlush) != 0 {
		t.Errorf(
			"PendingFlush length = %d, want 0",
			len(s.PendingFlush),
		)
	}
}

func TestRecordToolCall(t *testing.T) {
	s := NewMCPSession()
	s.RecordToolCall()
	if s.ToolCalls != 1 {
		t.Errorf("ToolCalls = %d, want 1", s.ToolCalls)
	}
	s.RecordToolCall()
	s.RecordToolCall()
	if s.ToolCalls != 3 {
		t.Errorf("ToolCalls = %d, want 3", s.ToolCalls)
	}
}

func TestRecordAdd(t *testing.T) {
	s := NewMCPSession()
	s.RecordAdd("task")
	s.RecordAdd("task")
	s.RecordAdd("decision")
	if s.AddsPerformed["task"] != 2 {
		t.Errorf(
			"task adds = %d, want 2",
			s.AddsPerformed["task"],
		)
	}
	if s.AddsPerformed["decision"] != 1 {
		t.Errorf(
			"decision adds = %d, want 1",
			s.AddsPerformed["decision"],
		)
	}
}

func TestQueuePendingUpdate(t *testing.T) {
	s := NewMCPSession()
	now := time.Now()
	s.QueuePendingUpdate(PendingUpdate{
		Type:     "task",
		Content:  "Build feature",
		QueuedAt: now,
	})
	if len(s.PendingFlush) != 1 {
		t.Fatalf(
			"PendingFlush length = %d, want 1",
			len(s.PendingFlush),
		)
	}
	pu := s.PendingFlush[0]
	if pu.Type != "task" {
		t.Errorf(
			"Type = %q, want %q",
			pu.Type, "task",
		)
	}
	if pu.Content != "Build feature" {
		t.Errorf(
			"Content = %q, want %q",
			pu.Content, "Build feature",
		)
	}
}