Search in sources :

Example 6 with DalHints

use of com.ctrip.platform.dal.dao.DalHints in project dal by ctripcorp.

the class DalTransactionManagerTest method testIsInTransaction.

@Test
public void testIsInTransaction() {
    try {
        final DalTransactionManager test = new DalTransactionManager(getDalConnectionManager());
        assertFalse(DalTransactionManager.isInTransaction());
        ConnectionAction<?> action = new ConnectionAction<Object>() {

            public Object execute() throws Exception {
                Assert.assertTrue(DalTransactionManager.isInTransaction());
                return null;
            }
        };
        action.operation = DalEventEnum.EXECUTE;
        test.doInTransaction(action, new DalHints());
    } catch (Exception e) {
        fail();
        e.printStackTrace();
    }
}
Also used : DalHints(com.ctrip.platform.dal.dao.DalHints) DalTransactionManager(com.ctrip.platform.dal.dao.client.DalTransactionManager) ConnectionAction(com.ctrip.platform.dal.dao.client.ConnectionAction) SQLException(java.sql.SQLException) Test(org.junit.Test)

Example 7 with DalHints

use of com.ctrip.platform.dal.dao.DalHints in project dal by ctripcorp.

the class DalTransactionManagerTest method testClearCurrentTransaction.

@Test
public void testClearCurrentTransaction() {
    final DalHints hints = new DalHints();
    try {
        final DalTransactionManager test = new DalTransactionManager(getDalConnectionManager());
        assertFalse(DalTransactionManager.isInTransaction());
        ConnectionAction<?> action = new ConnectionAction<Object>() {

            public Object execute() throws Exception {
                assertNotNull(DalTransactionManager.getCurrentDbMeta());
                return null;
            }
        };
        action.operation = DalEventEnum.EXECUTE;
        test.doInTransaction(action, hints);
    } catch (Exception e) {
        fail();
        e.printStackTrace();
    }
    assertFalse(DalTransactionManager.isInTransaction());
    assertNull(DalTransactionManager.getCurrentDbMeta());
}
Also used : DalHints(com.ctrip.platform.dal.dao.DalHints) DalTransactionManager(com.ctrip.platform.dal.dao.client.DalTransactionManager) ConnectionAction(com.ctrip.platform.dal.dao.client.ConnectionAction) SQLException(java.sql.SQLException) Test(org.junit.Test)

Example 8 with DalHints

use of com.ctrip.platform.dal.dao.DalHints in project dal by ctripcorp.

the class DalTransactionManagerTest method testRollbackListeners.

@Test
public void testRollbackListeners() {
    final DalHints hints = new DalHints();
    final DalTransactionListener testListener = new DalTransactionListener() {

        @Override
        public void beforeCommit() {
        }

        @Override
        public void beforeRollback() {
            Assert.assertTrue(DalTransactionManager.isInTransaction());
        }

        @Override
        public void afterCommit() {
            fail();
        }

        @Override
        public void afterRollback() {
            Assert.assertFalse(DalTransactionManager.isInTransaction());
        }
    };
    final DalTransactionListener testListener1 = new DalTransactionListener() {

        @Override
        public void beforeCommit() throws SQLException {
            throw new SQLException();
        }

        @Override
        public void beforeRollback() {
            Assert.assertTrue(DalTransactionManager.isInTransaction());
        }

        @Override
        public void afterCommit() {
            fail();
        }

        @Override
        public void afterRollback() {
            Assert.assertFalse(DalTransactionManager.isInTransaction());
        }
    };
    try {
        final DalTransactionManager test = new DalTransactionManager(getDalConnectionManager());
        ConnectionAction<?> action = new ConnectionAction<Object>() {

            public Object execute() throws Exception {
                DalTransactionManager.register(testListener);
                // The 2nd listener will cause transaction rollback
                DalTransactionManager.register(testListener1);
                return null;
            }
        };
        action.operation = DalEventEnum.EXECUTE;
        test.doInTransaction(action, hints);
        fail();
    } catch (Exception e) {
    }
}
Also used : DalTransactionListener(com.ctrip.platform.dal.dao.client.DalTransactionListener) DalHints(com.ctrip.platform.dal.dao.DalHints) DalTransactionManager(com.ctrip.platform.dal.dao.client.DalTransactionManager) SQLException(java.sql.SQLException) ConnectionAction(com.ctrip.platform.dal.dao.client.ConnectionAction) SQLException(java.sql.SQLException) Test(org.junit.Test)

Example 9 with DalHints

use of com.ctrip.platform.dal.dao.DalHints in project dal by ctripcorp.

the class DalTransactionManagerTest method testGetCurrentListeners.

@Test
public void testGetCurrentListeners() {
    final DalHints hints = new DalHints();
    final DalTransactionListener testListener = new DalTransactionListener() {

        @Override
        public void beforeCommit() {
        }

        @Override
        public void beforeRollback() {
        }

        @Override
        public void afterCommit() {
        }

        @Override
        public void afterRollback() {
        }
    };
    try {
        final DalTransactionManager test = new DalTransactionManager(getDalConnectionManager());
        DalTransactionManager.getCurrentListeners();
        fail();
    } catch (Exception e) {
    }
    try {
        final DalTransactionManager test = new DalTransactionManager(getDalConnectionManager());
        ConnectionAction<?> action = new ConnectionAction<Object>() {

            public Object execute() throws Exception {
                DalTransactionManager.register(testListener);
                DalTransactionManager.register(testListener);
                Assert.assertEquals(2, DalTransactionManager.getCurrentListeners().size());
                return null;
            }
        };
        action.operation = DalEventEnum.EXECUTE;
        test.doInTransaction(action, hints);
        try {
            DalTransactionManager.getCurrentListeners();
            fail();
        } catch (Exception e) {
        }
    } catch (Exception e) {
        e.printStackTrace();
        fail();
    }
}
Also used : DalTransactionListener(com.ctrip.platform.dal.dao.client.DalTransactionListener) DalHints(com.ctrip.platform.dal.dao.DalHints) DalTransactionManager(com.ctrip.platform.dal.dao.client.DalTransactionManager) ConnectionAction(com.ctrip.platform.dal.dao.client.ConnectionAction) SQLException(java.sql.SQLException) Test(org.junit.Test)

Example 10 with DalHints

use of com.ctrip.platform.dal.dao.DalHints in project dal by ctripcorp.

the class PartialQueryTableDaoUnitTest method testFindBySelectedField.

@Test
public void testFindBySelectedField() throws Exception {
    DalTableDao<Person> client = new DalTableDao<>(new DalDefaultJpaParser<>(Person.class));
    List<Integer> peopleIds = new ArrayList<>();
    peopleIds.add(1);
    peopleIds.add(2);
    peopleIds.add(3);
    List<Integer> cityIds = new ArrayList<>();
    cityIds.add(1);
    cityIds.add(2);
    cityIds.add(3);
    SelectSqlBuilder builder = new SelectSqlBuilder();
    builder.select("DataChange_LastTime", "CityID", "Name", "ProvinceID");
    builder.in("PeopleID", peopleIds, Types.INTEGER, false);
    builder.and();
    builder.in("CityID", cityIds, Types.INTEGER, false);
    try {
        List<Person> ret = client.query(builder, new DalHints().inAllShards().inTableShard(1));
        Assert.assertNull(ret.get(0).getCountryID());
        Assert.assertNull(ret.get(0).getPeopleID());
    } catch (DalException e) {
        Assert.fail();
    }
}
Also used : DalHints(com.ctrip.platform.dal.dao.DalHints) DalException(com.ctrip.platform.dal.exceptions.DalException) ArrayList(java.util.ArrayList) SelectSqlBuilder(com.ctrip.platform.dal.dao.sqlbuilder.SelectSqlBuilder) DalTableDao(com.ctrip.platform.dal.dao.DalTableDao) Test(org.junit.Test)

Aggregations

DalHints (com.ctrip.platform.dal.dao.DalHints)621 Test (org.junit.Test)473 SQLException (java.sql.SQLException)278 StatementParameters (com.ctrip.platform.dal.dao.StatementParameters)245 ArrayList (java.util.ArrayList)113 ClientTestModel (test.com.ctrip.platform.dal.dao.unitbase.ClientTestModel)57 Future (java.util.concurrent.Future)32 KeyHolder (com.ctrip.platform.dal.dao.KeyHolder)27 DataBase (com.ctrip.platform.dal.dao.configure.DataBase)26 DatabaseSelector (com.ctrip.platform.dal.dao.configure.DatabaseSelector)26 DalDefaultJpaParser (com.ctrip.platform.dal.dao.helper.DalDefaultJpaParser)25 AfterClass (org.junit.AfterClass)23 BeforeClass (org.junit.BeforeClass)23 List (java.util.List)22 DalTableDao (com.ctrip.platform.dal.dao.DalTableDao)20 BatchUpdateTask (com.ctrip.platform.dal.dao.task.BatchUpdateTask)18 After (org.junit.After)17 Before (org.junit.Before)17 HashMap (java.util.HashMap)16 Map (java.util.Map)16