better test for checking that spring workflows are detected

nflow-engine/src/test/java/com/nitorcreations/nflow/engine/service/SpringDummyTestWorkflow.java

@@ -3,11 +3,14 @@
 import static com.nitorcreations.nflow.engine.workflow.definition.NextAction.moveToState;
 import static com.nitorcreations.nflow.engine.workflow.definition.NextAction.stopInState;
 
+import org.springframework.stereotype.Component;
+
 import com.nitorcreations.nflow.engine.workflow.definition.NextAction;
 import com.nitorcreations.nflow.engine.workflow.definition.StateExecution;
 import com.nitorcreations.nflow.engine.workflow.definition.WorkflowDefinition;
 import com.nitorcreations.nflow.engine.workflow.definition.WorkflowStateType;
 
+@Component
 public class SpringDummyTestWorkflow extends WorkflowDefinition<SpringDummyTestWorkflow.SpringDummyTestState> {
 
   public static enum SpringDummyTestState implements com.nitorcreations.nflow.engine.workflow.definition.WorkflowState {

nflow-engine/src/test/java/com/nitorcreations/nflow/engine/service/WorkflowDefinitionServiceTest.java

@@ -8,7 +8,6 @@
 import static org.mockito.Mockito.when;
 
 import java.io.ByteArrayInputStream;
-import java.util.ArrayList;
 import java.util.List;
 
 import org.junit.Before;
@@ -54,15 +53,6 @@ public void initDuplicateWorkflows() throws Exception {
     service.initNonSpringWorkflowDefinitions();
   }
 
-  @Test
-  public void springWorkflowsWork() {
-    List<WorkflowDefinition<? extends WorkflowState>> definitions = service.getWorkflowDefinitions();
-    List<WorkflowDefinition<? extends WorkflowState>> list = new ArrayList<>();
-    list.add(new SpringDummyTestWorkflow());
-    service.setWorkflowDefinitions(list);
-    assertThat(definitions.size(), is(equalTo(1)));
-  }
-
   @Test
   public void demoWorkflowLoadedSuccessfully() {
     List<WorkflowDefinition<? extends WorkflowState>> definitions = service.getWorkflowDefinitions();

nflow-engine/src/test/java/com/nitorcreations/nflow/engine/service/WorkflowDefinitionServiceWithSpringTest.java

@@ -0,0 +1,67 @@
+package com.nitorcreations.nflow.engine.service;
+
+import static org.hamcrest.Matchers.equalTo;
+import static org.hamcrest.Matchers.is;
+import static org.junit.Assert.assertThat;
+import static org.mockito.Mockito.mock;
+
+import java.util.List;
+
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.ComponentScan;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.core.io.AbstractResource;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.support.AnnotationConfigContextLoader;
+
+import com.nitorcreations.nflow.engine.internal.config.NFlow;
+import com.nitorcreations.nflow.engine.internal.dao.ExecutorDao;
+import com.nitorcreations.nflow.engine.internal.dao.StatisticsDao;
+import com.nitorcreations.nflow.engine.internal.dao.WorkflowInstanceDao;
+import com.nitorcreations.nflow.engine.workflow.definition.WorkflowDefinition;
+import com.nitorcreations.nflow.engine.workflow.definition.WorkflowState;
+
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(loader = AnnotationConfigContextLoader.class)
+public class WorkflowDefinitionServiceWithSpringTest {
+
+  @Configuration
+  @ComponentScan(basePackageClasses = SpringDummyTestWorkflow.class)
+  static class ContextConfiguration {
+
+    @Bean
+    @NFlow
+    public AbstractResource nflowNonSpringWorkflowsListing() {
+      return null;
+    }
+
+    @Bean
+    public WorkflowInstanceDao workflowInstanceDao() {
+      return mock(WorkflowInstanceDao.class);
+    }
+
+    @Bean
+    public ExecutorDao executorDao() {
+      return mock(ExecutorDao.class);
+    }
+
+    @Bean
+    public StatisticsDao statisticsDao() {
+      return mock(StatisticsDao.class);
+    }
+  }
+
+  @Autowired
+  private WorkflowDefinitionService service;
+
+  @Test
+  public void springWorkflowDefinitionsAreDetected() {
+    List<WorkflowDefinition<? extends WorkflowState>> definitions = service.getWorkflowDefinitions();
+    assertThat(definitions.size(), is(equalTo(1)));
+    assertThat(definitions.get(0).getType(), is(new SpringDummyTestWorkflow().getType()));
+  }
+}