Skip to content

Commit

Permalink
Add unit test for ComposeClassStaticMethodAroundAdvice and ComposeCon…
Browse files Browse the repository at this point in the history
…structorAdvice (Issue #10958) (#13958)

* Add files via upload

* Rename shardingsphere-agent/shardingsphere-agent-core/src/test/java/org/apache/shardingsphere/agent/core/bytebuddy/transformer/ComposeClassStaticMethodAroundAdviceTest.java to shardingsphere-agent/shardingsphere-agent-core/src/test/java/org/apache/shardingsphere/agent/core/bytebuddy/transformer/advice/ComposeClassStaticMethodAroundAdviceTest.java

* Rename shardingsphere-agent/shardingsphere-agent-core/src/test/java/org/apache/shardingsphere/agent/core/bytebuddy/transformer/ComposeConstructorAdviceTest.java to shardingsphere-agent/shardingsphere-agent-core/src/test/java/org/apache/shardingsphere/agent/core/bytebuddy/transformer/advice/ComposeConstructorAdviceTest.java

* Update ComposeClassStaticMethodAroundAdviceTest.java

* Update ComposeConstructorAdviceTest.java

* Update ComposeClassStaticMethodAroundAdviceTest.java

* Update ComposeConstructorAdviceTest.java
  • Loading branch information
franco-h authored Dec 20, 2021
1 parent 2ca6beb commit fea1c95
Show file tree
Hide file tree
Showing 2 changed files with 111 additions and 0 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,66 @@
/*
* 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.shardingsphere.agent.core.bytebuddy.transformer.advice;

import org.apache.shardingsphere.agent.api.advice.ClassStaticMethodAroundAdvice;
import org.apache.shardingsphere.agent.api.result.MethodInvocationResult;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.junit.MockitoJUnitRunner;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import static org.mockito.Mockito.mock;

@RunWith(MockitoJUnitRunner.class)
public class ComposeClassStaticMethodAroundAdviceTest {

private ComposeClassStaticMethodAroundAdvice composeClassStaticMethodAroundAdvice;

@Test
public void beforeMethodTest() {
ClassStaticMethodAroundAdvice advice = mock(ClassStaticMethodAroundAdvice.class);
Method method = mock(Method.class);
MethodInvocationResult methodInvocationResult = mock(MethodInvocationResult.class);
List<ClassStaticMethodAroundAdvice> adviceList = new ArrayList<>(Arrays.asList(advice));
composeClassStaticMethodAroundAdvice = new ComposeClassStaticMethodAroundAdvice(adviceList);
composeClassStaticMethodAroundAdvice.beforeMethod(String.class, method, new Object[2], methodInvocationResult);
}

@Test
public void afterMethodTest() {
ClassStaticMethodAroundAdvice advice = mock(ClassStaticMethodAroundAdvice.class);
Method method = mock(Method.class);
MethodInvocationResult methodInvocationResult = mock(MethodInvocationResult.class);
List<ClassStaticMethodAroundAdvice> adviceList = new ArrayList<>(Arrays.asList(advice));
composeClassStaticMethodAroundAdvice = new ComposeClassStaticMethodAroundAdvice(adviceList);
composeClassStaticMethodAroundAdvice.afterMethod(String.class, method, new Object[2], methodInvocationResult);
}

@Test
public void onThrowingTest() {
ClassStaticMethodAroundAdvice advice = mock(ClassStaticMethodAroundAdvice.class);
Method method = mock(Method.class);
List<ClassStaticMethodAroundAdvice> adviceList = new ArrayList<>(Arrays.asList(advice));
composeClassStaticMethodAroundAdvice = new ComposeClassStaticMethodAroundAdvice(adviceList);
composeClassStaticMethodAroundAdvice.onThrowing(String.class, method, new Object[2], new NullPointerException("Null Pointer"));
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,45 @@
/*
* 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.shardingsphere.agent.core.bytebuddy.transformer.advice;

import org.apache.shardingsphere.agent.api.advice.AdviceTargetObject;
import org.apache.shardingsphere.agent.api.advice.ConstructorAdvice;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.junit.MockitoJUnitRunner;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

import static org.mockito.Mockito.mock;

@RunWith(MockitoJUnitRunner.class)
public class ComposeConstructorAdviceTest {

private ComposeConstructorAdvice composeConstructorAdvice;

@Test
public void onConstructorTest() {
ConstructorAdvice advice = mock(ConstructorAdvice.class);
AdviceTargetObject adviceTargetObject = mock(AdviceTargetObject.class);
List<ConstructorAdvice> adviceList = new ArrayList<>(Arrays.asList(advice));
composeConstructorAdvice = new ComposeConstructorAdvice(adviceList);
composeConstructorAdvice.onConstructor(adviceTargetObject, new Object[2]);
}
}

0 comments on commit fea1c95

Please sign in to comment.