Fixed HandlingRepositoryTest

src/test/java/se/citerus/dddsample/infrastructure/persistence/hibernate/HandlingEventRepositoryTest.java

@@ -1,5 +1,19 @@
 package se.citerus.dddsample.infrastructure.persistence.hibernate;
 
+import org.hibernate.SessionFactory;
+import org.hibernate.classic.Session;
+import org.junit.Before;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.jdbc.core.JdbcTemplate;
+import org.springframework.orm.hibernate3.HibernateTransactionManager;
+import org.springframework.test.context.ContextConfiguration;
+import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
+import org.springframework.test.context.transaction.TransactionConfiguration;
+import org.springframework.transaction.annotation.Transactional;
+import org.springframework.transaction.support.TransactionTemplate;
+import se.citerus.dddsample.application.util.SampleDataGenerator;
 import se.citerus.dddsample.domain.model.cargo.Cargo;
 import se.citerus.dddsample.domain.model.cargo.CargoRepository;
 import se.citerus.dddsample.domain.model.cargo.TrackingId;
@@ -9,52 +23,91 @@ import se.citerus.dddsample.domain.model.location.Location;
 import se.citerus.dddsample.domain.model.location.LocationRepository;
 import se.citerus.dddsample.domain.model.location.UnLocode;
 
+import javax.sql.DataSource;
+import java.lang.reflect.Field;
 import java.util.Date;
 import java.util.List;
 import java.util.Map;
 
-/*public class HandlingEventRepositoryTest extends AbstractRepositoryTest {
+import static org.junit.Assert.assertEquals;
 
-  HandlingEventRepository handlingEventRepository;
-  CargoRepository cargoRepository;
-  LocationRepository locationRepository;
+@RunWith(SpringJUnit4ClassRunner.class)
+@ContextConfiguration(value = {"/context-infrastructure-persistence.xml", "/context-domain.xml"})
+@TransactionConfiguration(transactionManager = "transactionManager")
+@Transactional
+public class HandlingEventRepositoryTest {
 
-  public void setHandlingEventRepository(HandlingEventRepository handlingEventRepository) {
-    this.handlingEventRepository = handlingEventRepository;
-  }
+    @Autowired
+    HandlingEventRepository handlingEventRepository;
 
-  public void setCargoRepository(CargoRepository cargoRepository) {
-    this.cargoRepository = cargoRepository;
-  }
+    @Autowired
+    CargoRepository cargoRepository;
 
-  public void setLocationRepository(LocationRepository locationRepository) {
-    this.locationRepository = locationRepository;
-  }
+    @Autowired
+    LocationRepository locationRepository;
 
-  public void testSave() {
-    Location location = locationRepository.find(new UnLocode("SESTO"));
+    @Autowired
+    SessionFactory sessionFactory;
 
-    Cargo cargo = cargoRepository.find(new TrackingId("XYZ"));
-    Date completionTime = new Date(10);
-    Date registrationTime = new Date(20);
-    HandlingEvent event = new HandlingEvent(cargo, completionTime, registrationTime, HandlingEvent.Type.CLAIM, location);
+    @Autowired
+    private DataSource dataSource;
 
-    handlingEventRepository.store(event);
+    @Autowired
+    private HibernateTransactionManager transactionManager;
 
-    flush();
+    private JdbcTemplate jdbcTemplate;
 
-    Map<String,Object> result = sjt.queryForMap("select * from HandlingEvent where id = ?", getLongId(event));
-    assertEquals(1L, result.get("CARGO_ID"));
-    assertEquals(new Date(10), result.get("COMPLETIONTIME"));
-    assertEquals(new Date(20), result.get("REGISTRATIONTIME"));
-    assertEquals("CLAIM", result.get("TYPE"));
-    // TODO: the rest of the columns
-  }
+    @Before
+    public void setup() {
+        jdbcTemplate = new JdbcTemplate(dataSource);
+        SampleDataGenerator.loadSampleData(jdbcTemplate, new TransactionTemplate(transactionManager));
+    }
 
-  public void testFindEventsForCargo() throws Exception {
-    TrackingId trackingId = new TrackingId("XYZ");
-    List<HandlingEvent> handlingEvents = handlingEventRepository.lookupHandlingHistoryOfCargo(trackingId).distinctEventsByCompletionTime();
-    assertEquals(12, handlingEvents.size());
-  }
+    @Test
+    public void testSave() {
+        Location location = locationRepository.find(new UnLocode("SESTO"));
 
-}*/
+        Cargo cargo = cargoRepository.find(new TrackingId("XYZ"));
+        Date completionTime = new Date(10);
+        Date registrationTime = new Date(20);
+        HandlingEvent event = new HandlingEvent(cargo, completionTime, registrationTime, HandlingEvent.Type.CLAIM, location);
+
+        handlingEventRepository.store(event);
+
+        flush();
+
+        Map<String, Object> result = jdbcTemplate.queryForMap("select * from HandlingEvent where id = ?", getLongId(event));
+        assertEquals(1L, result.get("CARGO_ID"));
+        assertEquals(new Date(10), result.get("COMPLETIONTIME"));
+        assertEquals(new Date(20), result.get("REGISTRATIONTIME"));
+        assertEquals("CLAIM", result.get("TYPE"));
+        // TODO: the rest of the columns
+    }
+
+    private void flush() {
+        sessionFactory.getCurrentSession().flush();
+    }
+
+    private Long getLongId(Object o) {
+        final Session session = sessionFactory.getCurrentSession();
+        if (session.contains(o)) {
+            return (Long) session.getIdentifier(o);
+        } else {
+            try {
+                Field id = o.getClass().getDeclaredField("id");
+                id.setAccessible(true);
+                return (Long) id.get(o);
+            } catch (Exception e) {
+                throw new RuntimeException();
+            }
+        }
+    }
+
+    @Test
+    public void testFindEventsForCargo() throws Exception {
+        TrackingId trackingId = new TrackingId("XYZ");
+        List<HandlingEvent> handlingEvents = handlingEventRepository.lookupHandlingHistoryOfCargo(trackingId).distinctEventsByCompletionTime();
+        assertEquals(12, handlingEvents.size());
+    }
+
+}