CommonServerMessageBlockResponse resp = mock(CommonServerMessageBlockResponse.class);

        // Configure the tree to throw transport error on first call, succeed on second
        when(tree.send(eq(req), eq(resp), anySet()))
                .thenThrow(new SmbException("transport error", new jcifs.util.transport.TransportException()))
                .thenReturn(resp);

        setTree(c, tree);

    public T initResponse(final CIFSContext tc) {
        final T resp = createResponse(tc, this);
        if (resp == null) {
            return null;
        }
        resp.setDigest(getDigest());
        setResponse(resp);

        final ServerMessageBlock2 n = getNext();
        if (n instanceof ServerMessageBlock2Request<?>) {
            resp.setNext(((ServerMessageBlock2Request<?>) n).initResponse(tc));
        }

    private SmbComWriteResponse resp;
    private Configuration cfgMock;

    @BeforeEach
    public void setUp() {
        // create a mock configuration that satisfies constructor requirements.
        cfgMock = mock(Configuration.class);
        resp = new SmbComWriteResponse(cfgMock);
    }

    @Test
    public void testInitialCountIsZero() {
        assertEquals(0, resp.getCount(), "Initial count should be zero");

	return DEK{
		KeyID:      name,
		Version:    resp[0].Version,
		Plaintext:  resp[0].Plaintext,
		Ciphertext: resp[0].Ciphertext,
	}, nil
}

func (c *kmsConn) Decrypt(ctx context.Context, req *DecryptRequest) ([]byte, error) {
	aad, err := req.AssociatedData.MarshalText()
	if err != nil {
		return nil, err
	}

	ciphertext, _ := parseCiphertext(req.Ciphertext)

            Smb2LogoffRequest req = newRequest();

            // Act
            Smb2LogoffResponse resp = req.createResponse(cifsContext, req);

            // Assert
            assertNotNull(resp, "Response should be created");
            assertTrue(resp instanceof Smb2LogoffResponse, "Response type should be Smb2LogoffResponse");
            verify(cifsContext, times(1)).getConfig();

	json.NewEncoder(w).Encode(map[string]string{
		"reason": fmt.Sprintf("%v", err),
	})
}

type Resp struct {
	User               string                 `json:"user"`
	MaxValiditySeconds int                    `json:"maxValiditySeconds"`
	Claims             map[string]interface{} `json:"claims"`
}

var tokens map[string]Resp = map[string]Resp{
	"aaa": {
		User:               "Alice",
		MaxValiditySeconds: 3600,

            }

            if (!resp.isReceived()) {
                throw new CIFSException("Did not receive response");
            }

            if (resp.getErrorCode() == 0x10B) {
                this.handle.markClosed();
            }
            if (resp.getErrorCode() == 0x10C) {
                resp.getNotifyInformation().clear();
            }

        final ServerMessageBlock2Response resp = (ServerMessageBlock2Response) response;
        synchronized (resp) {
            if (resp.isAsync() && !resp.isAsyncHandled() && resp.getStatus() == NtStatus.NT_STATUS_PENDING && resp.getAsyncId() != 0) {
                resp.setAsyncHandled(true);
                final boolean first = !req.isAsync();
                req.setAsyncId(resp.getAsyncId());

        Mockito.doAnswer(inv -> {
            Object resp = inv.getArgument(1);
            // Set private fields: status on SmbComTransactionResponse and available on TransPeekNamedPipeResponse
            Field statusField = jcifs.internal.smb1.trans.SmbComTransactionResponse.class.getDeclaredField("status");
            statusField.setAccessible(true);
            statusField.setInt(resp, status);

		}
	}
}

func fwdStatusToAPIError(statusCode int, resp *http.Response) *APIError {
	if statusCode < http.StatusBadRequest {
		return nil
	}
	desc := resp.Header.Get(xhttp.AmzFwdErrorMessage)
	if strings.TrimSpace(desc) == "" {
		apiErr := errorCodes.ToAPIErr(ErrInvalidRequest)
		return &apiErr
	}
	code := resp.Header.Get(xhttp.AmzFwdErrorCode)
	if strings.TrimSpace(code) == "" {