修改ui测试

sermant-io · Jan 2, 2025 · 939ff4e · 939ff4e
1 parent f3bf488
commit 939ff4e
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 6 deletions.
diff --git a/...transmit-plugin/src/test/java/io/sermant/router/transmit/wrapper/CallableWrapperTest.java b/...transmit-plugin/src/test/java/io/sermant/router/transmit/wrapper/CallableWrapperTest.java
@@ -43,8 +43,10 @@ public void testCanTransmit() throws Exception {
             return obj;
         }, new RequestTag(null), new RequestData(null, null, null), false);
 
-        Assert.assertNull(ThreadLocalUtils.getRequestData());
-        Assert.assertNull(ThreadLocalUtils.getRequestTag());
+        Assert.assertTrue(ThreadLocalUtils.getRequestData() == null
+                || ThreadLocalUtils.getRequestData().getPath() == null);
+        Assert.assertTrue(ThreadLocalUtils.getRequestTag() == null
+                || ThreadLocalUtils.getRequestTag().getTag().isEmpty());
 
         Assert.assertEquals(obj, wrapper.call());
 

diff --git a/...ugin/src/test/java/io/sermant/router/transmit/wrapper/RunnableAndCallableWrapperTest.java b/...ugin/src/test/java/io/sermant/router/transmit/wrapper/RunnableAndCallableWrapperTest.java
@@ -92,8 +92,10 @@ public void testCallCanTransmit() throws Exception {
             return obj;
         }, new RequestTag(null), new RequestData(null, null, null), false);
 
-        Assert.assertNull(ThreadLocalUtils.getRequestData());
-        Assert.assertNull(ThreadLocalUtils.getRequestTag());
+        Assert.assertTrue(ThreadLocalUtils.getRequestData() == null
+                || ThreadLocalUtils.getRequestData().getPath() == null);
+        Assert.assertTrue(ThreadLocalUtils.getRequestTag() == null
+                || ThreadLocalUtils.getRequestTag().getTag().isEmpty());
 
         Assert.assertEquals(obj, wrapper.call());
 

diff --git a/...transmit-plugin/src/test/java/io/sermant/router/transmit/wrapper/RunnableWrapperTest.java b/...transmit-plugin/src/test/java/io/sermant/router/transmit/wrapper/RunnableWrapperTest.java
@@ -41,8 +41,10 @@ public void testCanTransmit() {
                     || ThreadLocalUtils.getRequestTag().getTag().isEmpty());
         }, new RequestTag(null), new RequestData(null, null, null), false);
 
-        Assert.assertNull(ThreadLocalUtils.getRequestData());
-        Assert.assertNull(ThreadLocalUtils.getRequestTag());
+        Assert.assertTrue(ThreadLocalUtils.getRequestData() == null
+                || ThreadLocalUtils.getRequestData().getPath() == null);
+        Assert.assertTrue(ThreadLocalUtils.getRequestTag() == null
+                || ThreadLocalUtils.getRequestTag().getTag().isEmpty());
 
         wrapper.run();