Насмешка над свойством прокси-сервиса CGLIB не работает - PullRequest
Новая
       46

Насмешка над свойством прокси-сервиса CGLIB не работает

17 голосов
/ 27 января 2012

У меня возникла проблема при попытке смоделировать свойство службы из теста Junit: 

@ContextConfiguration("classpath:application-config.xml")
@RunWith(SpringJUnit4ClassRunner.class)
public class FooServiceTests {

    @Autowired
    private FooServiceImpl fooService;

    @Test
    public void testFoo() {
        String str = fooService.foo();
        assertEquals("Var", str);
    }

    @Before
    public void mockFooDao() throws Exception {
        FooDao mockFooDao = Mockito.mock(FooDao.class);
        Mockito.when(mockFooDao.foo()).thenReturn("Var");
        ReflectionTestUtils.setField(fooService, "fooDao", mockFooDao);
    }
}

Насмешка fooDao не имеет никакого эффекта, так как результат не является ожидаемым. Вот код службы и дао: 

@Service("fooService")
public class FooServiceImpl implements FooService {

    @Autowired
    protected FooDao fooDao;

    @Override
    public String foo() {
        return fooDao.foo();
    }
}

@Repository
public class FooDaoImpl implements FooDao {

    @Override
    public String foo() {
        return "foo";
    }
}

Как мы видим, реальный сервис предназначен для возврата «foo», но тест проверяет дао, поэтому сервис возвращает «var». Я знаю, что это связано с прокси-сервером CGLIB, но я не могу понять, как заставить его работать без использования установщика для свойства fooDao. Любая помощь будет оценена.

С уважением и заранее спасибо.

1 Ответ

39 голосов
/ 27 января 2012

Краткий ответ

Вы должны развернуть прокси и установить поле для целевого объекта: 

ReflectionTestUtils.setField(unwrapFooService(), "fooDao", mockFooDao);

unwrapFooService() можно определить следующим образом: 

private FooServiceImpl unwrapFooService() {
  if(AopUtils.isAopProxy(fooService) && fooService instanceof Advised) {
      Object target = ((Advised) fooService).getTargetSource().getTarget();
      return (FooServiceImpl)target;
  }
  return null;
}

... длинный

Проблема довольно сложная, но решаемая. Как вы уже догадались, это побочный эффект от использования прокси-серверов CGLIB. В принципе Spring создает подкласс вашего FooServiceImpl с именем, аналогичным FooServiceImpl$EnhancerByCGLIB. Этот подкласс содержит ссылку на исходный FooServiceImpl, а также ... все поля FooServiceImpl (что понятно - это подкласс).

Таким образом, на самом деле есть две переменные: FooServiceImpl$EnhancerByCGLIB.fooDao и FooServiceImpl.fooDao. Вы назначаете макет для первого, но ваш сервис использует последний ... Я написал об этой ловушке некоторое время назад.

...