Advertisement
Guest User

Untitled

a guest
Mar 3rd, 2017
100
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 8.34 KB | None | 0 0
  1. package com.innovmakers.voucherserver.job.common.util;
  2.  
  3. import com.innovmakers.middleware.foundation.datatype.ErrorCode;
  4. import com.innovmakers.middleware.foundation.datatype.Response;
  5. import com.innovmakers.middleware.foundation.persistence.PersistenceFactory;
  6. import com.innovmakers.middleware.foundation.request.RequestManager;
  7. import com.innovmakers.voucherserver.domain.*;
  8. import com.innovmakers.voucherserver.job.vouchergenerator.Generator;
  9. import com.innovmakers.voucherserver.operations.orders.OrderOperations;
  10. import com.innovmakers.voucherserver.operations.vouchers.VoucherOperations;
  11. import com.innovmakers.voucherserver.views.VoucherActivationView;
  12. import com.mysql.jdbc.jdbc2.optional.MysqlConnectionPoolDataSource;
  13. import org.junit.AfterClass;
  14. import org.junit.BeforeClass;
  15. import org.junit.Test;
  16.  
  17. import javax.naming.Context;
  18. import javax.naming.InitialContext;
  19. import javax.naming.NamingException;
  20. import javax.persistence.EntityManager;
  21.  
  22. import java.util.Date;
  23. import java.util.Random;
  24.  
  25. import static org.junit.Assert.assertTrue;
  26.  
  27. /**
  28. * Created by aninm
  29. */
  30. public class GeneratorJobTest {
  31.  
  32. @BeforeClass
  33. public static void setUp() throws NamingException {
  34. PersistenceFactory.init("voucherServerPersistence");
  35.  
  36. EntityManager entityManager = RequestManager.getRequest().getPersistence().getEntityManager();
  37.  
  38. setUp(entityManager);
  39. }
  40.  
  41. @AfterClass
  42. public static void release(){
  43. // PersistenceFactory.destroy();
  44. }
  45.  
  46.  
  47. private static void setUpDatasource() throws NamingException {
  48. System.setProperty(Context.INITIAL_CONTEXT_FACTORY,"org.apache.naming.java.javaURLContextFactory");
  49. System.setProperty(Context.URL_PKG_PREFIXES, "org.apache.naming");
  50.  
  51. InitialContext ic = new InitialContext();
  52.  
  53. ic.createSubcontext("java:");
  54. ic.createSubcontext("java:/jboss");
  55. ic.createSubcontext("java:/jboss/datasources");
  56.  
  57. MysqlConnectionPoolDataSource datasource = new MysqlConnectionPoolDataSource();
  58. datasource.setURL("jdbc:mysql://10.73.78.22:3306/voucherserver");
  59. datasource.setUser("root");
  60. datasource.setPassword("InnovMakers2010");
  61.  
  62. ic.bind("java:/jboss/datasources/voucherserver", datasource);
  63. }
  64.  
  65.  
  66. private static void setUp(final EntityManager entityManager) throws NamingException {
  67. setUpDatasource();
  68.  
  69. entityManager.getTransaction().begin();
  70. cleanTables(entityManager);
  71. insertRows(entityManager);
  72. entityManager.getTransaction().commit();
  73. }
  74.  
  75. private static void cleanTables(final EntityManager entityManager){
  76. entityManager.createNativeQuery("DELETE FROM UpdateArchivedVoucher").executeUpdate();
  77. entityManager.createNativeQuery("DELETE FROM ArchivedVoucher").executeUpdate();
  78.  
  79. entityManager.createNativeQuery("DELETE FROM UpdateVoucher").executeUpdate();
  80. entityManager.createNativeQuery("DELETE FROM Voucher").executeUpdate();
  81.  
  82. entityManager.createNativeQuery("DELETE FROM Batch").executeUpdate();
  83.  
  84. entityManager.createNativeQuery("DELETE FROM CreateOrder").executeUpdate();
  85. entityManager.createNativeQuery("DELETE FROM UpdateOrder").executeUpdate();
  86. entityManager.createNativeQuery("DELETE FROM Orders").executeUpdate();
  87. }
  88.  
  89.  
  90. private static void insertRows(final EntityManager entityManager){
  91.  
  92. Distributor distributor = entityManager.find(Distributor.class, 1L);//name: NiteCrest, database id:1
  93. Administrator administrator = entityManager.find(Administrator.class, 2071001L);//username: dl, database id:2071001
  94. VoucherType voucherType = entityManager.find(VoucherType.class, 1L);//username: dl, database id:1
  95.  
  96. Date date = new Date(0);
  97.  
  98. //create order with one batch
  99. Order order = new Order();
  100. order.setDescription("description test");
  101. order.setStatus(OrderStatus.PROCESSING);
  102. order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
  103. order.setCreatedAt(date);
  104. order.setUpdatedAt(date);
  105. order.setDistributor(distributor);
  106. entityManager.persist(order);
  107.  
  108. CreateOrder createOrder = new CreateOrder();
  109. createOrder.setAdministrator(administrator);
  110. createOrder.setOrder(order);
  111. createOrder.setCreatedAt(date);
  112. entityManager.persist(createOrder);
  113.  
  114. Batch batch = new Batch();
  115. batch.setCreatedAt(date);
  116. batch.setName(order.getDescription());
  117. batch.setOrder(order);
  118. batch.setQuantity(10);
  119. batch.setStatus(BatchStatus.PROCESSING);
  120. batch.setStatusMessage("status message test");
  121. batch.setUpdatedAt(date);
  122. batch.setVoucherType(voucherType);
  123. batch.setExpiryDate(date);
  124.  
  125. entityManager.persist(batch);
  126.  
  127. //create order with two batches
  128. order = new Order();
  129. order.setDescription("description test");
  130. order.setStatus(OrderStatus.PROCESSING);
  131. order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
  132. order.setCreatedAt(date);
  133. order.setUpdatedAt(date);
  134. order.setDistributor(distributor);
  135. entityManager.persist(order);
  136.  
  137. createOrder = new CreateOrder();
  138. createOrder.setAdministrator(administrator);
  139. createOrder.setOrder(order);
  140. createOrder.setCreatedAt(date);
  141. entityManager.persist(createOrder);
  142.  
  143. batch = new Batch();
  144. batch.setCreatedAt(date);
  145. batch.setName(order.getDescription());
  146. batch.setOrder(order);
  147. batch.setQuantity(20);
  148. batch.setStatus(BatchStatus.PROCESSING);
  149. batch.setStatusMessage("status message test");
  150. batch.setUpdatedAt(date);
  151. batch.setVoucherType(voucherType);
  152. batch.setExpiryDate(date);
  153.  
  154. entityManager.persist(batch);
  155.  
  156. batch = new Batch();
  157. batch.setCreatedAt(date);
  158. batch.setName(order.getDescription());
  159. batch.setOrder(order);
  160. batch.setQuantity(40);
  161. batch.setStatus(BatchStatus.PROCESSING);
  162. batch.setStatusMessage("status message test");
  163. batch.setUpdatedAt(date);
  164. batch.setVoucherType(voucherType);
  165. batch.setExpiryDate(date);
  166.  
  167. entityManager.persist(batch);
  168.  
  169. //create order with three batches
  170. order = new Order();
  171. order.setDescription("description test");
  172. order.setStatus(OrderStatus.PROCESSING);
  173. order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
  174. order.setCreatedAt(date);
  175. order.setUpdatedAt(date);
  176. order.setDistributor(distributor);
  177. entityManager.persist(order);
  178.  
  179. createOrder = new CreateOrder();
  180. createOrder.setAdministrator(administrator);
  181. createOrder.setOrder(order);
  182. createOrder.setCreatedAt(date);
  183. entityManager.persist(createOrder);
  184.  
  185. batch = new Batch();
  186. batch.setCreatedAt(date);
  187. batch.setName(order.getDescription());
  188. batch.setOrder(order);
  189. batch.setQuantity(30);
  190. batch.setStatus(BatchStatus.PROCESSING);
  191. batch.setStatusMessage("status message test");
  192. batch.setUpdatedAt(date);
  193. batch.setVoucherType(voucherType);
  194. batch.setExpiryDate(date);
  195.  
  196. entityManager.persist(batch);
  197.  
  198. batch = new Batch();
  199. batch.setCreatedAt(date);
  200. batch.setName(order.getDescription());
  201. batch.setOrder(order);
  202. batch.setQuantity(60);
  203. batch.setStatus(BatchStatus.PROCESSING);
  204. batch.setStatusMessage("status message test");
  205. batch.setUpdatedAt(date);
  206. batch.setVoucherType(voucherType);
  207. batch.setExpiryDate(date);
  208.  
  209. entityManager.persist(batch);
  210.  
  211. batch = new Batch();
  212. batch.setCreatedAt(date);
  213. batch.setName(order.getDescription());
  214. batch.setOrder(order);
  215. batch.setQuantity(90);
  216. batch.setStatus(BatchStatus.PROCESSING);
  217. batch.setStatusMessage("status message test");
  218. batch.setUpdatedAt(date);
  219. batch.setVoucherType(voucherType);
  220. batch.setExpiryDate(date);
  221.  
  222. entityManager.persist(batch);
  223. }
  224.  
  225.  
  226. @Test
  227. public void insertSuccess(){
  228. Generator generator = new Generator("java:/jboss/datasources/voucherserver");
  229. generator.run();
  230. }
  231.  
  232.  
  233.  
  234.  
  235. }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement