JobManagerContextTest.java

/*
 * This program is free software; you can redistribute it and/or modify it under the
 * terms of the GNU General Public License, version 2 as published by the Free Software
 * Foundation.
 *
 * You should have received a copy of the GNU General Public License along with this
 * program; if not, you can obtain a copy at http://www.gnu.org/licenses/gpl-2.0.html
 * or from the Free Software Foundation, Inc.,
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * This program 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.
 *
 *
 * Copyright 2006 - 2016 Pentaho Corporation.  All rights reserved.
 */

package org.pentaho.reporting.platform.plugin;

import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.pentaho.platform.api.engine.IPentahoSession;
import org.pentaho.platform.engine.core.system.PentahoSessionHolder;
import org.pentaho.platform.engine.core.system.PentahoSystem;
import org.pentaho.reporting.engine.classic.core.MasterReport;
import org.pentaho.reporting.libraries.resourceloader.ResourceException;
import org.pentaho.reporting.platform.plugin.async.AsyncExecutionStatus;
import org.pentaho.reporting.platform.plugin.async.IAsyncReportState;
import org.pentaho.reporting.platform.plugin.async.IPentahoAsyncExecutor;
import org.powermock.api.mockito.PowerMockito;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import java.io.IOException;
import java.util.UUID;

import static junit.framework.TestCase.assertFalse;
import static junit.framework.TestCase.assertTrue;
import static org.mockito.Matchers.anyString;
import static org.powermock.api.mockito.PowerMockito.mock;
import static org.powermock.api.mockito.PowerMockito.when;

@RunWith( PowerMockRunner.class )
@PrepareForTest( { ReportCreator.class, PentahoSessionHolder.class } )
public class JobManagerContextTest {

  private IAsyncReportState state;
  private MasterReport report;
  private UUID uuid;
  private IPentahoSession session;

  @Before
  public void before() throws ResourceException, IOException {
    PowerMockito.mockStatic( ReportCreator.class, PentahoSessionHolder.class );
    session = mock( IPentahoSession.class );
    uuid = UUID.randomUUID();
    final IPentahoAsyncExecutor executor = mock( IPentahoAsyncExecutor.class );
    state = mock( IAsyncReportState.class );
    report = mock( MasterReport.class );
    when( executor.getReportState( uuid, session ) ).thenReturn( state );
    PentahoSystem.registerObject( executor, IPentahoAsyncExecutor.class );
    when( ReportCreator.createReportByName( anyString() ) ).thenReturn( report );
    when( PentahoSessionHolder.getSession() ).thenReturn( session );
  }

  @After
  public void after() {
    state = null;
    report = null;
    uuid = null;
    session = null;
    PentahoSystem.clearObjectFactory();
  }


  @Test
  public void testNeedRecalculateFinished() throws ResourceException, IOException, JobManager.ContextFailedException {
    when( state.getStatus() ).thenReturn( AsyncExecutionStatus.FINISHED );
    final JobManager jobManager = new JobManager();
    final JobManager.ExecutionContext executionContext = jobManager.getContext( uuid.toString() );
    assertTrue( executionContext.needRecalculation( Boolean.TRUE ) );
  }

  @Test
  public void testNoNeedRecalculateFinished() throws ResourceException, IOException, JobManager.ContextFailedException {
    when( state.getStatus() ).thenReturn( AsyncExecutionStatus.FINISHED );
    final JobManager jobManager = new JobManager();
    final JobManager.ExecutionContext executionContext = jobManager.getContext( uuid.toString() );
    assertFalse( executionContext.needRecalculation( Boolean.FALSE ) );
  }

  @Test
  public void testNeedRecalculateReportLevelLimit()
    throws ResourceException, IOException, JobManager.ContextFailedException {
    when( state.getStatus() ).thenReturn( AsyncExecutionStatus.FINISHED );
    final JobManager jobManager = new JobManager();
    final JobManager.ExecutionContext executionContext = jobManager.getContext( uuid.toString() );
    when( report.getQueryLimit() ).thenReturn( 100 );
    assertTrue( executionContext.needRecalculation( Boolean.FALSE ) );
  }


  @Test
  public void testNeedRecalculateReportLimitReached()
    throws ResourceException, IOException, JobManager.ContextFailedException {
    when( state.getStatus() ).thenReturn( AsyncExecutionStatus.FINISHED );
    when( state.getIsQueryLimitReached() ).thenReturn( true );
    final JobManager jobManager = new JobManager();
    final JobManager.ExecutionContext executionContext = jobManager.getContext( uuid.toString() );
    assertTrue( executionContext.needRecalculation( Boolean.FALSE ) );
  }


}