ScheduleHelperTest.java

/*!
* This program is free software; you can redistribute it and/or modify it under the
* terms of the GNU Lesser General Public License, version 2.1 as published by the Free Software
* Foundation.
*
* You should have received a copy of the GNU Lesser General Public License along with this
* program; if not, you can obtain a copy at http://www.gnu.org/licenses/old-licenses/lgpl-2.1.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 Lesser General Public License for more details.
*
* Copyright (c) 2002-2016 Pentaho Corporation..  All rights reserved.
*/

package org.pentaho.mantle.client.dialogs.scheduling;

import com.google.gwt.http.client.RequestBuilder;
import com.google.gwt.json.client.JSONObject;
import com.google.gwt.json.client.JSONString;
import com.google.gwtmockito.GwtMockitoTestRunner;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mortbay.util.ajax.JSON;
import org.pentaho.mantle.client.workspace.JsJob;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith( GwtMockitoTestRunner.class )
public class ScheduleHelperTest {
  public static final String JOB_ID = "JobID123";

  @Test
  public void testBuildJobRequest() throws Exception {
    JsJob job = null;
    JSONObject requestPayload = null;

    // null job so use job URL
    ScheduleHelper helper = mock( ScheduleHelper.class );
    RequestBuilder requestBuilder = ScheduleHelper.buildRequestForJob( job, requestPayload );
    assertTrue( requestBuilder.getUrl().endsWith( ScheduleHelper.JOB_SCHEDULER_URL ) );
    assertNull( requestPayload );
    assertHeaders( requestBuilder );

    // no Job ID so use job URL
    job = mock( JsJob.class );
    when( job.getJobId() ).thenReturn( null );
    requestBuilder = ScheduleHelper.buildRequestForJob( job, requestPayload );
    assertTrue( requestBuilder.getUrl().endsWith( ScheduleHelper.JOB_SCHEDULER_URL ) );
    assertNull( requestPayload );
    assertHeaders( requestBuilder );

    // Job ID so use update URL, payload is null so nothing is set
    job = mock( JsJob.class );
    when( job.getJobId() ).thenReturn( JOB_ID );
    requestBuilder = ScheduleHelper.buildRequestForJob( job, requestPayload );
    assertTrue( requestBuilder.getUrl().endsWith( ScheduleHelper.UPDATE_JOB_SCHEDULER_URL ) );
    assertNull( requestPayload );
    assertHeaders( requestBuilder );

    // Job ID so use update URL, payload has Job ID
    job = mock( JsJob.class );
    when( job.getJobId() ).thenReturn( JOB_ID );
    requestPayload = mock( JSONObject.class );
    requestBuilder = ScheduleHelper.buildRequestForJob( job, requestPayload );
    assertTrue( requestBuilder.getUrl().endsWith( ScheduleHelper.UPDATE_JOB_SCHEDULER_URL ) );
    assertNotNull( requestPayload );
    verify( requestPayload ).put( "jobId", new JSONString( JOB_ID ) );
    assertHeaders( requestBuilder );
  }

  public void assertHeaders( RequestBuilder builder ) {
    assertEquals( "01 Jan 1970 00:00:00 GMT", builder.getHeader( "If-Modified-Since" ) );
    assertEquals( "application/json", builder.getHeader( "Content-Type" ) );
  }
}