|
16 | 16 | */ |
17 | 17 | package org.apache.rocketmq.acl.common; |
18 | 18 |
|
19 | | -import com.alibaba.fastjson.JSONObject; |
| 19 | +import com.alibaba.fastjson2.JSONObject; |
20 | 20 | import org.apache.commons.lang3.StringUtils; |
21 | 21 | import org.apache.rocketmq.acl.plain.PlainAccessData; |
22 | 22 | import org.apache.rocketmq.common.PlainAccessConfig; |
|
32 | 32 | import java.util.Collections; |
33 | 33 | import java.util.List; |
34 | 34 | import java.util.Map; |
| 35 | +import java.util.Objects; |
35 | 36 | import java.util.UUID; |
36 | 37 |
|
| 38 | +import static org.junit.Assert.assertEquals; |
| 39 | +import static org.junit.Assert.assertNotNull; |
| 40 | +import static org.junit.Assert.assertTrue; |
| 41 | + |
37 | 42 | public class AclUtilsTest { |
38 | 43 |
|
39 | 44 | @Test |
@@ -296,4 +301,25 @@ public void getAclRPCHookTest() throws IOException { |
296 | 301 | Assert.assertNull(incompleteContRPCHook); |
297 | 302 | } |
298 | 303 | } |
| 304 | + |
| 305 | + @Test |
| 306 | + public void testGetAclRPCHookByFileName() { |
| 307 | + RPCHook actual = AclUtils.getAclRPCHook(Objects.requireNonNull(AclUtilsTest.class.getResource("/acl_hook/plain_acl.yml")).getPath()); |
| 308 | + assertNotNull(actual); |
| 309 | + assertTrue(actual instanceof AclClientRPCHook); |
| 310 | + assertAclClientRPCHook((AclClientRPCHook) actual); |
| 311 | + } |
| 312 | + |
| 313 | + @Test |
| 314 | + public void testGetAclRPCHookByInputStream() { |
| 315 | + RPCHook actual = AclUtils.getAclRPCHook(Objects.requireNonNull(AclUtilsTest.class.getResourceAsStream("/acl_hook/plain_acl.yml"))); |
| 316 | + assertNotNull(actual); |
| 317 | + assertTrue(actual instanceof AclClientRPCHook); |
| 318 | + assertAclClientRPCHook((AclClientRPCHook) actual); |
| 319 | + } |
| 320 | + |
| 321 | + private void assertAclClientRPCHook(final AclClientRPCHook actual) { |
| 322 | + assertEquals("rocketmq2", actual.getSessionCredentials().getAccessKey()); |
| 323 | + assertEquals("12345678", actual.getSessionCredentials().getSecretKey()); |
| 324 | + } |
299 | 325 | } |
0 commit comments