PagerUtilsTest_Count_SQLServer_0.java

package com.alibaba.druid.bvt.sql;

import junit.framework.TestCase;

import org.junit.Assert;

import com.alibaba.druid.sql.PagerUtils;
import com.alibaba.druid.util.JdbcConstants;

public class PagerUtilsTest_Count_SQLServer_0 extends TestCase {

    public void test_sqlserver_0() throws Exception {
        String sql = "select * from t";
        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
        Assert.assertEquals("SELECT COUNT(*)\n" + //
                            "FROM t", result);
    }

    public void test_sqlserver_1() throws Exception {
        String sql = "select id, name from t";
        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
        Assert.assertEquals("SELECT COUNT(*)\n" + //
                            "FROM t", result);
    }

    public void test_sqlserver_2() throws Exception {
        String sql = "select id, name from t order by id";
        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
        Assert.assertEquals("SELECT COUNT(*)\n" + //
                            "FROM t", result);
    }

    public void test_sqlserver_group_0() throws Exception {
        String sql = "select type, count(*) from t group by type";
        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
        Assert.assertEquals("SELECT COUNT(*)" + //
                            "\nFROM (SELECT type, COUNT(*)" + //
                            "\n\tFROM t" + //
                            "\n\tGROUP BY type" + //
                            "\n\t) ALIAS_COUNT", result);
    }

    public void test_sqlserver_union_0() throws Exception {
        String sql = "select id, name from t1 union select id, name from t2 order by id";
        String result = PagerUtils.count(sql, JdbcConstants.SQL_SERVER);
        Assert.assertEquals("SELECT COUNT(*)\n" + //
                            "FROM (SELECT id, name" + //
                            "\n\tFROM t1" + //
                            "\n\tUNION" + //
                            "\n\tSELECT id, name" + //
                            "\n\tFROM t2" + //
                            "\n\t) ALIAS_COUNT", result);
    }
}