Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- package com.innovmakers.voucherserver.job.common.util;
- import com.innovmakers.middleware.foundation.datatype.ErrorCode;
- import com.innovmakers.middleware.foundation.datatype.Response;
- import com.innovmakers.middleware.foundation.persistence.PersistenceFactory;
- import com.innovmakers.middleware.foundation.request.RequestManager;
- import com.innovmakers.voucherserver.domain.*;
- import com.innovmakers.voucherserver.job.vouchergenerator.Generator;
- import com.innovmakers.voucherserver.operations.orders.OrderOperations;
- import com.innovmakers.voucherserver.operations.vouchers.VoucherOperations;
- import com.innovmakers.voucherserver.views.VoucherActivationView;
- import com.mysql.jdbc.jdbc2.optional.MysqlConnectionPoolDataSource;
- import org.junit.AfterClass;
- import org.junit.BeforeClass;
- import org.junit.Test;
- import javax.naming.Context;
- import javax.naming.InitialContext;
- import javax.naming.NamingException;
- import javax.persistence.EntityManager;
- import java.util.Date;
- import java.util.Random;
- import static org.junit.Assert.assertTrue;
- /**
- * Created by aninm
- */
- public class GeneratorJobTest {
- @BeforeClass
- public static void setUp() throws NamingException {
- PersistenceFactory.init("voucherServerPersistence");
- EntityManager entityManager = RequestManager.getRequest().getPersistence().getEntityManager();
- setUp(entityManager);
- }
- @AfterClass
- public static void release(){
- // PersistenceFactory.destroy();
- }
- private static void setUpDatasource() throws NamingException {
- System.setProperty(Context.INITIAL_CONTEXT_FACTORY,"org.apache.naming.java.javaURLContextFactory");
- System.setProperty(Context.URL_PKG_PREFIXES, "org.apache.naming");
- InitialContext ic = new InitialContext();
- ic.createSubcontext("java:");
- ic.createSubcontext("java:/jboss");
- ic.createSubcontext("java:/jboss/datasources");
- MysqlConnectionPoolDataSource datasource = new MysqlConnectionPoolDataSource();
- datasource.setURL("jdbc:mysql://10.73.78.22:3306/voucherserver");
- datasource.setUser("root");
- datasource.setPassword("InnovMakers2010");
- ic.bind("java:/jboss/datasources/voucherserver", datasource);
- }
- private static void setUp(final EntityManager entityManager) throws NamingException {
- setUpDatasource();
- entityManager.getTransaction().begin();
- cleanTables(entityManager);
- insertRows(entityManager);
- entityManager.getTransaction().commit();
- }
- private static void cleanTables(final EntityManager entityManager){
- entityManager.createNativeQuery("DELETE FROM UpdateArchivedVoucher").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM ArchivedVoucher").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM UpdateVoucher").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM Voucher").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM Batch").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM CreateOrder").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM UpdateOrder").executeUpdate();
- entityManager.createNativeQuery("DELETE FROM Orders").executeUpdate();
- }
- private static void insertRows(final EntityManager entityManager){
- Distributor distributor = entityManager.find(Distributor.class, 1L);//name: NiteCrest, database id:1
- Administrator administrator = entityManager.find(Administrator.class, 2071001L);//username: dl, database id:2071001
- VoucherType voucherType = entityManager.find(VoucherType.class, 1L);//username: dl, database id:1
- Date date = new Date(0);
- //create order with one batch
- Order order = new Order();
- order.setDescription("description test");
- order.setStatus(OrderStatus.PROCESSING);
- order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
- order.setCreatedAt(date);
- order.setUpdatedAt(date);
- order.setDistributor(distributor);
- entityManager.persist(order);
- CreateOrder createOrder = new CreateOrder();
- createOrder.setAdministrator(administrator);
- createOrder.setOrder(order);
- createOrder.setCreatedAt(date);
- entityManager.persist(createOrder);
- Batch batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(10);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- //create order with two batches
- order = new Order();
- order.setDescription("description test");
- order.setStatus(OrderStatus.PROCESSING);
- order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
- order.setCreatedAt(date);
- order.setUpdatedAt(date);
- order.setDistributor(distributor);
- entityManager.persist(order);
- createOrder = new CreateOrder();
- createOrder.setAdministrator(administrator);
- createOrder.setOrder(order);
- createOrder.setCreatedAt(date);
- entityManager.persist(createOrder);
- batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(20);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(40);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- //create order with three batches
- order = new Order();
- order.setDescription("description test");
- order.setStatus(OrderStatus.PROCESSING);
- order.setCreationMode(OrderCreationMode.ADMINISTRATOR);
- order.setCreatedAt(date);
- order.setUpdatedAt(date);
- order.setDistributor(distributor);
- entityManager.persist(order);
- createOrder = new CreateOrder();
- createOrder.setAdministrator(administrator);
- createOrder.setOrder(order);
- createOrder.setCreatedAt(date);
- entityManager.persist(createOrder);
- batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(30);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(60);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- batch = new Batch();
- batch.setCreatedAt(date);
- batch.setName(order.getDescription());
- batch.setOrder(order);
- batch.setQuantity(90);
- batch.setStatus(BatchStatus.PROCESSING);
- batch.setStatusMessage("status message test");
- batch.setUpdatedAt(date);
- batch.setVoucherType(voucherType);
- batch.setExpiryDate(date);
- entityManager.persist(batch);
- }
- @Test
- public void insertSuccess(){
- Generator generator = new Generator("java:/jboss/datasources/voucherserver");
- generator.run();
- }
- }
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement