GaInputStepTest.java

/*! ******************************************************************************
 *
 * Pentaho Data Integration
 *
 * Copyright (C) 2002-2016 by Pentaho : http://www.pentaho.com
 *
 *******************************************************************************
 *
 * 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.
 *
 ******************************************************************************/

package org.pentaho.di.trans.steps.googleanalytics;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNull;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.anyInt;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.spy;
import static org.mockito.Mockito.when;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.apache.commons.lang.reflect.FieldUtils;
import org.junit.Test;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.pentaho.di.trans.Trans;
import org.pentaho.di.trans.TransMeta;
import org.pentaho.di.trans.step.StepMeta;

import com.google.api.services.analytics.Analytics;
import com.google.api.services.analytics.model.GaData;

/**
 * @author Andrey Khayrutdinov
 */
public class GaInputStepTest {

  @Test
  public void getNextDataEntry_WithPaging() throws Exception {
    final int recordsCount = 30;

    final String stepName = "GaInputStepTest";

    StepMeta stepMeta = new StepMeta( stepName, stepName, new GaInputStepMeta() );

    Trans trans = mock( Trans.class );

    TransMeta transMeta = mock( TransMeta.class );
    when( transMeta.findStep( stepName ) ).thenReturn( stepMeta );

    GaInputStepData data = new GaInputStepData();

    GaInputStep step = new GaInputStep( stepMeta, data, 0, transMeta, trans );

    FieldUtils.writeField( FieldUtils.getField( GaInputStep.class, "data", true ), step, data, true );

    Analytics.Data.Ga.Get mockQuery = prepareMockQuery( recordsCount );
    step = spy( step );
    doReturn( mockQuery ).when( step ).getQuery( any( Analytics.class ) );

    for ( int i = 0; i < recordsCount; i++ ) {
      List<String> next = step.getNextDataEntry();
      assertEquals( Integer.toString( i + 1 ), next.get( 0 ) );
    }
    assertNull( step.getNextDataEntry() );
  }

  private Analytics.Data.Ga.Get prepareMockQuery( int recordsCount ) throws Exception {
    final MockQueryAssistant assistant = new MockQueryAssistant( recordsCount );
    assistant.setLimit( 10 );

    Analytics.Data.Ga.Get get = mock( Analytics.Data.Ga.Get.class );
    when( get.setStartIndex( anyInt() ) ).thenAnswer( new Answer<Object>() {
      @Override
      public Object answer( InvocationOnMock invocation ) throws Throwable {
        assistant.setStartIndex( (Integer) invocation.getArguments()[ 0 ] );
        return null;
      }
    } );
    when( get.getStartIndex() ).thenAnswer( new Answer<Integer>() {
      @Override
      public Integer answer( InvocationOnMock invocation ) throws Throwable {
        return assistant.getStartIndex();
      }
    } );
    when( get.execute() ).thenAnswer( new Answer<GaData>() {
      @Override
      public GaData answer( InvocationOnMock invocation ) throws Throwable {
        return assistant.execute();
      }
    } );

    return get;
  }


  private static class MockQueryAssistant {
    private final int recordsCount;
    private Integer startIndex;
    private Integer limit;

    public MockQueryAssistant( int recordsCount ) {
      this.recordsCount = recordsCount;
    }

    public void setStartIndex( Integer startIndex ) {
      this.startIndex = startIndex;
    }

    public Integer getStartIndex() {
      return startIndex;
    }

    public void setLimit( Integer limit ) {
      this.limit = limit;
    }

    public GaData execute() {
      GaData result = new GaData();
      result.setTotalResults( recordsCount );
      result.setItemsPerPage( limit );

      List<List<String>> rows = new ArrayList<List<String>>();
      int start = ( startIndex == null ) ? 1 : startIndex;
      int end = Math.min( start + ( limit == null ? 1000 : limit ), recordsCount + 1 );
      while ( start < end ) {
        rows.add( Collections.singletonList( Integer.toString( start ) ) );
        start++;
      }

      result.setRows( rows );
      return result;
    }
  }
}