You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@dubbo.apache.org by al...@apache.org on 2021/08/23 05:44:29 UTC
[dubbo] branch master updated: Fix mock:true configuration. (#8554)
This is an automated email from the ASF dual-hosted git repository.
albumenj pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/dubbo.git
The following commit(s) were added to refs/heads/master by this push:
new fd80172 Fix mock:true configuration. (#8554)
fd80172 is described below
commit fd80172bc20d2d8be58938062c58d252956d8e44
Author: changfu <ch...@gmail.com>
AuthorDate: Mon Aug 23 13:44:18 2021 +0800
Fix mock:true configuration. (#8554)
* fix issue8516
* add license to file
---
.../org/apache/dubbo/rpc/support/MockInvoker.java | 10 +++++---
.../org/apache/dubbo/rpc/support/DemoServiceA.java | 21 ++++++++++++++++
.../apache/dubbo/rpc/support/DemoServiceAMock.java | 28 +++++++++++++++++++++
.../org/apache/dubbo/rpc/support/DemoServiceB.java | 21 ++++++++++++++++
.../apache/dubbo/rpc/support/DemoServiceBMock.java | 29 ++++++++++++++++++++++
.../apache/dubbo/rpc/support/MockInvokerTest.java | 25 +++++++++++++++++++
6 files changed, 131 insertions(+), 3 deletions(-)
diff --git a/dubbo-rpc/dubbo-rpc-api/src/main/java/org/apache/dubbo/rpc/support/MockInvoker.java b/dubbo-rpc/dubbo-rpc-api/src/main/java/org/apache/dubbo/rpc/support/MockInvoker.java
index da1895b..690f483 100644
--- a/dubbo-rpc/dubbo-rpc-api/src/main/java/org/apache/dubbo/rpc/support/MockInvoker.java
+++ b/dubbo-rpc/dubbo-rpc-api/src/main/java/org/apache/dubbo/rpc/support/MockInvoker.java
@@ -156,14 +156,18 @@ final public class MockInvoker<T> implements Invoker<T> {
}
@SuppressWarnings("unchecked")
- private Invoker<T> getInvoker(String mockService) {
+ private Invoker<T> getInvoker(String mock) {
+ Class<T> serviceType = (Class<T>) ReflectUtils.forName(url.getServiceInterface());
+
+ final boolean isDefault = ConfigUtils.isDefault(mock);
+ // convert to actual mock service name
+ String mockService = isDefault ? serviceType.getName() + "Mock" : mock;
Invoker<T> invoker = (Invoker<T>) MOCK_MAP.get(mockService);
if (invoker != null) {
return invoker;
}
- Class<T> serviceType = (Class<T>) ReflectUtils.forName(url.getServiceInterface());
- T mockObject = (T) getMockObject(mockService, serviceType);
+ T mockObject = (T) getMockObject(mock, serviceType);
invoker = PROXY_FACTORY.getInvoker(mockObject, serviceType, url);
if (MOCK_MAP.size() < 10000) {
MOCK_MAP.put(mockService, invoker);
diff --git a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceA.java b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceA.java
new file mode 100644
index 0000000..5431836
--- /dev/null
+++ b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceA.java
@@ -0,0 +1,21 @@
+/*
+ * 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.dubbo.rpc.support;
+
+public interface DemoServiceA {
+ String methodA();
+}
diff --git a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceAMock.java b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceAMock.java
new file mode 100644
index 0000000..23c623a
--- /dev/null
+++ b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceAMock.java
@@ -0,0 +1,28 @@
+/*
+ * 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.dubbo.rpc.support;
+
+/**
+ * default mock service for DemoServiceA
+ */
+public class DemoServiceAMock implements DemoServiceA{
+ public static final String MOCK_VALUE = "mockA";
+ @Override
+ public String methodA() {
+ return MOCK_VALUE;
+ }
+}
diff --git a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceB.java b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceB.java
new file mode 100644
index 0000000..a2350a5
--- /dev/null
+++ b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceB.java
@@ -0,0 +1,21 @@
+/*
+ * 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.dubbo.rpc.support;
+
+public interface DemoServiceB {
+ String methodB();
+}
diff --git a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceBMock.java b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceBMock.java
new file mode 100644
index 0000000..ffc199d
--- /dev/null
+++ b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/DemoServiceBMock.java
@@ -0,0 +1,29 @@
+/*
+ * 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.dubbo.rpc.support;
+
+/**
+ * default mock service for DemoServiceA
+ */
+public class DemoServiceBMock implements DemoServiceB {
+ public static final String MOCK_VALUE = "mockB";
+
+ @Override
+ public String methodB() {
+ return MOCK_VALUE;
+ }
+}
diff --git a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/MockInvokerTest.java b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/MockInvokerTest.java
index b40951a..646c789 100644
--- a/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/MockInvokerTest.java
+++ b/dubbo-rpc/dubbo-rpc-api/src/test/java/org/apache/dubbo/rpc/support/MockInvokerTest.java
@@ -66,6 +66,31 @@ public class MockInvokerTest {
}
@Test
+ public void testGetDefaultObject() {
+ // test methodA in DemoServiceAMock
+ final Class<DemoServiceA> demoServiceAClass = DemoServiceA.class;
+ URL url = URL.valueOf("remote://1.2.3.4/" + demoServiceAClass.getName());
+ url = url.addParameter(MOCK_KEY, "force:true");
+ MockInvoker mockInvoker = new MockInvoker(url, demoServiceAClass);
+
+ RpcInvocation invocation = new RpcInvocation();
+ invocation.setMethodName("methodA");
+ Assertions.assertEquals(new HashMap<>(),
+ mockInvoker.invoke(invocation).getObjectAttachments());
+
+ // test methodB in DemoServiceBMock
+ final Class<DemoServiceB> demoServiceBClass = DemoServiceB.class;
+ url = URL.valueOf("remote://1.2.3.4/" + demoServiceBClass.getName());
+ url = url.addParameter(MOCK_KEY, "force:true");
+ mockInvoker = new MockInvoker(url, demoServiceBClass);
+ invocation = new RpcInvocation();
+ invocation.setMethodName("methodB");
+ Assertions.assertEquals(new HashMap<>(),
+ mockInvoker.invoke(invocation).getObjectAttachments());
+ }
+
+
+ @Test
public void testInvokeThrowsRpcException1() {
URL url = URL.valueOf("remote://1.2.3.4/" + String.class.getName());
MockInvoker mockInvoker = new MockInvoker(url, null);