diff --git a/src/test/java/com/profiler/common/util/HeaderUtilsTest.java b/src/test/java/com/profiler/common/util/HeaderUtilsTest.java new file mode 100644 index 000000000..07c19efd3 --- /dev/null +++ b/src/test/java/com/profiler/common/util/HeaderUtilsTest.java @@ -0,0 +1,28 @@ +package com.profiler.common.util; + +import com.profiler.common.dto.Header; +import com.profiler.common.util.HeaderUtils; +import org.apache.thrift.TException; +import org.junit.Assert; +import org.junit.Test; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +public class HeaderUtilsTest { + + private final Logger logger = LoggerFactory.getLogger(this.getClass().getName()); + + @Test + public void validateSignature() throws TException { + Header header = new Header(); + Assert.assertTrue(HeaderUtils.validateSignature(header.getSignature())); + + + Header error = new Header((byte) 0x11, (byte) 0x20, (short) 1); + Assert.assertTrue(!HeaderUtils.validateSignature(error.getSignature())); + + + logger.info(header.toString()); + } + +} diff --git a/src/test/java/com/profiler/common/util/SpanUtilsTest.java b/src/test/java/com/profiler/common/util/SpanUtilsTest.java index a5955a501..e0f1ef1b8 100644 --- a/src/test/java/com/profiler/common/util/SpanUtilsTest.java +++ b/src/test/java/com/profiler/common/util/SpanUtilsTest.java @@ -9,6 +9,13 @@ import org.junit.Test; * */ public class SpanUtilsTest { + @Test + public void testGetTraceIndexRowKeyWhiteSpace() throws Exception { + String agentId = "test test"; + long time = System.currentTimeMillis(); + check(agentId, time); + } + @Test public void testGetTraceIndexRowKey1() throws Exception { String agentId = "test"; @@ -49,7 +56,7 @@ public class SpanUtilsTest { byte[] traceIndexRowKey = SpanUtils.getTraceIndexRowKey(span); - String agentId = Bytes.toString(traceIndexRowKey, 0, agentId0.length()); + String agentId = Bytes.toString(traceIndexRowKey, 0, 24).trim(); Assert.assertEquals(agentId0, agentId); long time = Bytes.toLong(traceIndexRowKey, SpanUtils.AGENT_NAME_LIMIT);