values = s3Client.parsePath("bucket/path");
        assertEquals("bucket", values[0]);
        assertEquals("path", values[1]);

        values = s3Client.parsePath("bucket/path1/path2");
        assertEquals("bucket", values[0]);
        assertEquals("path1/path2", values[1]);

        values = s3Client.parsePath("bucket/");
        assertEquals("bucket", values[0]);

        }

        @Test
        @DisplayName("All RPC packet flags should have unique values")
        void testRpcPacketFlagsUniqueness() {
            // Verify that RPC packet flags are unique and follow bit pattern
            assertEquals(1, Integer.bitCount(DcerpcConstants.RPC_C_PF_BROADCAST), "RPC_C_PF_BROADCAST should be a single bit");

import org.opensearch.search.aggregations.AggregationBuilders;
import org.opensearch.search.aggregations.BucketOrder;
import org.opensearch.search.aggregations.bucket.histogram.DateHistogramInterval;
import org.opensearch.search.aggregations.bucket.histogram.Histogram;
import org.opensearch.search.aggregations.bucket.terms.Terms;
import org.opensearch.search.aggregations.metrics.Avg;
import org.opensearch.search.aggregations.metrics.Cardinality;

        // When: Request another connection
        // Note: Real SmbTransportImpl will report as disconnected without actual socket
        SmbTransportImpl second = testPool.getSmbTransport(ctx, address, 445, false);

        // Then: Will create new connection since real transport has no socket
        assertNotSame(first, second, "Should create new connection when first is disconnected");
    }

    @Test

import org.opensearch.search.aggregations.bucket.filter.FilterAggregationBuilder;
import org.opensearch.search.aggregations.bucket.global.GlobalAggregationBuilder;
import org.opensearch.search.aggregations.bucket.histogram.DateHistogramAggregationBuilder;
import org.opensearch.search.aggregations.bucket.missing.MissingAggregationBuilder;
import org.opensearch.search.aggregations.bucket.range.DateRangeAggregationBuilder;

    private SmbTree tree;

    @BeforeEach
    void resetMocks() {
        Mockito.reset(session, transport, tree);
    }

    // Happy path: isInUse returns the mocked state
    @ParameterizedTest
    @DisplayName("isInUse reflects current mocked state")
    @ValueSource(booleans = { true, false })
    void isInUse_variants(boolean inUse) {
        when(session.isInUse()).thenReturn(inUse);

        try {
            final Bucket existingBucket = storage.get(bucket);
            if (existingBucket == null) {
                storage.create(BucketInfo.newBuilder(bucket).build());
                logger.info("Created storage bucket: {}", bucket);
            }
        } catch (final Exception e) {
            logger.warn("Failed to ensure bucket exists: {}", bucket, e);
        }
    }

    @Override

        // Test that the constructor correctly initializes the type and length fields.
        SessionRetargetResponsePacket packet = new SessionRetargetResponsePacket();
        assertEquals(SessionServicePacket.SESSION_RETARGET_RESPONSE, packet.type);
        assertEquals(6, packet.length);
    }

    @Test
    void writeTrailerWireFormatShouldReturnZero() {

    @Test
    @DisplayName("isDisconnected / isFailed reflect socket and state")
    void connectionStateChecks() throws Exception {
        // Arrange: simulate connected state and open socket
        setField(transport, "state", 3); // connected
        Socket s = mock(Socket.class);
        when(s.isClosed()).thenReturn(false);
        setField(transport, "socket", s);

        // Act & Assert

        int totalRead = packet.readWireFormat(bais, data, 0);

        assertEquals(14, totalRead); // 4 header + 10 trailer
        assertEquals(0x85, packet.type);
        assertEquals(10, packet.length);
    }

    @Test
    @DisplayName("writeHeaderWireFormat should write correct header")
    void testWriteHeaderWireFormat() {
        packet.type = SessionServicePacket.SESSION_REQUEST;