|
25 | 25 | import java.util.concurrent.ConcurrentHashMap; |
26 | 26 | import java.util.function.Function; |
27 | 27 |
|
28 | | -import static java.lang.StackWalker.Option.RETAIN_CLASS_REFERENCE; |
29 | 28 | import static java.util.Objects.requireNonNull; |
30 | 29 |
|
31 | 30 | public class TestPolicyManager extends PolicyManager { |
@@ -117,30 +116,15 @@ boolean isTriviallyAllowed(Class<?> requestingClass) { |
117 | 116 | if (isTriviallyAllowingTestCode && isTestCode(requestingClass)) { |
118 | 117 | return true; |
119 | 118 | } |
120 | | - if(super.isTriviallyAllowed(requestingClass)){ |
121 | | - return true; |
122 | | - }; |
123 | | - return isStackWithoutEntitlements(); |
| 119 | + return super.isTriviallyAllowed(requestingClass); |
124 | 120 | } |
125 | 121 |
|
126 | 122 | @Override |
127 | 123 | protected Collection<Path> getComponentPathsFromClass(Class<?> requestingClass) { |
128 | 124 | return classpath; // required to grant read access to the production source and test resources |
129 | 125 | } |
130 | 126 |
|
131 | | - private static boolean hasWithoutEntitlements(Class<?> clazz) { |
132 | | - return clazz.getAnnotation(ESTestCase.WithoutEntitlements.class) != null; |
133 | | - } |
134 | | - |
135 | | - private static boolean isStackWithoutEntitlements() { |
136 | | - return StackWalker.getInstance(RETAIN_CLASS_REFERENCE) |
137 | | - .walk(frames -> frames |
138 | | - .map(StackWalker.StackFrame::getDeclaringClass) |
139 | | - .filter(c -> isEntitlementClass(c) == false) |
140 | | - .anyMatch(TestPolicyManager::hasWithoutEntitlements)); |
141 | | - } |
142 | | - |
143 | | - private static boolean isEntitlementClass(Class<?> requestingClass) { |
| 127 | + private boolean isEntitlementClass(Class<?> requestingClass) { |
144 | 128 | return requestingClass.getPackageName().startsWith("org.elasticsearch.entitlement") |
145 | 129 | && (requestingClass.getName().contains("Test") == false); |
146 | 130 | } |
|
0 commit comments