List<SortBuilder<?>> sorts = queryContext.sortBuilders();
        assertNotNull(sorts);
        assertTrue(sorts.isEmpty());

        SortBuilder<?> sort = SortBuilders.fieldSort("field");
        queryContext.addSorts(sort);

        sorts = queryContext.sortBuilders();
        assertEquals(1, sorts.size());
        assertEquals(sort, sorts.get(0));
    }

    public void test_getSortFields() {
        String[] fields = { "sort1", "sort2", "sort3" };
        queryFieldConfig.setSortFields(fields);

        String[] result = queryFieldConfig.getSortFields();
        assertSame(fields, result);
    }

    @Test
    public void test_setSortFields() {
        String[] fields = { "sort1", "sort2", "sort3" };
        queryFieldConfig.setSortFields(fields);

        assertTrue(queryFieldConfig.facetFieldSet.contains("facet3"));
    }

    /**
     * Test that setting sort fields creates both array and Set.
     */
    @Test
    public void test_setSortFields_createsSet() {
        String[] fields = { "sort1", "sort2", "sort3" };
        queryFieldConfig.setSortFields(fields);

        // Verify array is set
        assertNotNull(queryFieldConfig.sortFields);

            short maxCpu = 0;
            for (final NodeStats nodeStats : response.getNodes()) {
                final OsStats os = nodeStats.getOs();
                if (os != null && os.getCpu() != null) {
                    final short percent = os.getCpu().getPercent();
                    if (percent > maxCpu) {

     */
    private static class TestableLoadControlFilter extends LoadControlFilter {
        private short cpuPercent = 0;
        private int webLoadControl = 0;
        private int apiLoadControl = 0;

        public void setCpuPercent(final short cpuPercent) {
            this.cpuPercent = cpuPercent;
        }

        public void setWebLoadControl(final int webLoadControl) {

        assertEquals("\"longKey\":9876543210", buf.toString());
    }

    // Test append method with Short value
    @Test
    public void test_append_shortValue() {
        StringBuilder buf = new StringBuilder();
        Supplier<Object> supplier = () -> Short.valueOf((short) 42);

        monitorTarget.append(buf, "shortKey", supplier);
        assertEquals("\"shortKey\":42", buf.toString());
    }

    protected String productVersion;

    /** The end-of-life timestamp for the application. */
    protected long eolTime;

    private short systemCpuPercent;

    private long systemCpuCheckTime;

    private long systemCpuCheckInterval = 1000L;

    private volatile short searchEngineCpuPercent;

    private volatile TimeoutTask loadControlMonitorTask;

    /** A map of listeners for configuration updates. */

            method.setAccessible(true);

            org.opensearch.search.sort.SortBuilder<?> result = (org.opensearch.search.sort.SortBuilder<?>) method.invoke(queryHelper,
                    "timestamp", org.opensearch.search.sort.SortOrder.DESC);

            assertNotNull(result);

        // System.out.println("size: " + getObjectSize(""));
        assertEquals(24L, MemoryUtil.sizeOf(Integer.MAX_VALUE));
        assertEquals(24L, MemoryUtil.sizeOf(Long.MAX_VALUE));
        assertEquals(24L, MemoryUtil.sizeOf(Short.MAX_VALUE));
        assertEquals(24L, MemoryUtil.sizeOf(Float.MAX_VALUE));
        assertEquals(24L, MemoryUtil.sizeOf(Double.MAX_VALUE));
        assertEquals(24L, MemoryUtil.sizeOf(Byte.MAX_VALUE));

    @Test
    public void test_buildAssistantHistoryContent_truncated_shortContent() {
        final ChatMessage msg = ChatMessage.assistantMessage("Short");
        final String result = chatClient.testBuildAssistantHistoryContent(msg, "truncated", 500, 500);
        assertEquals("Short", result);
    }

    // ========== extractHistory tests ==========

    @Test
    public void test_extractHistory_fullMode() {