remove context finally

apache · Sep 13, 2024 · 3e6d5b1 · 3e6d5b1
1 parent 4527215
commit 3e6d5b1
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/server/src/test/java/org/apache/gravitino/server/web/rest/TestFilesetOperations.java b/server/src/test/java/org/apache/gravitino/server/web/rest/TestFilesetOperations.java
@@ -517,9 +517,13 @@ public void testGetFileLocation() {
           .thenAnswer(
               (Answer<String>)
                   invocation -> {
-                    CallerContext context = CallerContext.CallerContextHolder.get();
-                    callerContextMap.putAll(context.context());
-                    return subPath2;
+                    try {
+                      CallerContext context = CallerContext.CallerContextHolder.get();
+                      callerContextMap.putAll(context.context());
+                      return subPath2;
+                    } finally {
+                      CallerContext.CallerContextHolder.remove();
+                    }
                   });
       Response resp4 =
           target(filesetPath(metalake, catalog, schema) + "fileset3/location")