sample tests

spring-webflow-samples/booking-faces/src/test/java/org/springframework/webflow/samples/booking/BookingFlowExecutionTests.java

@@ -0,0 +1,92 @@
+package org.springframework.webflow.samples.booking;
+
+import org.easymock.EasyMock;
+import org.springframework.faces.model.converter.FacesConversionService;
+import org.springframework.webflow.config.FlowDefinitionResource;
+import org.springframework.webflow.config.FlowDefinitionResourceFactory;
+import org.springframework.webflow.core.collection.LocalAttributeMap;
+import org.springframework.webflow.core.collection.MutableAttributeMap;
+import org.springframework.webflow.test.MockExternalContext;
+import org.springframework.webflow.test.MockFlowBuilderContext;
+import org.springframework.webflow.test.execution.AbstractXmlFlowExecutionTests;
+
+public class BookingFlowExecutionTests extends AbstractXmlFlowExecutionTests {
+
+    private BookingService bookingService;
+
+    private boolean persistCalled;
+
+    protected void setUp() {
+	bookingService = EasyMock.createMock(BookingService.class);
+    }
+
+    @Override
+    protected FlowDefinitionResource getResource(FlowDefinitionResourceFactory resourceFactory) {
+	return resourceFactory.createFileResource("src/main/webapp/WEB-INF/flows/booking/booking.xml");
+    }
+
+    @Override
+    protected void configureFlowBuilderContext(MockFlowBuilderContext builderContext) {
+	builderContext.registerBean("bookingService", bookingService);
+	builderContext.getFlowBuilderServices().setConversionService(new FacesConversionService());
+    }
+
+    public void testStartBookingFlow() {
+	Booking booking = createTestBooking();
+
+	EasyMock.expect(bookingService.createBooking(1L, "keith")).andReturn(booking);
+
+	EasyMock.replay(bookingService);
+
+	MutableAttributeMap input = new LocalAttributeMap();
+	input.put("hotelId", "1");
+	MockExternalContext context = new MockExternalContext();
+	context.setCurrentUser("keith");
+	startFlow(input, context);
+
+	assertCurrentStateEquals("enterBookingDetails");
+	assertResponseWrittenEquals("enterBookingDetails", context);
+	assertTrue(getRequiredFlowAttribute("booking") instanceof Booking);
+
+	EasyMock.verify(bookingService);
+    }
+
+    public void testEnterBookingDetails_Proceed() {
+	setCurrentState("enterBookingDetails");
+	getFlowScope().put("booking", createTestBooking());
+
+	MockExternalContext context = new MockExternalContext();
+	context.setEventId("proceed");
+	resumeFlow(context);
+
+	assertCurrentStateEquals("reviewBooking");
+	assertResponseWrittenEquals("reviewBooking", context);
+    }
+
+    public void testReviewBooking_Confirm() {
+	setCurrentState("reviewBooking");
+	getFlowScope().put("booking", createTestBooking());
+	getFlowScope().put("entityManager", new Object() {
+	    public void persist(Booking booking) {
+		persistCalled = true;
+	    }
+	});
+	MockExternalContext context = new MockExternalContext();
+	context.setEventId("confirm");
+	resumeFlow(context);
+
+	assertTrue(persistCalled);
+	assertFlowExecutionEnded();
+	assertFlowExecutionOutcomeEquals("bookingConfirmed");
+    }
+
+    private Booking createTestBooking() {
+	Hotel hotel = new Hotel();
+	hotel.setId(1L);
+	hotel.setName("Jameson Inn");
+	User user = new User("keith", "pass", "Keith Donald");
+	Booking booking = new Booking(hotel, user);
+	return booking;
+    }
+
+}