Apache Camel的單元測試

<dependency>
            <groupId>org.apache.camel</groupId>
            <artifactId>camel-test-spring-junit5</artifactId>
            <version>${camel.version}</version>
            <scope>test</scope>
</dependency>

String fileInbound = UriComponentsBuilder.newInstance().scheme("file")
                .path(Paths.get(inboundPath).toAbsolutePath().toString())
                .queryParam("include", "^.*\\.(xlsx|csv)$")
                .build().toString();

        from(fileInbound).id("SAMPLE_FILE_ROUTE")
                .log("Received file: ${header.CamelFileName}")
                .choice()
                .when((exchange) -> exchange.getIn().getHeader("CamelFileName").toString()
                        .endsWith("csv")).setHeader("status", simple("SUCCESS")).log("Hurray !! Its a CSV !!")
                .otherwise().setHeader("status", simple("FAILURE")).log("Oh No!! Its an XLSX..")
                .end();

//Extend the class with CamelTestSupport which Camel Test provides
class SampleFileRouteTest extends CamelTestSupport {
  
  //Override the createRouteBuilder() and return an object of the actual route class under test
  @Override
    protected RoutesBuilder createRouteBuilder() throws Exception {
        return new SampleFileRoute("src/test/resources");
    }
  
  //Set-Up the mock route to listenForCompletion response once the route is triggered
  @Override
    protected void doPostSetup() throws Exception {
        AdviceWith.adviceWith(context, "SAMPLE_FILE_ROUTE",
                a -> a.onCompletion().onCompleteOnly().to("mock:listenForComplete"));
    }
  
  //And Finally the test cases for all 3 scenarios, namely Success/Fail and Invalid File
  @Test
    void testForSuccessValidFile() throws IOException, InterruptedException {
        inboundFile = new File(inboundPath.toString(), "test.csv");
        inboundFile.createNewFile();
        listenForComplete.expectedHeaderReceived("status", "SUCCESS");
        listenForComplete.assertIsSatisfied();
    }

    @Test
    void testForFailValidFile() throws IOException, InterruptedException {
        inboundFile = new File(inboundPath.toString(), "test.xlsx");
        inboundFile.createNewFile();
        listenForComplete.expectedHeaderReceived("status", "FAILURE");
        listenForComplete.assertIsSatisfied();
    }

    @Test
    void testForInvalidFile() throws IOException, InterruptedException {
        inboundFile = new File(inboundPath.toString(), "test.txt");
        inboundFile.createNewFile();
        listenForComplete.expectedHeaderReceived("status","SUCCESS");
        listenForComplete.assertIsNotSatisfied();
    }
  
}
 

//Prepare the respective inbound and outbound URLS
String fileInbound = UriComponentsBuilder.newInstance().scheme("file")
        .path(inboundPath)
        .build().toString();

String outboundLocation = UriComponentsBuilder.newInstance()
        .scheme("sftp").host(sftpInfo.getHost()).port(sftpInfo.getPort()).userInfo(sftpInfo.getUser())
        .path(sftpInfo.getPath()).queryParam("preferredAuthentications", "password")
        .queryParam("password", sftpInfo.getPassword())
        .queryParam("jschLoggingLevel", "WARN")
        .build().toString();


//Use Camel-FTP Support and effect the transfer.. namely pick from the inbound and send to output
//and observe success/failures
from(fileInbound).id("SAMPLE_SFTP_ROUTE")
        .log("Starting SFTP Transfer for File: ${header.CamelFileName}")
        .to(outboundLocation)
        .log("SFTP Transfer Success for File: ${header.CamelFileName}");

onException(Exception.class)
        .maximumRedeliveries(2)
        .redeliveryDelay(20)
        .log("SFTP Transfer Failure for File: ${header.CamelFileName}")
        .handled(true);

//Extend the test class with Test Support Camel provides
class SampleSftpRouteTest extends CamelTestSupport {
  
    //Start the in-memory SFTP Server using Fake-Sftp-Server-Lambda and make it listen
    @Override
    @BeforeEach
    public void setUp() throws Exception {
        withSftpServer(server -> {
            server.addUser("tst", "tst");
            sftpInfo = SftpInfo.builder().host("0.0.0.0").port(server.getPort()).path("/").build();
        });
        super.setUp();
    }
  
    
    //Test Both the Success and Fail Scenarios.
    //For Success - Test if the file is transfered successful upon created in I/P Path
    //For Failure - Test if the route throws exception upon start and the file transfer does'nt happen
    @Test
    void testsftpSuccess() throws Exception {
        inboundFile = new File(inboundPath.toString(), "test.txt");
        inboundFile.createNewFile();
        await().atMost(10, TimeUnit.SECONDS);
        withSftpServer(server -> assertTrue(server.existsFile("/test.txt")));
        assertFalse(inboundFile.exists());
    }

    @Test
    void testsftpFailure() throws Exception {
        sftpInfo.setUser("user");
        sftpInfo.setPassword("wrong");
        inboundFile = new File(inboundPath.toString(), "test.txt");
        inboundFile.createNewFile();
        await().atMost(10, TimeUnit.SECONDS);
        withSftpServer(server -> assertFalse(server.existsFile("/test.txt")));
        assertTrue(inboundFile.exists());
    }

}
 

//Build the route as below. It all starts from India and proceeds to Delhi, Mumbai, Kolkata and Chennai
//In each and every stop, it goes into individual logics and has it executed.
//The block of individual logic for Delhi is also given below.
  from("direct:India").id("SAMPLE_MULTIHOP_ROUTE")
                .log("Welcome to India !!")
                .to("direct:Delhi")
                .to("direct:Mumbai")
                .to("direct:Kolkata")
                .to("direct:Chennai");

  from("direct:Delhi")
          .bean(Delhi.class);

  from("direct:Mumbai")
          .bean(Mumbai.class);

  from("direct:Kolkata")
          .bean(Kolkata.class);

  from("direct:Chennai")
          .bean(Chennai.class);

//Execution Logic inside Delhi.class
@Component
public class Delhi {

    @Handler
    public void handler(@RequestBody String name) {
        System.out.println("Hello Mr " + name + "!!! Welcome to Delhi !!");
    }
}

@ExtendWith(MockitoExtension.class)
class SampleMultiHopRouteTest extends CamelTestSupport {

    @InjectMocks
    private SampleMultiHopRoute sampleMultiHopRoute;

    @Mock
    private Delhi delhi;

    @Mock
    private Mumbai mumbai;

    @Mock
    private Kolkata kolkata;

    @Mock
    private Chennai chennai;

    @Override
    protected RoutesBuilder createRouteBuilder() throws Exception {
        Registry registry = context.getRegistry();
        registry.bind("delhi", Delhi.class, delhi);
        registry.bind("mumbai", Mumbai.class, mumbai);
        registry.bind("kolkata", Kolkata.class, kolkata);
        registry.bind("chennai", Chennai.class, chennai);
        return sampleMultiHopRoute;
    }

    @Test
    void testMultiHopRoute() {
        doNothing().when(delhi).handler("Junit");
        doNothing().when(mumbai).handler("Junit");
        doNothing().when(kolkata).handler("Junit");
        doNothing().when(chennai).handler("Junit");
        template.sendBody("direct:India", "Junit");
        verify(delhi).handler("Junit");
        verify(mumbai).handler("Junit");
        verify(kolkata).handler("Junit");
        verify(chennai).handler("Junit");
    }
}