FilterChainSelectClauseTest.java

/*
 * Copyright (c) 2017 OBiBa. All rights reserved.
 *
 * This program and the accompanying materials
 * are made available under the terms of the GNU Public License v3.0.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package org.obiba.magma.filter.views;

import org.junit.Test;
import org.obiba.magma.Variable;
import org.obiba.magma.filter.FilterChain;
import org.obiba.magma.views.SelectClause;

import static org.easymock.EasyMock.createMock;
import static org.easymock.EasyMock.expect;
import static org.easymock.EasyMock.replay;
import static org.easymock.EasyMock.verify;
import static org.fest.assertions.api.Assertions.assertThat;

public class FilterChainSelectClauseTest {
  //
  // Test Methods
  //

  @SuppressWarnings("unchecked")
  @Test
  public void testSelectWithIncludingFilterChain() {
    FilterChain<Variable> filterChainMock = (FilterChain<Variable>) createMock(FilterChain.class);
    Variable variableMock = createMock(Variable.class);

    expect(filterChainMock.filter(variableMock)).andReturn(variableMock);
    replay(filterChainMock);

    SelectClause filterChainSelectClause = new FilterChainSelectClause(filterChainMock);
    boolean result = filterChainSelectClause.select(variableMock);

    // Verify behaviour.
    verify(filterChainMock);

    // Verify state.
    assertThat(result).isTrue();
  }

  @SuppressWarnings("unchecked")
  @Test
  public void testSelectWithExcludingFilterChain() {
    FilterChain<Variable> filterChainMock = (FilterChain<Variable>) createMock(FilterChain.class);
    Variable variableMock = createMock(Variable.class);

    expect(filterChainMock.filter(variableMock)).andReturn(null);
    replay(filterChainMock);

    SelectClause filterChainSelectClause = new FilterChainSelectClause(filterChainMock);
    boolean result = filterChainSelectClause.select(variableMock);

    // Verify behaviour.
    verify(filterChainMock);

    // Verify state.
    assertThat(result).isFalse();

  }
}