if globalSiteReplicationSys.isEnabled() && cred.AccessKey != siteReplicatorSvcAcc {
		nsecret, err := getTokenSigningKey()
		if err != nil {
			return nil, toAPIErrorCode(r.Context(), err)
		}
		// sign root's temporary accounts also with site replicator creds
		if cred.ParentUser != globalActiveCred.AccessKey || cred.IsTemp() {
			secret = nsecret
		}
	}
	if cred.IsServiceAccount() {
		token = cred.SessionToken

            return 0;
        }

        @Override
        protected int readParameterWordsWireFormat(byte[] src, int srcIndex) {
            return 34; // Return proper word count for extended response
        }
    }

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.isUseBatching()).thenReturn(true);
    }

    @Test

	errNonTimestampArg   = errors.New("Expected a timestamp argument")
)

func (e *FuncExpr) getFunctionName() FuncName {
	switch {
	case e.SFunc != nil:
		return FuncName(strings.ToUpper(e.SFunc.FunctionName))
	case e.Count != nil:
		return aggFnCount
	case e.Cast != nil:
		return sqlFnCast
	case e.Substring != nil:
		return sqlFnSubstring
	case e.Extract != nil:
		return sqlFnExtract
	case e.Trim != nil:
		return sqlFnTrim

                    } catch (RuntimeException e) {
                        failureCount.incrementAndGet();
                    } catch (Exception e) {
                        // Other exceptions also count as failures in this context
                        failureCount.incrementAndGet();
                    }
                });
            }

            // Submit task to wipe key

      override fun contentLength(): Long = if (chunked) -1L else size

      override fun writeTo(sink: BufferedSink) {
        var count = 0
        while (count < size) {
          sink.write(buffer, 0, Math.min(size - count, writeSize.toLong()).toInt())
          count += writeSize
        }
      }
    }
  }

  @Test
  fun emptyResponseBody() {
    server.enqueue(

                    appendJson("user", entity.getUser(), buf).append(',');
                    appendJson("search-word", entity.getSearchWord(), buf).append(',');
                    appendJson("hit-count", entity.getHitCount(), buf).append(',');
                    appendJson("query-page-size", entity.getQueryPageSize(), buf).append(',');
                    appendJson("query-offset", entity.getQueryOffset(), buf).append(',');

        // Write structure size (9) - indicates error response
        SMBUtil.writeInt2(9, buffer, bufferIndex);
        // Write error context and error data
        buffer[bufferIndex + 2] = 1; // Error context count
        buffer[bufferIndex + 3] = 0; // Reserved
        SMBUtil.writeInt4(0, buffer, bufferIndex + 4); // ByteCount (0 for error)

        // Create a custom test response that tracks if readErrorResponse was called

      Suppliers.ofInstance(
          new StatsCounter() {
            @Override
            public void recordHits(int count) {}

            @Override
            public void recordMisses(int count) {}

            @SuppressWarnings("GoodTime") // b/122668874
            @Override
            public void recordLoadSuccess(long loadTime) {}

   * time it is closed. In such calls [byteCount] is the number of bytes returned to the
   * application. This may be smaller than the resource's byte count if were read to completion.
   *
   * This method is always invoked after [responseBodyStart].
   */
  open fun responseBodyEnd(
    call: Call,
    byteCount: Long,
  ) {
  }

  /**

      object : WebSocketListener() {
        override fun onFailure(
          webSocket: WebSocket,
          t: Throwable,
          response: Response?,
        ) {
          if (attempts.count > 0) {
            clientListener.setNextEventDelegate(this)
            webSockets.add(client.newWebSocket(request, clientListener))
            attempts.countDown()
          }
        }
      }