模拟 CGLIB 代理服务的属性不起作用

Question

我在尝试从 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 没有效果，因为结果不是预期的。下面是服务和 dao 的代码：

@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”，但测试模拟了 dao，因此服务返回“var”。我知道这是一个与 CGLIB 代理相关的东西，但我不知道如何在不使用 fooDao 属性的设置器的情况下使其工作。任何帮助将不胜感激。

提前致谢。

Answer 1

简答

您必须unwrap the proxy 并在目标对象上设置字段：

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。您正在为前者分配一个模拟，但您的服务使用后者...我 wrote 不久前谈到了这个陷阱。