return asMocks ? Mockito.spy(registry) : registry;
}
@SuppressWarnings({ "unchecked", "rawtypes" })
private void spyOnDefs(final MasterDefRegistryImpl registry) throws QuickFixException {
final MockUtil mockUtil = new MockUtil();
for (DefRegistry<?> subReg : registry.getAllRegistries()) {
Mockito.doAnswer(new Answer<Definition>() {
@Override
public Definition answer(InvocationOnMock invocation) throws Throwable {
Definition ret = (Definition) invocation.callRealMethod();
if (ret == null) {
return ret;
}
if (mockUtil.isMock(ret)) {
return ret;
} else {
ret = Mockito.spy(ret);
registry.addLocalDef(ret);
return ret;