Function TestExtractJSON

pkg/memory/session_summary_test.go:406–467

(t *testing.T)

}

func TestExtractJSON(t *testing.T) {
	tests := []struct {
		name     string
		input    string
		expected string
	}{
		{
			name:     "plain JSON",
			input:    `{"key": "value"}`,
			expected: `{"key": "value"}`,
		},
		{
			name:     "JSON in code block",
			input:    "```json\n{\"key\": \"value\"}\n```",
			expected: `{"key": "value"}`,
		},
		{
			name:     "JSON with text before",
			input:    "Here is the JSON:\n{\"key\": \"value\"}",
			expected: `{"key": "value"}`,
		},
		{
			name:     "JSON with text after",
			input:    "{\"key\": \"value\"}\nThat's it!",
			expected: `{"key": "value"}`,
		},
		{
			name:     "nested JSON",
			input:    `{"outer": {"inner": "value"}}`,
			expected: `{"outer": {"inner": "value"}}`,
		},
		{
			name:     "JSON with string containing braces",
			input:    `{"message": "This is a {test} string"}`,
			expected: `{"message": "This is a {test} string"}`,
		},
		{
			name:     "JSON with escaped quotes",
			input:    `{"message": "She said \"hello\""}`,
			expected: `{"message": "She said \"hello\""}`,
		},
		{
			name:     "JSON with multiple objects (should extract first)",
			input:    `{"first": "object"} {"second": "object"}`,
			expected: `{"first": "object"}`,
		},
		{
			name:     "complex nested JSON",
			input:    `{"summary": "test", "data": {"nested": {"deep": "value"}}, "array": [1, 2, 3]}`,
			expected: `{"summary": "test", "data": {"nested": {"deep": "value"}}, "array": [1, 2, 3]}`,
		},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			result := extractJSON(tt.input)
			if result != tt.expected {
				t.Errorf("Expected %s, got %s", tt.expected, result)