mirror of https://github.com/apache/cloudstack.git
82 lines
2.9 KiB
Java
82 lines
2.9 KiB
Java
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
//
|
|
|
|
package com.cloud.utils.db;
|
|
|
|
import static org.junit.Assert.assertTrue;
|
|
|
|
import java.util.ArrayList;
|
|
|
|
import org.junit.Test;
|
|
|
|
import com.cloud.utils.Pair;
|
|
import com.cloud.utils.db.SearchCriteria.Func;
|
|
import com.cloud.utils.db.SearchCriteria.Op;
|
|
|
|
|
|
public class GroupByTest {
|
|
|
|
protected static final String EXPECTED_QUERY = "BASE GROUP BY FIRST(TEST_TABLE.TEST_COLUMN), MAX(TEST_TABLE.TEST_COLUMN) HAVING COUNT(TEST_TABLE2.TEST_COLUMN2) > ? ";
|
|
@Test
|
|
public void testToSql() {
|
|
// Prepare
|
|
final StringBuilder sb = new StringBuilder("BASE");
|
|
final GroupByExtension groupBy = new GroupByExtension(new SearchBaseExtension(String.class, String.class));
|
|
|
|
final Attribute att = new Attribute("TEST_TABLE", "TEST_COLUMN");
|
|
final Pair<Func, Attribute> pair1 = new Pair<SearchCriteria.Func, Attribute>(SearchCriteria.Func.FIRST, att);
|
|
final Pair<Func, Attribute> pair2 = new Pair<SearchCriteria.Func, Attribute>(SearchCriteria.Func.MAX, att);
|
|
groupBy._groupBys = new ArrayList<Pair<Func, Attribute>>();
|
|
groupBy._groupBys.add(pair1);
|
|
groupBy._groupBys.add(pair2);
|
|
groupBy.having(SearchCriteria.Func.COUNT, att, Op.GT, "SOME_VALUE");
|
|
|
|
// Execute
|
|
groupBy.toSql(sb);
|
|
|
|
// Assert
|
|
assertTrue("It didn't create the expected SQL query.", sb.toString().equals(EXPECTED_QUERY));
|
|
}
|
|
}
|
|
|
|
class GroupByExtension extends GroupBy<SearchBaseExtension, String, String> {
|
|
|
|
public GroupByExtension(final SearchBaseExtension builder) {
|
|
super(builder);
|
|
_builder = builder;
|
|
}
|
|
|
|
@Override
|
|
protected void init(final SearchBaseExtension builder) {
|
|
}
|
|
}
|
|
|
|
@SuppressWarnings({"rawtypes", "unchecked"})
|
|
class SearchBaseExtension extends SearchBase<SearchBaseExtension, String, String>{
|
|
|
|
SearchBaseExtension(final Class entityType, final Class resultType) {
|
|
super(entityType, resultType);
|
|
_specifiedAttrs = new ArrayList<Attribute>();
|
|
_specifiedAttrs.add(new Attribute("TEST_TABLE2", "TEST_COLUMN2"));
|
|
}
|
|
|
|
@Override
|
|
protected void init(final Class<String> entityType, final Class<String> resultType) {
|
|
}
|
|
}
|