diff --git a/src/main/java/com/syncleus/ferma/framefactories/annotation/AbstractAnnotationFrameFactory.java b/src/main/java/com/syncleus/ferma/framefactories/annotation/AbstractAnnotationFrameFactory.java index 61071199232ed245f3f54a9b6fc3375d51893025..88bbaa5652385853d20599741864c4a3b40a1bd0 100644 --- a/src/main/java/com/syncleus/ferma/framefactories/annotation/AbstractAnnotationFrameFactory.java +++ b/src/main/java/com/syncleus/ferma/framefactories/annotation/AbstractAnnotationFrameFactory.java @@ -39,7 +39,7 @@ public class AbstractAnnotationFrameFactory implements FrameFactory { private final ReflectionCache reflectionCache; private final Map<Class, Class> constructedClassCache = new HashMap<>(); - public AbstractAnnotationFrameFactory(final ReflectionCache reflectionCache, Set<MethodHandler> handlers) { + protected AbstractAnnotationFrameFactory(final ReflectionCache reflectionCache, Set<MethodHandler> handlers) { this.reflectionCache = reflectionCache; for(MethodHandler handler : handlers) this.methodHandlers.put(handler.getAnnotationType(), handler); diff --git a/src/test/java/com/syncleus/ferma/framefactories/annotation/AnnotationFrameFactoryTest.java b/src/test/java/com/syncleus/ferma/framefactories/annotation/AnnotationFrameFactoryTest.java index 736899f5c11d18ae1254f989d7b340dbe83fd61a..8e3c0d4a76c86f381c0e50f162e544894c66c426 100644 --- a/src/test/java/com/syncleus/ferma/framefactories/annotation/AnnotationFrameFactoryTest.java +++ b/src/test/java/com/syncleus/ferma/framefactories/annotation/AnnotationFrameFactoryTest.java @@ -85,7 +85,8 @@ public class AnnotationFrameFactoryTest { .when(custom.processMethod(Mockito.any(), Mockito.any(), Mockito.any())) .thenAnswer(inv -> inv.getArgumentAt(0, DynamicType.Builder.class)) .getMock(); - AnnotationFrameFactory frameFactory = new AnnotationFrameFactory(new ReflectionCache(), Collections.singleton(custom)); + AbstractAnnotationFrameFactory frameFactory = new AbstractAnnotationFrameFactory(new ReflectionCache(), Collections.singleton(custom)) { + }; DelegatingFramedGraph framedGraph = new DelegatingFramedGraph(fg.getBaseGraph(), frameFactory, new PolymorphicTypeResolver()); framedGraph.addFramedVertex(God.class); Mockito.verify(custom, Mockito.atLeast(0)).getAnnotationType();