}
    }

    @Nested
    @DisplayName("Credit Tests")
    class CreditTests {

        @Test
        @DisplayName("Should return credit as granted credits")
        void testGetGrantedCredits() {
            int credits = 10;
            response.setCreditForTest(credits);

            assertEquals(credits, response.getGrantedCredits());
        }
    }

    @Nested

    /**
     * Grant a send credit
     */
    public void grantSendCredit() {
        sendCredits.incrementAndGet();
    }

    /**
     * Grant a receive credit
     */
    public void grantReceiveCredit() {
        receiveCredits.incrementAndGet();
    }

    /**
     * Get number of available send credits
     *
     * @return available send credits
     */

    }

    /**
     * Gets the credit count for this message.
     *
     * @return the credit
     */
    public final int getCredit() {
        return this.credit;
    }

    /**
     * Sets the credit count for this message.
     *
     * @param credit
     *            the credit to set
     */
    public final void setCredit(final int credit) {
        this.credit = credit;
    }

        this.creditsGranted = 0;
    }

    /**
     * Get initial number of credits to request
     *
     * @return initial credits
     */
    public int getInitialCredits() {
        return initialCredits;
    }

    /**
     * Set initial credits
     *
     * @param initialCredits initial credits to request
     */
    public void setInitialCredits(int initialCredits) {

    @Test
    @DisplayName("Test setRequestCredits sets credit value")
    void testSetRequestCredits() {
        // Given
        Smb2CancelRequest request = new Smb2CancelRequest(mockConfig, 1L, 0L);
        int credits = 10;

        // When
        request.setRequestCredits(credits);

        // Then
        assertEquals(credits, request.getCredit(), "Credits should be set correctly");
    }

    @Test

        assertEquals(expectedCreditCost, actualCreditCost, "Credit cost should match the mocked value.");
        verify(mockRequest, times(1)).getCreditCost(); // Verify method was called once
    }

    @Test
    void testSetRequestCredits() {
        // Test case for setRequestCredits method
        int credits = 5;
        mockRequest.setRequestCredits(credits);

                Thread.currentThread().interrupt();
                throw new IOException("Interrupted waiting for send credit", e);
            }
        }
        
        if (!rdmaConnection.canSend()) {
            throw new IOException("Timeout waiting for RDMA send credit");
        }
    }
    
    @Override
    public void disconnect() throws IOException {
        try {

     */
    public static final int SMB2_GLOBAL_CAP_DFS = 0x1;

    /**
     * Server supports leasing
     */
    public static final int SMB2_GLOBAL_CAP_LEASING = 0x2;

    /**
     * Server supports multi-credit operations
     */
    public static final int SMB2_GLOBAL_CAP_LARGE_MTU = 0x4;

    /**
     * Server supports multi-channel connections
     */
    public static final int SMB2_GLOBAL_CAP_MULTI_CHANNEL = 0x8;

        }
    }

    @Nested
    @DisplayName("Credit Management Tests")
    class CreditManagementTests {

        @Test
        @DisplayName("Should get granted credits from credit field")
        void testGetGrantedCredits() {
            int credits = 15;
            echoResponse.setCredit(credits);

            assertEquals(credits, echoResponse.getGrantedCredits());
        }
    }

pull requests directly into the master branch. Instead, once a pull request is
ready for merging, we'll make the appropriate changes in the internal codebase
and, when the change is synced out, give the pull request author credit for the
commit.

Contributor License Agreement
-----------------------------

Contributions to any Google project must be accompanied by a Contributor