ExportSingleTrackTest.java

/*
 * Copyright 2012 Google Inc.
 * 
 * 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 com.google.android.apps.mytracks.endtoendtest.common;

import com.google.android.apps.mytracks.TrackListActivity;
import com.google.android.apps.mytracks.endtoendtest.EndToEndTestUtils;
import com.google.android.apps.mytracks.endtoendtest.GoogleUtils;
import com.google.android.maps.mytracks.R;
import com.google.api.services.drive.model.File;

import android.app.Instrumentation;
import android.test.ActivityInstrumentationTestCase2;

import java.util.List;

/**
 * Tests export single track to Google. Each test would export one track to
 * Google Maps, Google Fusion Tables and Google Spreadsheets.
 * 
 * @author youtaol
 */
public class ExportSingleTrackTest extends ActivityInstrumentationTestCase2<TrackListActivity> {

  private Instrumentation instrumentation;
  private TrackListActivity activityMyTracks;

  public ExportSingleTrackTest() {
    super(TrackListActivity.class);
  }

  @Override
  protected void setUp() throws Exception {
    super.setUp();
    instrumentation = getInstrumentation();
    activityMyTracks = getActivity();

    GoogleUtils.deleteAllMaps(activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1);
    GoogleUtils.deleteDriveTestFiles(
        activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1);
    EndToEndTestUtils.setupForAllTest(instrumentation, activityMyTracks);
  }

  /**
   * Checks all services and send to google.
   */
  public void testCreateAndSendTrack_send() {
    // Create a new track.
    EndToEndTestUtils.createSimpleTrack(1, false);
    instrumentation.waitForIdleSync();
    checkSendTrackToGoogle();
  }

  /**
   * Checks all services and send to google.
   */
  public void testCreateAndSendTrack_sendPausedTrack() {
    EndToEndTestUtils.deleteAllTracks();
    EndToEndTestUtils.createTrackWithPause(3);
    instrumentation.waitForIdleSync();
    checkSendTrackToGoogle();
  }

  /**
   * Sends two tracks with same activity. Two tracks should be in the same
   * spreadsheet.
   */
  public void testSendTwoTracksWithSameActivity() {
    String testActivity = "(TestActivity)";
    EndToEndTestUtils.activityType = testActivity;
    GoogleUtils.deleteSpreadsheets(
        EndToEndTestUtils.trackListActivity.getApplicationContext(), GoogleUtils.ACCOUNT_1);
    EndToEndTestUtils.createSimpleTrack(1, false);
    boolean result = sendToGoogle(activityMyTracks.getString(R.string.export_google_maps))
        && sendToGoogle(activityMyTracks.getString(R.string.export_google_fusion_tables))
        && sendToGoogle(activityMyTracks.getString(R.string.export_google_spreadsheets));
    result = result && sendToGoogle(activityMyTracks.getString(R.string.export_google_maps))
        && sendToGoogle(activityMyTracks.getString(R.string.export_google_fusion_tables))
        && sendToGoogle(activityMyTracks.getString(R.string.export_google_spreadsheets));

    // Result is true mean has account bound with this device and send
    // successful.
    if (result) {
      List<File> fileList = GoogleUtils.searchSpreadsheets(
          activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1);
      assertEquals(1, fileList.size());
    }
  }

  /**
   * Checks the process of sending track to google.
   */
  private void checkSendTrackToGoogle() {
    if (!sendToGoogle(activityMyTracks.getString(R.string.export_google_maps))
        || !sendToGoogle(activityMyTracks.getString(R.string.export_google_fusion_tables))
        || !sendToGoogle(activityMyTracks.getString(R.string.export_google_spreadsheets))) {
      return;
    }
    // Check whether all data is correct on Google Drive, Maps, Documents, and
    // Spreadsheet.
    // assertTrue(SyncTestUtils.checkFile(EndToEndTestUtils.trackName, true,
    // SyncTestUtils.getGoogleDrive(activityMyTracks.getApplicationContext())));
    assertTrue(GoogleUtils.deleteMaps(
        activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1,
        EndToEndTestUtils.trackName));
    assertTrue(GoogleUtils.deleteFusionTables(
        activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1,
        EndToEndTestUtils.trackName));
    assertTrue(GoogleUtils.deleteSpreadsheetsRow(
        activityMyTracks.getApplicationContext(), GoogleUtils.ACCOUNT_1,
        EndToEndTestUtils.trackName));
  }

  /**
   * Sends a track to Google.
   * 
   * @param exportDestination destination to export
   * @return true means send successfully
   */
  private boolean sendToGoogle(String exportDestination) {
    EndToEndTestUtils.findMenuItem(activityMyTracks.getString(R.string.menu_export), true);
    EndToEndTestUtils.SOLO.waitForText(activityMyTracks.getString(R.string.export_title));
    instrumentation.waitForIdleSync();

    EndToEndTestUtils.SOLO.clickOnText(exportDestination);

    instrumentation.waitForIdleSync();
    EndToEndTestUtils.getButtonOnScreen(activityMyTracks.getString(R.string.menu_export), true,
        true);

    instrumentation.waitForIdleSync();
    if (!GoogleUtils.isAccountAvailable()) {
      return false;
    }

    // Following check the process of "Send to Google", the second condition is
    // in case the export has done.
    assertTrue(EndToEndTestUtils.SOLO.waitForText(activityMyTracks
        .getString(R.string.generic_progress_title))
        || EndToEndTestUtils.SOLO.waitForText(activityMyTracks
            .getString(R.string.generic_success_title)));

    // Waiting the send is finish. Should double check it for the progress
    // dialog may disappear and display again while switch to next send(There
    // are four sends currently, there are send to drive, maps, fusion table and
    // spreadsheet).
    while (EndToEndTestUtils.SOLO.waitForText(
        activityMyTracks.getString(R.string.generic_progress_title), 1,
        EndToEndTestUtils.SHORT_WAIT_TIME)
        || EndToEndTestUtils.SOLO.waitForText(
            activityMyTracks.getString(R.string.generic_progress_title), 1,
            EndToEndTestUtils.SHORT_WAIT_TIME)) {}

    if (exportDestination.endsWith(activityMyTracks.getString(R.string.export_google_maps))) {
      // Check whether the result dialog is display.
      assertTrue(EndToEndTestUtils.SOLO.waitForText(activityMyTracks
          .getString(R.string.share_track_share_url)));
    } else {
      assertTrue(EndToEndTestUtils.SOLO.waitForText(activityMyTracks
          .getString(R.string.generic_success_title)));
    }
    EndToEndTestUtils
        .getButtonOnScreen(activityMyTracks.getString(R.string.generic_ok), true, true);
    return true;
  }

  /**
   * Sends a large track to Google.
   */
  public void testSendLargeTrackToGoogle() {
    EndToEndTestUtils.deleteAllTracks();
    EndToEndTestUtils.createSimpleTrack(200, false);
    instrumentation.waitForIdleSync();
    checkSendTrackToGoogle();
  }

  @Override
  protected void tearDown() throws Exception {
    EndToEndTestUtils.activityType = EndToEndTestUtils.DEFAULT_ACTIVITY_TYPE;
    EndToEndTestUtils.SOLO.finishOpenedActivities();
    super.tearDown();
  }

}