Fix AbstractChannelScannerTest

springwolf · Dec 17, 2021 · 54d3b1f · 54d3b1f
1 parent cf45a63
commit 54d3b1f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/...o/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java b/...o/github/stavshamir/springwolf/asyncapi/scanners/channels/AbstractChannelScannerTest.java
@@ -7,6 +7,7 @@
 import io.github.stavshamir.springwolf.asyncapi.scanners.components.ComponentsScanner;
 import io.github.stavshamir.springwolf.asyncapi.types.channel.operation.message.Message;
 import io.github.stavshamir.springwolf.asyncapi.types.channel.operation.message.PayloadReference;
+import io.github.stavshamir.springwolf.configuration.AsyncApiDocket;
 import io.github.stavshamir.springwolf.schemas.DefaultSchemasService;
 import lombok.Data;
 import lombok.NoArgsConstructor;
@@ -39,8 +40,12 @@ public class AbstractChannelScannerTest {
     @MockBean
     private ComponentsScanner componentsScanner;
 
+    @MockBean
+    private AsyncApiDocket docket;
+
     private void setClassToScan(Class<?> classToScan) {
         Set<Class<?>> classesToScan = singleton(classToScan);
+        when(docket.getBasePackage()).thenReturn("io.github.stavshamir.springwolf.asyncapi.scanners.channels");
         when(componentsScanner.scanForComponents(anyString())).thenReturn(classesToScan);
     }
 

diff --git a/...t/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java b/...t/java/io/github/stavshamir/springwolf/asyncapi/scanners/channels/TestChannelScanner.java
@@ -9,11 +9,6 @@
 
 public class TestChannelScanner extends AbstractChannelScanner<AbstractChannelScannerTest.TestChannelListener> {
 
-    @Override
-    protected String getBasePackage() {
-        return "io.github.stavshamir.springwolf.asyncapi.scanners.channels";
-    }
-
     @Override
     protected Class<AbstractChannelScannerTest.TestChannelListener> getListenerAnnotationClass() {
         return AbstractChannelScannerTest.TestChannelListener.class;