Транзакционная аннотация позволяет избежать издевательств

У меня есть файл правил drools, который использует классы обслуживания в правилах. Итак, одно правило делает что-то вроде этого:

eval (countryService.getCountryById(1)!= null)

В службе валидации, которая аннотируется с помощью @service и @Transactional (распространение = распространение .SUPPORTS), файл drools используется в statelessKnowledgebase и добавляются факты, которые должны использоваться в слюнотечении. После этого вызывается session.execute(факсы) и запускается правило.

Чтобы проверить правила, я хотел бы заглушить countryService.getCountryById(). Нет большой проблемы с использованием mockito. Сделано это для другой службы, которая также использует настройку слюни, и она отлично работала. Однако в данном конкретном случае countryService не был заглушен, и я не мог понять, почему. Проведя много времени и проверив мой код, я обнаружил, что использование @Transactional над сервисом или отсутствие этой аннотации внесли разницу. Отсутствие @Transaction, сделанное mockito, искалечило систему countryservice без каких-либо проблем, имея @transactional на месте, из-за которого mockito терпит неудачу (без какой-либо ошибки или намека), впрыскивая макет, чтобы использовать исходный объект countryservice.

Мой вопрос в том, почему эта аннотация вызывает эту проблему. Почему при запуске @Transactional вы не можете вводить макеты в mockito? Я заметил, что mockito терпит неудачу, когда я отлаживаю и проверяю countryService, когда он добавляется как глобальный к сеансу спуска. Я вижу следующую разницу, когда я проверяю countryservice в своем debugwindow:

  • с @transactional: countryService имеет значение CountryService $$ EnhancerByCGLIB $$ b80dbb7b

  • без @transactional: countryService имеет значение CountryService $$ EnhancerByMockitoWithCGLIB $$ 27f34dc1

В дополнение к @transactional моя точка останова в методе countryservice getCountryById найдена, и отладчик останавливается в этой точке останова, но без @transactional моя точка останова пропускается, поскольку mockito обходит ее.

ValidationService:

@Service
@Transactional(propagation=Propagation.SUPPORTS)
public class ValidationService 
{
  @Autowired
  private CountryService countryService;

  public void validateFields(Collection<Object> facts)
  {
    KnowledgeBase knowledgeBase = (KnowledgeBase)AppContext.getApplicationContext().getBean(knowledgeBaseName); 
    StatelessKnowledgeSession session = knowledgeBase.newStatelessKnowledgeSession();
    session.setGlobal("countryService", countryService);
    session.execute(facts);

  }

И тестовый класс:

public class TestForeignAddressPostalCode extends BaseTestDomainIntegration
{

  private final Collection<Object> postalCodeMinLength0 = new ArrayList<Object>();

  @Mock
  protected CountryService countryService;

  @InjectMocks
  private ValidationService level2ValidationService;


  @BeforeMethod(alwaysRun=true)
  protected void setup()
  {
    // Get the object under test (here the determination engine)
    level2ValidationService = (ValidationService) getAppContext().getBean("validationService");
    // and replace the services as documented above.
    MockitoAnnotations.initMocks(this);

    ForeignAddress foreignAddress = new ForeignAddress();
    foreignAddress.setCountryCode("7029");
    foreignAddress.setForeignPostalCode("foreign");

    // mock country to be able to return a fixed id
    Country country = mock(Country.class);
    foreignAddress.setLand(country);
    doReturn(Integer.valueOf(1)).when(country).getId();

    doReturn(country).when(countryService).getCountryById(anyInt());

    ContextualAddressBean context = new ContextualAddressBean(foreignAddress, "", AddressContext.CORRESPONDENCE_ADDRESS);
    postalCodeMinLength0.add(context);
  }

  @Test
  public void PostalCodeMinLength0_ExpectError()
  {
    // Execute
    level2ValidationService.validateFields(postalCodeMinLength0, null);

  }

Любая идея, что делать, если я хочу сохранить эту аннотацию @transactional, но также смогу заглушить методы countryservice?

С уважением,

Майкл

Ответ 1

Случается, что ваш ValidationService завернут в JdkDynamicAopProxy, поэтому, когда Mockito идет, чтобы внедрить насмешки в службу, он не видит никаких полей для их ввода. Вам нужно сделать одну из двух вещей:

  • Перед началом работы с Spring контекстом приложения и проверьте только Validation Service, заставляя вас высмеивать каждую зависимость.
  • Или разверните свою реализацию из JdkDynamicAopProxy и обработайте сами инъекции.

Пример кода:

@Before
public void setup() throws Exception {
    MockitoAnnotations.initMocks(this);
    ValidationService validationService = (ValidationService) unwrapProxy(level2ValidationService);
    ReflectionTestUtils.setField(validationService, "countryService", countryService);
}

public static final Object unwrapProxy(Object bean) throws Exception {
    /*
     * If the given object is a proxy, set the return value as the object
     * being proxied, otherwise return the given object.
     */
    if (AopUtils.isAopProxy(bean) && bean instanceof Advised) {
        Advised advised = (Advised) bean;
        bean = advised.getTargetSource().getTarget();
    }
    return bean;
}

Запись в блоге

Ответ 2

Обратите внимание, что с Spring 4.3.1 ReflectionTestUtils следует автоматически развернуть прокси. Так

ReflectionTestUtils.setField(validationService, "countryService", countryService);

теперь должен работать, даже если ваш countryService аннотируется с помощью @Transactional, @Cacheable... (т.е. скрывается за прокси во время выполнения)

Связанная проблема: SPR-14050

Ответ 3

Альтернативным решением является добавление макетного объекта в контекст Spring до того, как Spring проложит все вместе, чтобы он уже был введен до начала тестов. Модифицированный тест может выглядеть примерно так:

@RunWith(SpringJUnit4ClassRunner.class)
@ContextConfiguration(classes = { Application.class, MockConfiguration.class })
public class TestForeignAddressPostalCode extends BaseTestDomainIntegration
{

  public static class MockConfiguration {

      @Bean
      @Primary
      public CountryService mockCountryService() {
        return mock(CountryService.class);
      }

  }

  @Autowired
  protected CountryService mockCountryService;

  @Autowired
  private ValidationService level2ValidationService;

  @BeforeMethod(alwaysRun=true)
  protected void setup()
  {

    // set up you mock stubs here
    // ...

Аннотация @Primary важна, чтобы ваш новый макет CountryService имел приоритет для инъекций, заменив обычный. Это может иметь непреднамеренные побочные эффекты, однако, если класс вводится в несколько мест.

Ответ 4

На основе ответа SuperSaiyen я создал класс утилиты для удаления, чтобы сделать его более простым и безопасным по типу:

import org.mockito.Mockito;
import org.springframework.aop.framework.Advised;
import org.springframework.aop.support.AopUtils;
import org.springframework.test.util.ReflectionTestUtils;

@SuppressWarnings("unchecked")
public class SpringBeanMockUtil {
  /**
   * If the given object is a proxy, set the return value as the object being proxied, otherwise return the given
   * object.
   */
  private static <T> T unwrapProxy(T bean) {
    try {
      if (AopUtils.isAopProxy(bean) && bean instanceof Advised) {
        Advised advised = (Advised) bean;
        bean = (T) advised.getTargetSource().getTarget();
      }
      return bean;
    }
    catch (Exception e) {
      throw new RuntimeException("Could not unwrap proxy!", e);
    }
  }

  public static <T> T mockFieldOnBean(Object beanToInjectMock, Class<T> classToMock) {
    T mocked = Mockito.mock(classToMock);
    ReflectionTestUtils.setField(unwrapProxy(beanToInjectMock), null, mocked, classToMock);
    return mocked;
  }
}

Использование простое, только в начале вашего тестового метода, вызовите метод mockFieldOnBean(Object beanToInjectMock, Class<T> classToMock) с помощью bean, который вы хотите ввести в макет, и класс объекта, который следует издеваться. Пример:

Скажем, у вас есть bean с типом SomeService, который содержит autwired bean of SomeOtherService, что-то вроде:

@Component
public class SomeService {
  @Autowired
  private SomeOtherService someOtherService;

  // some other stuff
}

Чтобы высмеять SomeOtherService на SomeService bean, используйте следующее:

@RunWith(SpringJUnit4ClassRunner.class)
public class TestClass {

  @Autowired
  private SomeService someService;

  @Test
  public void sampleTest() throws Exception {
    SomeOtherService someOtherServiceMock = SpringBeanMockUtil.mockFieldOnBean(someService, SomeOtherService.class);

    doNothing().when(someOtherServiceMock).someMethod();

    // some test method(s)

    verify(someOtherServiceMock).someMethod();
  }
}

все должно работать так, как должно.