This is an automated email from the ASF dual-hosted git repository.
xiaoyu pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-shenyu.git
The following commit(s) were added to refs/heads/master by this push:
new f51c80b Add integrated test for cryptor request plugin (#2284)
f51c80b is described below
commit f51c80b23594938fa2fdfb351485efc5d2854209
Author: Kunshuai Zhu <[email protected]>
AuthorDate: Sat Oct 30 11:14:45 2021 +0800
Add integrated test for cryptor request plugin (#2284)
---
.../shenyu-integrated-test-http/pom.xml | 6 +
.../http/combination/CryptorRequestPluginTest.java | 136 +++++++++++++++++++++
2 files changed, 142 insertions(+)
diff --git a/shenyu-integrated-test/shenyu-integrated-test-http/pom.xml
b/shenyu-integrated-test/shenyu-integrated-test-http/pom.xml
index 973eb94..5473df0 100644
--- a/shenyu-integrated-test/shenyu-integrated-test-http/pom.xml
+++ b/shenyu-integrated-test/shenyu-integrated-test-http/pom.xml
@@ -95,6 +95,12 @@
</dependency>
<!-- apache shenyu param_mapping plugin end-->
+ <dependency>
+ <groupId>org.apache.shenyu</groupId>
+ <artifactId>shenyu-spring-boot-starter-plugin-cryptor</artifactId>
+ <version>${project.version}</version>
+ </dependency>
+
<!-- https://mvnrepository.com/artifact/com.google.code.gson/gson -->
<dependency>
<groupId>com.google.code.gson</groupId>
diff --git
a/shenyu-integrated-test/shenyu-integrated-test-http/src/test/java/org/apache/shenyu/integrated/test/http/combination/CryptorRequestPluginTest.java
b/shenyu-integrated-test/shenyu-integrated-test-http/src/test/java/org/apache/shenyu/integrated/test/http/combination/CryptorRequestPluginTest.java
new file mode 100644
index 0000000..1b0a12a
--- /dev/null
+++
b/shenyu-integrated-test/shenyu-integrated-test-http/src/test/java/org/apache/shenyu/integrated/test/http/combination/CryptorRequestPluginTest.java
@@ -0,0 +1,136 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one or more
+ * contributor license agreements. See the NOTICE file distributed with
+ * this work for additional information regarding copyright ownership.
+ * The ASF licenses this file to You under the Apache License, Version 2.0
+ * (the "License"); you may not use this file except in compliance with
+ * the License. You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.apache.shenyu.integrated.test.http.combination;
+
+import com.google.gson.JsonObject;
+import org.apache.shenyu.common.dto.ConditionData;
+import org.apache.shenyu.common.enums.OperatorEnum;
+import org.apache.shenyu.common.enums.ParamTypeEnum;
+import org.apache.shenyu.common.enums.PluginEnum;
+import org.apache.shenyu.common.utils.JsonUtils;
+import org.apache.shenyu.integratedtest.common.AbstractPluginDataInit;
+import org.apache.shenyu.integratedtest.common.dto.UserDTO;
+import org.apache.shenyu.integratedtest.common.helper.HttpHelper;
+import org.apache.shenyu.plugin.cryptor.dto.CryptorRuleHandle;
+import org.apache.shenyu.plugin.cryptor.strategy.RsaStrategy;
+import org.apache.shenyu.web.controller.LocalPluginController.RuleLocalData;
+import org.junit.After;
+import org.junit.Before;
+import org.junit.Test;
+
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
+
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertThat;
+
+public class CryptorRequestPluginTest extends AbstractPluginDataInit {
+
+ private static final String RSA_PRIVATE_KEY =
"MIIBVQIBADANBgkqhkiG9w0BAQEFAASCAT8wggE7AgEAAkEAvEXyUDh5qliWhM6KrpTFi1OXumoJQzMfSr8XjfKa/kHKb1uxr7N8lJd3I850m2IYrxckFCQW6nrnRKctm"
+ +
"iMgZQIDAQABAkBEFbdvMz0sUST9mgOk5sAZhn1UOIxo9M/YJArMlnNehqQs3Pv8RD6ASisgs19XnBhcUNdl2ecfxddp7OVQ6PVxAiEA+XagQdbwkFrEjUsPqPQTweKkc7aoVGJfifEGWvCKtAcCIQDBNN0K5vlVV5YKnA5WtDAN"
+ +
"K31oRGjTJe5D94IRYRHlMwIgQUH++jo4BAs6j5urJQ90e6vGSV7m+ewiAfvDJdb28dECIQCW9r9grWlVDcLnN1jc1p5VLA4pUoq1sYWjBdpTyg05kQIhAJUSfs1+gDFtCRekCwvn5QEnxJVJXD6SQPAPkEHS9Drb";
+
+ private static final String RSA_PUBLIC_KEY =
"MFwwDQYJKoZIhvcNAQEBBQADSwAwSAJBALxF8lA4eapYloTOiq6UxYtTl7pqCUMzH0q/F43ymv5Bym9bsa+zfJSXdyPOdJtiGK8XJBQkFup650SnLZojIGUCAwEAAQ==";
+
+ private static final String TEST_PATH = "/http/test/payment";
+
+ private static final String TEST_USER_ID = "10001";
+
+ private static final String TEST_USER_NAME = "user_name";
+
+ private static final RsaStrategy RSA_STRATEGY = new RsaStrategy();
+
+ @Before
+ public void setup() throws IOException {
+ String pluginResult = initPlugin(PluginEnum.CRYPTOR_REQUEST.getName(),
null);
+ assertThat(pluginResult, is("success"));
+ }
+
+ @Test
+ public void testDecryptRequest() throws Exception {
+ String selectorAndRulesResult =
initSelectorAndRules(PluginEnum.CRYPTOR_REQUEST.getName(),
+ "", buildSelectorConditionList(),
buildRuleLocalDataList("data", "decrypt"));
+ assertThat(selectorAndRulesResult, is("success"));
+
+ JsonObject jsonObject = new JsonObject();
+ jsonObject.addProperty("userId", TEST_USER_ID);
+ jsonObject.addProperty("userName", TEST_USER_NAME);
+ JsonObject request = new JsonObject();
+ request.addProperty("data", RSA_STRATEGY.encrypt(RSA_PUBLIC_KEY,
jsonObject.toString()));
+ UserDTO actualUser = HttpHelper.INSTANCE.postGateway(TEST_PATH,
request, UserDTO.class);
+
+ assertThat(actualUser.getUserId(), is(TEST_USER_ID));
+ assertThat(actualUser.getUserName(), is(TEST_USER_NAME));
+ }
+
+ @Test
+ public void testEncryptRequest() throws Exception {
+ String selectorAndRulesResult =
initSelectorAndRules(PluginEnum.CRYPTOR_REQUEST.getName(),
+ "", buildSelectorConditionList(),
buildRuleLocalDataList("userId", "encrypt"));
+ assertThat(selectorAndRulesResult, is("success"));
+
+ JsonObject request = new JsonObject();
+ request.addProperty("userId", TEST_USER_ID);
+ request.addProperty("userName", TEST_USER_NAME);
+ UserDTO actualUser = HttpHelper.INSTANCE.postGateway(TEST_PATH,
request, UserDTO.class);
+
+ assertThat(RSA_STRATEGY.decrypt(RSA_PRIVATE_KEY,
actualUser.getUserId()), is(TEST_USER_ID));
+ assertThat(actualUser.getUserName(), is(TEST_USER_NAME));
+ }
+
+ private List<ConditionData> buildSelectorConditionList() {
+ ConditionData conditionData = new ConditionData();
+ conditionData.setParamType(ParamTypeEnum.URI.getName());
+ conditionData.setOperator(OperatorEnum.EQ.getAlias());
+ conditionData.setParamValue(TEST_PATH);
+ return Collections.singletonList(conditionData);
+ }
+
+ private List<RuleLocalData> buildRuleLocalDataList(final String
fieldNames, final String way) {
+ List<RuleLocalData> ruleLocalDataList = new ArrayList<>();
+ ruleLocalDataList.add(buildRuleLocalData(fieldNames, way));
+ return ruleLocalDataList;
+ }
+
+ private RuleLocalData buildRuleLocalData(final String fieldNames, final
String way) {
+ final RuleLocalData ruleLocalData = new RuleLocalData();
+
+ CryptorRuleHandle cryptorRuleHandle = new CryptorRuleHandle();
+ cryptorRuleHandle.setDecryptKey(RSA_PRIVATE_KEY);
+ cryptorRuleHandle.setEncryptKey(RSA_PUBLIC_KEY);
+ cryptorRuleHandle.setStrategyName("rsa");
+ cryptorRuleHandle.setFieldNames(fieldNames);
+ cryptorRuleHandle.setWay(way);
+
+ ruleLocalData.setRuleHandler(JsonUtils.toJson(cryptorRuleHandle));
+ ConditionData conditionData = new ConditionData();
+ conditionData.setParamType(ParamTypeEnum.URI.getName());
+ conditionData.setOperator(OperatorEnum.EQ.getAlias());
+ conditionData.setParamValue(TEST_PATH);
+
ruleLocalData.setConditionDataList(Collections.singletonList(conditionData));
+ return ruleLocalData;
+ }
+
+ @After
+ public void clean() throws IOException {
+ String cleanResult =
cleanPluginData(PluginEnum.CRYPTOR_REQUEST.getName());
+ assertThat(cleanResult, is("success"));
+ }
+}
