@eeacms/volto-eea-chatbot 1.0.9 → 1.0.11

package/src/ChatBlock/tests/ClaimSegments.test.jsx

@@ -0,0 +1,206 @@
+import renderer from 'react-test-renderer';
+import '@testing-library/jest-dom/extend-expect';
+import { ClaimSegments } from '../components/markdown/ClaimSegments';
+
+// Mock semantic-ui-react
+jest.mock('semantic-ui-react', () => ({
+  Tab: ({ panes, activeIndex }) => (
+    <div data-testid="tab">
+      <div data-testid="tab-menu">
+        {panes.map((pane, i) => (
+          <div
+            key={i}
+            data-testid={`menu-item-${i}`}
+            className={pane.menuItem.className}
+            onClick={pane.menuItem.onClick}
+            onKeydown={() => {}}
+            role="tab"
+            tabIndex={i === activeIndex ? 0 : -1}
+            aria-selected={i === activeIndex}
+            aria-controls={`tab-pane-${i}`}
+          >
+            {pane.menuItem.content}
+          </div>
+        ))}
+      </div>
+      <div data-testid="tab-content">{panes[activeIndex]?.render()}</div>
+    </div>
+  ),
+  TabPane: ({ children }) => <div data-testid="tab-pane">{children}</div>,
+}));
+
+// Mock RenderClaimView
+jest.mock('../components/markdown/RenderClaimView', () => ({
+  RenderClaimView: () => (
+    <div data-testid="render-claim-view">RenderClaimView</div>
+  ),
+}));
+
+describe('ClaimSegments', () => {
+  const defaultProps = {
+    segmentIds: [1, 2],
+    segments: {
+      1: { id: 1, startOffset: 0, endOffset: 10 },
+      2: { id: 2, startOffset: 15, endOffset: 25 },
+    },
+    citedSources: [
+      {
+        id: 'source1',
+        semantic_identifier: 'Source Document 1',
+        link: 'https://example.com/1',
+        source_type: 'web',
+        halloumiContext: 'This is the context text for source 1.',
+        index: 1,
+      },
+    ],
+  };
+
+  it('renders with basic props', () => {
+    const component = renderer.create(<ClaimSegments {...defaultProps} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('renders with multiple sources', () => {
+    const props = {
+      ...defaultProps,
+      citedSources: [
+        {
+          id: 'source1',
+          semantic_identifier: 'Source 1',
+          link: 'https://example.com/1',
+          source_type: 'web',
+          halloumiContext: 'Context for source 1.',
+          index: 1,
+        },
+        {
+          id: 'source2',
+          semantic_identifier: 'Source 2',
+          link: 'https://example.com/2',
+          source_type: 'file',
+          halloumiContext: 'Context for source 2.',
+          index: 2,
+        },
+      ],
+      segments: {
+        1: { id: 1, startOffset: 0, endOffset: 10 },
+        2: { id: 2, startOffset: 22, endOffset: 32 },
+      },
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles empty segmentIds', () => {
+    const props = {
+      ...defaultProps,
+      segmentIds: [],
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles null segmentIds', () => {
+    const props = {
+      ...defaultProps,
+      segmentIds: null,
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles missing segments in segments object', () => {
+    // Suppress console.warn for this test
+    const originalWarn = console.warn;
+    console.warn = jest.fn();
+
+    const props = {
+      ...defaultProps,
+      segmentIds: [1, 999], // 999 doesn't exist
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+    expect(console.warn).toHaveBeenCalled();
+
+    console.warn = originalWarn;
+  });
+
+  it('renders source without link', () => {
+    const props = {
+      ...defaultProps,
+      citedSources: [
+        {
+          id: 'source1',
+          semantic_identifier: 'Source 1',
+          link: null,
+          source_type: 'file',
+          halloumiContext: 'Context for source 1.',
+          index: 1,
+        },
+      ],
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('renders with file source type icon', () => {
+    const props = {
+      ...defaultProps,
+      citedSources: [
+        {
+          id: 'source1',
+          semantic_identifier: 'Internal Document',
+          link: 'https://example.com/1',
+          source_type: 'file',
+          halloumiContext: 'Context text.',
+          index: 1,
+        },
+      ],
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('filters out sources without matching snippets', () => {
+    const props = {
+      segmentIds: [1],
+      segments: {
+        1: { id: 1, startOffset: 0, endOffset: 10 },
+      },
+      citedSources: [
+        {
+          id: 'source1',
+          semantic_identifier: 'Source 1',
+          link: null,
+          source_type: 'web',
+          halloumiContext: 'Short context.',
+          index: 1,
+        },
+        {
+          id: 'source2',
+          semantic_identifier: 'Source 2 (no snippets)',
+          link: null,
+          source_type: 'web',
+          halloumiContext: 'This source has no matching segments.',
+          index: 2,
+        },
+      ],
+    };
+
+    const component = renderer.create(<ClaimSegments {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+});

package/src/ChatBlock/tests/CustomToolRenderer.test.jsx

@@ -0,0 +1,241 @@
+import React from 'react';
+import renderer, { act } from 'react-test-renderer';
+import '@testing-library/jest-dom/extend-expect';
+import { CustomToolRenderer } from '../packets/renderers/CustomToolRenderer';
+import { PacketType } from '../types/streamingModels';
+
+describe('CustomToolRenderer', () => {
+  const mockChildRenderer = (result) => (
+    <div data-testid="renderer-result">
+      <div data-testid="status">{result.status}</div>
+      <div data-testid="content">{result.content}</div>
+    </div>
+  );
+
+  it('renders running state when not complete', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'MyTool',
+          },
+        },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+    expect(props.onComplete).not.toHaveBeenCalled();
+  });
+
+  it('renders complete state', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'Calculator',
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'Calculator',
+            response_type: 'calculation_result',
+            data: { result: 42 },
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('renders with multiple deltas', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'MultiStepTool',
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'MultiStepTool',
+            response_type: 'step1',
+            data: { step: 1, status: 'processing' },
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'MultiStepTool',
+            response_type: 'step2',
+            data: { step: 2, status: 'complete' },
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles delta without response_type', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'SimpleTool',
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'SimpleTool',
+            data: { result: 'done' },
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles delta without data', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'NoDataTool',
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'NoDataTool',
+            response_type: 'status',
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('uses default tool name when not provided', () => {
+    const props = {
+      packets: [{ ind: 1, obj: { type: PacketType.SECTION_END } }],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles empty packets array', () => {
+    const props = {
+      packets: [],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('calls onComplete when section end is received', () => {
+    const onComplete = jest.fn();
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: { type: PacketType.CUSTOM_TOOL_START, tool_name: 'Test' },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete,
+      children: mockChildRenderer,
+    };
+
+    act(() => {
+      renderer.create(<CustomToolRenderer {...props} />);
+    });
+    expect(onComplete).toHaveBeenCalled();
+  });
+
+  it('renders complex nested data correctly', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_START,
+            tool_name: 'DataTool',
+          },
+        },
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.CUSTOM_TOOL_DELTA,
+            tool_name: 'DataTool',
+            response_type: 'complex',
+            data: {
+              nested: {
+                array: [1, 2, 3],
+                object: { key: 'value' },
+              },
+            },
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<CustomToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+});

package/src/ChatBlock/tests/FetchToolRenderer.test.jsx

@@ -0,0 +1,161 @@
+import React from 'react';
+import renderer, { act } from 'react-test-renderer';
+import '@testing-library/jest-dom/extend-expect';
+import { FetchToolRenderer } from '../packets/renderers/FetchToolRenderer';
+import { PacketType } from '../types/streamingModels';
+
+describe('FetchToolRenderer', () => {
+  const mockChildRenderer = (result) => (
+    <div data-testid="renderer-result">
+      <div data-testid="status">{result.status}</div>
+      <div data-testid="content">{result.content}</div>
+    </div>
+  );
+
+  it('renders fetching state when not complete', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.FETCH_TOOL_START,
+            queries: ['search query'],
+            documents: null,
+          },
+        },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+    expect(props.onComplete).not.toHaveBeenCalled();
+  });
+
+  it('renders complete state', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.FETCH_TOOL_START,
+            queries: ['search query'],
+            documents: [
+              {
+                document_id: 'doc1',
+                semantic_identifier: 'Document 1',
+                link: 'https://example.com/1',
+              },
+            ],
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('renders with multiple queries', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.FETCH_TOOL_START,
+            queries: ['first query', 'second query', 'third query'],
+            documents: null,
+          },
+        },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('renders with multiple documents', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.FETCH_TOOL_START,
+            queries: ['query'],
+            documents: [
+              { document_id: 'doc1' },
+              { document_id: 'doc2' },
+              { document_id: 'doc3' },
+            ],
+          },
+        },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles empty queries and documents', () => {
+    const props = {
+      packets: [
+        {
+          ind: 1,
+          obj: {
+            type: PacketType.FETCH_TOOL_START,
+            queries: null,
+            documents: null,
+          },
+        },
+      ],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('handles empty packets array', () => {
+    const props = {
+      packets: [],
+      onComplete: jest.fn(),
+      children: mockChildRenderer,
+    };
+
+    const component = renderer.create(<FetchToolRenderer {...props} />);
+    const json = component.toJSON();
+    expect(json).toMatchSnapshot();
+  });
+
+  it('calls onComplete when section end is received', () => {
+    const onComplete = jest.fn();
+    const props = {
+      packets: [
+        { ind: 1, obj: { type: PacketType.FETCH_TOOL_START, queries: [] } },
+        { ind: 1, obj: { type: PacketType.SECTION_END } },
+      ],
+      onComplete,
+      children: mockChildRenderer,
+    };
+
+    act(() => {
+      renderer.create(<FetchToolRenderer {...props} />);
+    });
+    expect(onComplete).toHaveBeenCalled();
+  });
+});