fix tests

apache · Sep 14, 2023 · f139824 · f139824
1 parent 6711c9a
commit f139824
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/hbase-server/src/main/java/org/apache/hadoop/hbase/namequeues/impl/SlowLogQueueService.java b/hbase-server/src/main/java/org/apache/hadoop/hbase/namequeues/impl/SlowLogQueueService.java
@@ -185,6 +185,9 @@ public void consumeEventFromDisruptor(NamedQueuePayload namedQueuePayload) {
 
   private static Collection<HBaseProtos.NameBytesPair>
     buildNameBytesPairs(Map<String, byte[]> attributes) {
+    if (attributes == null) {
+      return Collections.emptySet();
+    }
     return attributes.entrySet().stream().map(attr -> HBaseProtos.NameBytesPair.newBuilder()
       .setName(attr.getKey()).setValue(ByteString.copyFrom(attr.getValue())).build())
       .collect(Collectors.toSet());

diff --git a/hbase-server/src/test/java/org/apache/hadoop/hbase/namequeues/TestNamedQueueRecorder.java b/hbase-server/src/test/java/org/apache/hadoop/hbase/namequeues/TestNamedQueueRecorder.java
@@ -756,20 +756,19 @@ public long getSize() {
 
       @Override
       public RPCProtos.RequestHeader getHeader() {
-        RPCProtos.RequestHeader.Builder builder = RPCProtos.RequestHeader.newBuilder();
-        REQUEST_HEADERS.forEach(builder::addAttribute);
-        return builder.build();
+        return null;
       }
 
       @Override
       public Map<String, byte[]> getConnectionAttributes() {
-        return CONNECTION_HEADERS.stream().collect(Collectors.toMap(
-          nameBytesPair -> nameBytesPair.getName(), nameBytesPair -> nameBytesPair.toByteArray()));
+        return CONNECTION_HEADERS.stream().collect(Collectors
+          .toMap(HBaseProtos.NameBytesPair::getName, pair -> pair.getValue().toByteArray()));
       }
 
       @Override
       public Map<String, byte[]> getRequestAttributes() {
-        return null;
+        return REQUEST_HEADERS.stream().collect(Collectors.toMap(HBaseProtos.NameBytesPair::getName,
+          pair -> pair.getValue().toByteArray()));
       }
 
       @Override