ass2-service/courseplan/src/test/java/dst/ass2/service/courseplan/TestDataInserter.java

   1 package dst.ass2.service.courseplan;
   2
   3 import dst.ass1.jpa.model.IModelFactory;
   4 import dst.ass1.jpa.tests.TestData;
   5 import org.slf4j.Logger;
   6 import org.slf4j.LoggerFactory;
   7 import org.springframework.transaction.PlatformTransactionManager;
   8 import org.springframework.transaction.support.TransactionTemplate;
   9
  10 import javax.persistence.EntityManager;
  11
  12 public class TestDataInserter {
  13     private static final Logger LOG = LoggerFactory.getLogger(TestDataInserter.class);
  14
  15     private PlatformTransactionManager transactionManager;
  16     private IModelFactory modelFactory;
  17     private TestData testData;
  18
  19     public TestDataInserter(TestData testData, IModelFactory modelFactory, PlatformTransactionManager transactionManager) {
  20         this.testData = testData;
  21         this.modelFactory = modelFactory;
  22         this.transactionManager = transactionManager;
  23     }
  24
  25     public void insertTestData(EntityManager em) {
  26         LOG.info("Inserting test data...");
  27         TransactionTemplate tx = new TransactionTemplate(transactionManager);
  28         tx.execute(status -> {
  29             testData.insert(modelFactory, em);
  30             return null;
  31         });
  32     }
  33
  34 }