me.yanaga.querydsl.args.core.single.SingleComparableArgumentTest.java Source code

Introduction

Here is the source code for me.yanaga.querydsl.args.core.single.SingleComparableArgumentTest.java

Source

package me.yanaga.querydsl.args.core.single;

/*
 * #%L
 * querydsl-args-core
 * %%
 * Copyright (C) 2014 - 2015 Edson Yanaga
 * %%
 * Licensed 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.
 * #L%
 */

import com.mysema.query.BooleanBuilder;
import com.mysema.query.jpa.impl.JPAQuery;
import me.yanaga.querydsl.args.core.TestConfig;
import me.yanaga.querydsl.args.core.model.CustomComparableType;
import me.yanaga.querydsl.args.core.model.Person;
import me.yanaga.querydsl.args.core.model.QPerson;
import org.springframework.test.context.ContextConfiguration;
import org.springframework.test.context.testng.AbstractTransactionalTestNGSpringContextTests;
import org.testng.annotations.BeforeMethod;
import org.testng.annotations.Test;

import javax.persistence.EntityManager;
import javax.persistence.PersistenceContext;
import java.time.LocalDate;

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

@ContextConfiguration(classes = TestConfig.class)
public class SingleComparableArgumentTest extends AbstractTransactionalTestNGSpringContextTests {

    @PersistenceContext
    private EntityManager entityManager;

    @BeforeMethod
    public void setUp() {
        Person person = new Person();
        person.setOneCustomComparableType(CustomComparableType.of(LocalDate.of(2015, 2, 26)));
        person.setAnotherCustomComparableType(CustomComparableType.of(LocalDate.of(2015, 3, 23)));
        entityManager.persist(person);
    }

    @Test
    public void testAppendDefaultWithOneArgument() {
        CustomComparableType value = CustomComparableType.of(LocalDate.of(2015, 2, 26));
        SingleComparableArgument<CustomComparableType> argument = SingleComparableArgument.of(value);
        BooleanBuilder builder = new BooleanBuilder();
        argument.append(builder, QPerson.person.oneCustomComparableType);
        SingleArgument.of(value).append(builder, QPerson.person.oneCustomComparableType);
        Person result = new JPAQuery(entityManager).from(QPerson.person).where(builder)
                .uniqueResult(QPerson.person);
        assertThat(result.getOneCustomComparableType()).isEqualTo(value);
    }

    @Test
    public void testAppendGoeWithTwoArguments() {
        SingleComparableArgument<CustomComparableType> argument = SingleComparableArgument
                .of(CustomComparableType.of(LocalDate.of(2015, 3, 30)));
        BooleanBuilder builder = new BooleanBuilder();
        argument.append(builder, QPerson.person.oneCustomComparableType,
                QPerson.person.anotherCustomComparableType);
        Person result = new JPAQuery(entityManager).from(QPerson.person).where(builder)
                .uniqueResult(QPerson.person);
        assertThat(result).isNull();
    }

}