GroupBaseCRUDTest.java

package com.taobao.tddl.qatest.group;

import java.util.Date;
import java.util.List;
import java.util.Map;

import org.junit.Assert;
import org.junit.Test;
import org.springframework.dao.DataAccessException;

import com.taobao.tddl.common.GroupDataSourceRouteHelper;
import com.taobao.tddl.qatest.util.DateUtil;

public class GroupBaseCRUDTest extends GroupTestCase {

    @Test
    public void insertByPreStTest() {
        String sql = "insert into normaltbl_0001 (pk,gmt_create) values (?,?)";
        Object[] arguments = new Object[] { RANDOM_ID, time };
        tddlJT.update(sql, arguments);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=?", new Object[] { RANDOM_ID });
        Assert.assertEquals(time, String.valueOf(re.get("gmt_create")));
    }

    @Test
    public void updateByPreStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        String sql = "update normaltbl_0001 set gmt_create=? where pk=?";
        Object[] arguments = new Object[] { nextDay, RANDOM_ID };
        tddlJT.update(sql, arguments);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=?", new Object[] { RANDOM_ID });
        Assert.assertEquals(nextDay, String.valueOf(re.get("gmt_create")));
    }

    @Test
    public void queryByPreStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        String sql = "select * from normaltbl_0001 where pk=?";
        Object[] arguments = new Object[] { RANDOM_ID };
        Map re = tddlJT.queryForMap(sql, arguments);
        Assert.assertEquals(time, String.valueOf(re.get("gmt_create")));
    }

    @Test
    public void deleteByPreStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        String sql = "delete from normaltbl_0001 where pk=?";
        Object[] arguments = new Object[] { RANDOM_ID };
        tddlJT.update(sql, arguments);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        List re = tddlJT.queryForList("select * from normaltbl_0001 where pk=?", new Object[] { RANDOM_ID });
        Assert.assertEquals(0, re.size());
    }

    @Test
    public void replaceByPreStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        String sql = "replace into normaltbl_0001 (pk,gmt_create) values (?,?)";
        tddlJT.update(sql, new Object[] { RANDOM_ID, nextDay });
        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=?", new Object[] { RANDOM_ID });
        Assert.assertEquals(nextDay, String.valueOf(re.get("gmt_create")));
    }

    @Test
    public void insertByStTest() {
        String sql = "insert into normaltbl_0001 (pk,gmt_create) values (" + RANDOM_ID + ",CURDATE())";
        tddlJT.update(sql);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=" + RANDOM_ID);
        Assert.assertEquals(time, DateUtil.formatDate((Date) re.get("gmt_create"), DateUtil.DATE_FULLHYPHEN));
    }

    @Test
    public void updateByStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        String sql = "update normaltbl_0001 set gmt_create= ADDDATE(CURDATE(),INTERVAL 1 DAY) where pk=" + RANDOM_ID;
        tddlJT.update(sql);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=" + RANDOM_ID);
        Assert.assertEquals(nextDay, DateUtil.formatDate((Date) re.get("gmt_create"), DateUtil.DATE_FULLHYPHEN));
    }

    @Test
    public void queryByStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=" + RANDOM_ID);
        Assert.assertEquals(time, DateUtil.formatDate((Date) re.get("gmt_create"), DateUtil.DATE_FULLHYPHEN));
    }

    @Test
    public void deleteByStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });

        String sql = "delete from normaltbl_0001 where pk=" + RANDOM_ID;
        tddlJT.update(sql);

        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        List re = tddlJT.queryForList("select * from normaltbl_0001 where pk=" + RANDOM_ID);
        Assert.assertEquals(0, re.size());
    }

    @Test
    public void replaceByStTest() {
        prepareData(tddlJT, "insert into normaltbl_0001 (pk,gmt_create) values (?,?)", new Object[] { RANDOM_ID, time });
        String sql = "replace into normaltbl_0001 (gmt_create,pk) values (ADDDATE(CURDATE(),INTERVAL 1 DAY),"
                     + RANDOM_ID + ")";
        tddlJT.update(sql);
        GroupDataSourceRouteHelper.executeByGroupDataSourceIndex(0);
        Map re = tddlJT.queryForMap("select * from normaltbl_0001 where pk=" + RANDOM_ID);
        Assert.assertEquals(nextDay, DateUtil.formatDate((Date) re.get("gmt_create"), DateUtil.DATE_FULLHYPHEN));
    }

    @Test
    public void noneExistTblTest() {
        String sql = "insert into normaltbl_noneExist (pk,gmt_create) values (?,?)";
        try {
            tddlJT.update(sql, new Object[] { RANDOM_ID, time });
            Assert.fail();
        } catch (DataAccessException e) {
        }
    }
}