com.epam.ta.reportportal.database.search.CriteriaMapTest.java Source code

Introduction

Here is the source code for com.epam.ta.reportportal.database.search.CriteriaMapTest.java

Source

/*
 * Copyright 2016 EPAM Systems
 *
 *
 * This file is part of EPAM Report Portal.
 * https://github.com/reportportal/commons-dao
 *
 * Report Portal is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Report Portal is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Report Portal.  If not, see <http://www.gnu.org/licenses/>.
 */

package com.epam.ta.reportportal.database.search;

import static org.assertj.core.api.Assertions.assertThat;

import com.epam.ta.reportportal.database.entity.item.TestItem;
import com.epam.ta.reportportal.database.entity.statistics.IssueCounter;
import com.mongodb.DBObject;
import org.junit.Test;

import com.epam.ta.reportportal.database.entity.Project;
import org.springframework.data.mongodb.core.query.Query;

public class CriteriaMapTest {

    @Test
    public void projectCriteria() {
        CriteriaMap<Project> projectCriteriaMap = new CriteriaMap<>(Project.class);
        assertThat(projectCriteriaMap.getCriteriaHolder("configuration").getQueryCriteria())
                .isEqualTo("configuration");
        assertThat(projectCriteriaMap.getCriteriaHolder("name").getQueryCriteria()).isEqualTo("name");
        assertThat(projectCriteriaMap.getCriteriaHolder("configuration$entryType").getQueryCriteria())
                .isEqualTo("configuration.entryType");
        assertThat(projectCriteriaMap.getCriteriaHolder("users").getQueryCriteria()).isEqualTo("users");
        assertThat(projectCriteriaMap.getCriteriaHolder("creationDate").getQueryCriteria())
                .isEqualTo("creationDate");
    }

    @Test
    public void dynamicCriteriaTest() {
        CriteriaMap<TestItem> testItemCriteriaMap = new CriteriaMap<>(TestItem.class);

        assertThat(testItemCriteriaMap.getCriteriaHolderUnchecked(IssueCounter.PRODUCT_BUG_CRITERIA)).isPresent();

        Filter filter = Filter.builder()
                .withCondition(FilterCondition.builder().withCondition(Condition.EQUALS)
                        .withSearchCriteria("statistics$defects$product_bug$total").withValue("10").build())
                .withTarget(TestItem.class).build();
        Query q = QueryBuilder.newBuilder().with(filter).build();

        assertThat(q.getQueryObject().get("statistics.issueCounter.productBug.total")).isEqualTo(10L);

    }

    @Test
    public void dynamicCriteriaCastTest() {
        Filter filter = Filter.builder()
                .withCondition(FilterCondition.builder().withCondition(Condition.GREATER_THAN_OR_EQUALS)
                        .withSearchCriteria("statistics$defects$product_bug$total").withValue("10").build())
                .withTarget(TestItem.class).build();
        Query q = QueryBuilder.newBuilder().with(filter).build();

        assertThat(((DBObject) q.getQueryObject().get("statistics.issueCounter.productBug.total")).get("$gte"))
                .isEqualTo(10L);

    }
}