Search in sources :

Example 1 with CountingAfterReturningAdvice

use of cn.taketoday.aop.CountingAfterReturningAdvice in project today-framework by TAKETODAY.

the class AbstractAopProxyTests method testAfterReturningAdvisorIsNotInvokedOnException.

@Test
public void testAfterReturningAdvisorIsNotInvokedOnException() {
    CountingAfterReturningAdvice car = new CountingAfterReturningAdvice();
    TestBean target = new TestBean();
    ProxyFactory pf = new ProxyFactory(target);
    pf.addAdvice(new NopInterceptor());
    pf.addAdvice(car);
    assertThat(pf.getAdvisors()[1].getAdvice()).as("Advice was wrapped in Advisor and added").isEqualTo(car);
    ITestBean proxied = (ITestBean) createProxy(pf);
    assertThat(car.getCalls()).isEqualTo(0);
    int age = 10;
    proxied.setAge(age);
    assertThat(proxied.getAge()).isEqualTo(age);
    assertThat(car.getCalls()).isEqualTo(2);
    Exception exc = new Exception();
    // On exception it won't be invoked
    assertThatExceptionOfType(Throwable.class).isThrownBy(() -> proxied.exceptional(exc)).satisfies(ex -> assertThat(ex).isSameAs(exc));
    assertThat(car.getCalls()).isEqualTo(2);
}
Also used : CountingAfterReturningAdvice(cn.taketoday.aop.CountingAfterReturningAdvice) ITestBean(cn.taketoday.beans.testfixture.beans.ITestBean) NopInterceptor(cn.taketoday.aop.NopInterceptor) SerializableNopInterceptor(cn.taketoday.aop.SerializableNopInterceptor) ITestBean(cn.taketoday.beans.testfixture.beans.ITestBean) TestBean(cn.taketoday.beans.testfixture.beans.TestBean) ProxyFactory(cn.taketoday.aop.framework.ProxyFactory) AopConfigException(cn.taketoday.aop.framework.AopConfigException) LockedException(cn.taketoday.aop.mixin.LockedException) FileNotFoundException(java.io.FileNotFoundException) SQLException(java.sql.SQLException) MarshalException(java.rmi.MarshalException) Assertions.assertThatIllegalStateException(org.assertj.core.api.Assertions.assertThatIllegalStateException) IOException(java.io.IOException) UndeclaredThrowableException(java.lang.reflect.UndeclaredThrowableException) Assertions.assertThatIllegalArgumentException(org.assertj.core.api.Assertions.assertThatIllegalArgumentException) Test(org.junit.jupiter.api.Test)

Example 2 with CountingAfterReturningAdvice

use of cn.taketoday.aop.CountingAfterReturningAdvice in project today-framework by TAKETODAY.

the class AbstractAopProxyTests method testSerializableTargetAndAdvice.

@Test
public void testSerializableTargetAndAdvice() throws Throwable {
    SerializablePerson personTarget = new SerializablePerson();
    personTarget.setName("jim");
    personTarget.setAge(26);
    assertThat(SerializationTestUtils.isSerializable(personTarget)).isTrue();
    ProxyFactory pf = new ProxyFactory(personTarget);
    CountingThrowsAdvice cta = new CountingThrowsAdvice();
    pf.addAdvice(new SerializableNopInterceptor());
    // Try various advice types
    pf.addAdvice(new CountingBeforeAdvice());
    pf.addAdvice(new CountingAfterReturningAdvice());
    pf.addAdvice(cta);
    Person p = (Person) createAopProxy(pf).getProxy();
    p.echo(null);
    assertThat(cta.getCalls()).isEqualTo(0);
    try {
        p.echo(new IOException());
    } catch (IOException ex) {
    /* expected */
    }
    assertThat(cta.getCalls()).isEqualTo(1);
    // Will throw exception if it fails
    Person p2 = SerializationTestUtils.serializeAndDeserialize(p);
    assertThat(p2).isNotSameAs(p);
    assertThat(p2.getName()).isEqualTo(p.getName());
    assertThat(p2.getAge()).isEqualTo(p.getAge());
    assertThat(AopUtils.isAopProxy(p2)).as("Deserialized object is an AOP proxy").isTrue();
    Advised a1 = (Advised) p;
    Advised a2 = (Advised) p2;
    // Check we can manipulate state of p2
    assertThat(a2.getAdvisors().length).isEqualTo(a1.getAdvisors().length);
    // This should work as SerializablePerson is equal
    assertThat(p2).as("Proxies should be equal, even after one was serialized").isEqualTo(p);
    assertThat(p).as("Proxies should be equal, even after one was serialized").isEqualTo(p2);
    // Check we can add a new advisor to the target
    NopInterceptor ni = new NopInterceptor();
    p2.getAge();
    assertThat(ni.getCount()).isEqualTo(0);
    a2.addAdvice(ni);
    p2.getAge();
    assertThat(ni.getCount()).isEqualTo(1);
    cta = (CountingThrowsAdvice) a2.getAdvisors()[3].getAdvice();
    p2.echo(null);
    assertThat(cta.getCalls()).isEqualTo(1);
    try {
        p2.echo(new IOException());
    } catch (IOException ignored) {
    }
    assertThat(cta.getCalls()).isEqualTo(2);
}
Also used : CountingAfterReturningAdvice(cn.taketoday.aop.CountingAfterReturningAdvice) NopInterceptor(cn.taketoday.aop.NopInterceptor) SerializableNopInterceptor(cn.taketoday.aop.SerializableNopInterceptor) SerializableNopInterceptor(cn.taketoday.aop.SerializableNopInterceptor) SerializablePerson(cn.taketoday.aop.support.NameMatchMethodPointcutTests.SerializablePerson) ProxyFactory(cn.taketoday.aop.framework.ProxyFactory) Advised(cn.taketoday.aop.framework.Advised) IOException(java.io.IOException) Person(cn.taketoday.aop.support.NameMatchMethodPointcutTests.Person) SerializablePerson(cn.taketoday.aop.support.NameMatchMethodPointcutTests.SerializablePerson) Test(org.junit.jupiter.api.Test)

Aggregations

CountingAfterReturningAdvice (cn.taketoday.aop.CountingAfterReturningAdvice)2 NopInterceptor (cn.taketoday.aop.NopInterceptor)2 SerializableNopInterceptor (cn.taketoday.aop.SerializableNopInterceptor)2 ProxyFactory (cn.taketoday.aop.framework.ProxyFactory)2 IOException (java.io.IOException)2 Test (org.junit.jupiter.api.Test)2 Advised (cn.taketoday.aop.framework.Advised)1 AopConfigException (cn.taketoday.aop.framework.AopConfigException)1 LockedException (cn.taketoday.aop.mixin.LockedException)1 Person (cn.taketoday.aop.support.NameMatchMethodPointcutTests.Person)1 SerializablePerson (cn.taketoday.aop.support.NameMatchMethodPointcutTests.SerializablePerson)1 ITestBean (cn.taketoday.beans.testfixture.beans.ITestBean)1 TestBean (cn.taketoday.beans.testfixture.beans.TestBean)1 FileNotFoundException (java.io.FileNotFoundException)1 UndeclaredThrowableException (java.lang.reflect.UndeclaredThrowableException)1 MarshalException (java.rmi.MarshalException)1 SQLException (java.sql.SQLException)1 Assertions.assertThatIllegalArgumentException (org.assertj.core.api.Assertions.assertThatIllegalArgumentException)1 Assertions.assertThatIllegalStateException (org.assertj.core.api.Assertions.assertThatIllegalStateException)1