DBHelperTest.java

package org.test4j.module.database.utility;

import java.io.FileNotFoundException;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;
import java.util.Iterator;

import mockit.Mock;

import org.junit.Test;
import org.test4j.hamcrest.matcher.string.StringMode;
import org.test4j.junit.Test4J;
import org.test4j.junit.annotations.DataFrom;
import org.test4j.tools.commons.ResourceHelper;

@SuppressWarnings({ "rawtypes", "serial" })
public class DBHelperTest extends Test4J {

    @Test
    public void testParseSQL() throws FileNotFoundException {
        String lines = ResourceHelper.readFromFile("org/test4j/module/database/util/testParseSQL.sql");
        String[] statments = DBHelper.parseSQL(lines);
        want.array(statments).sizeEq(6);
    }

    @Test
    public void testParseSQL_SINGLE_LINE_NOTE_REGEX() {
        String line = "insert into tdd_user(id,first_name,last_name,my_date,sarary)--本行注释\n"
                + "values(3,'name3','last3','2011-03-20',2336);";
        String[] statments = DBHelper.parseSQL(line);
        want.array(statments)
                .sizeEq(1)
                .hasItems(
                        "insert into tdd_user(id,first_name,last_name,my_date,sarary) values(3,'name3','last3','2011-03-20',2336)");
    }

    @Test
    public void testParseSQL_MULTI_LINE_NOTE_REGEX() {
        String line = "insert into tdd_user(id,first_name,last_name,my_date,sarary)\n"
                + "values(2,'name2','last2',/**插入注释**/'2011-03-19',2335);";
        String[] statments = DBHelper.parseSQL(line);
        want.array(statments)
                .sizeEq(1)
                .hasItems(
                        "insert into tdd_user(id,first_name,last_name,my_date,sarary) values(2,'name2','last2','2011-03-19',2335)");
    }

    @Test
    public void testParseSQL_LineHasEnterChar() {
        String line = "insert into tdd_user(id,first_name,last_name,my_date,sarary)\n"
                + "values(3,'name\n3','last3','2011-03-20',2336);";
        String statement = "insert into tdd_user(id,first_name,last_name,my_date,sarary) values(3,'name\n3','last3','2011-03-20',2336)";
        String[] statments = DBHelper.parseSQL(line);
        want.array(statments).sizeEq(1);
        want.string(statments[0]).isEqualTo(statement);
    }

    @Test
    // "测试数据库字段为大写的时候，返回的camelName的正确性")
    public void testGetCamelFieldName() {
        ResultSetMetaData rsmeta = new MockUp<ResultSetMetaData>() {
            @Mock
            public String getColumnName(int column) throws SQLException {
                return "UUID_NAME";
            }
        }.getMockInstance();
        String uuid = reflector.invoke(DBHelper.class, "getCamelFieldName", rsmeta, 1, true);
        want.string(uuid).isEqualTo("uuidName");
    }

    @Test
    @DataFrom("dataGetWhereCondiction")
    public void testGetWhereCondiction(DataMap map, String where) {
        String result = DBHelper.getWhereCondiction(map);
        want.string(result).eq(where, StringMode.SameAsSpace);
    }

    public static Iterator dataGetWhereCondiction() {
        return new DataIterator() {
            {
                data(null, "");
                data(new DataMap(), "");
                data(new DataMap() {
                    {
                        this.put("key1", null);
                    }
                }, " where key1=?");
                data(new DataMap() {
                    {
                        this.put("key1", null);
                        this.put("key2", null);
                    }
                }, " where key1=? and key2=?");
                data(new DataMap() {
                    {
                        this.put("key1", null);
                        this.put("key2", null);
                        this.put("key3", null);
                    }
                }, " where key1=? and key2=? and key3=?");
            }
        };
    }
}