| /* |
| * 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.samples.governance; |
| |
| import org.apache.dubbo.rpc.RpcException; |
| import org.apache.dubbo.samples.governance.api.DemoService; |
| |
| import org.junit.Assert; |
| import org.junit.Before; |
| import org.junit.BeforeClass; |
| import org.junit.Test; |
| import org.junit.runner.RunWith; |
| import org.springframework.beans.factory.annotation.Autowired; |
| import org.springframework.test.context.ContextConfiguration; |
| import org.springframework.test.context.junit4.SpringJUnit4ClassRunner; |
| |
| |
| @RunWith(SpringJUnit4ClassRunner.class) |
| @ContextConfiguration(locations = {"classpath:/spring/dubbo-demo-consumer.xml"}) |
| public class DemoServiceInterfaceLevelIT { |
| @Autowired |
| private DemoService demoService; |
| |
| @BeforeClass |
| public static void setUp() throws Exception { |
| ZKTools.initClient(); |
| } |
| |
| private static boolean init = false; |
| |
| @Before |
| public void cleanAllRules() throws Exception { |
| if(!init) { |
| UpgradeUtil.writeForceInterfaceRule(); |
| checkIfNotified(); |
| init = true; |
| } |
| ZKTools.cleanup(); |
| } |
| |
| @Test(expected = RpcException.class) |
| public void testWithoutRule() throws Exception { |
| Thread.sleep(2000); |
| demoService.sayHello("world", 1200); |
| } |
| |
| @Test(expected = RpcException.class) |
| public void testWithServiceRuleWithInsufficientTimeout() throws Exception { |
| ZKTools.generateServiceLevelOverride(600); |
| Thread.sleep(2000); |
| demoService.sayHello("world", 800); |
| } |
| |
| @Test |
| public void testWithServiceRuleWithSufficientTimeout() throws Exception { |
| ZKTools.generateServiceLevelOverride(6000); |
| Thread.sleep(2000); |
| Assert.assertTrue(demoService.sayHello("world", 1200).startsWith("Hello world")); |
| } |
| |
| @Test(expected = RpcException.class) |
| public void testWithAppRuleWithInsufficientTimeout() throws Exception { |
| ZKTools.generateApplicationLevelOverride(600); |
| Thread.sleep(2000); |
| demoService.sayHello("world", 800); |
| } |
| |
| @Test |
| public void testWithAppRuleWithSufficientTimeout() throws Exception { |
| ZKTools.generateApplicationLevelOverride(6000); |
| Thread.sleep(2000); |
| Assert.assertTrue(demoService.sayHello("world", 1200).startsWith("Hello world")); |
| } |
| |
| private static void checkIfNotified() throws InterruptedException { |
| for (int i = 0; i < 50; i++) { |
| if (FrameworkStatusReporterImpl.getReport().size() == 1) { |
| FrameworkStatusReporterImpl.clearReport(); |
| return; |
| } |
| Thread.sleep(100); |
| } |
| } |
| } |