From 54641cbcd8701fbd45a31cc5bfadcae2deb032f1 Mon Sep 17 00:00:00 2001
From: youtaol <youtaol@localhost>
Date: Fri, 11 Jan 2013 14:54:17 +0800
Subject: [PATCH] Change ExportAndImportTest to make e2e test cases can run
 well with new code change(More import options).

---
 .../endtoendtest/EndToEndTestUtils.java       |  2 +-
 .../endtoendtest/ExportAndImportTest.java     | 48 ++++++++++++++++---
 2 files changed, 42 insertions(+), 8 deletions(-)

diff --git a/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/EndToEndTestUtils.java b/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/EndToEndTestUtils.java
index c15b0c883..97af872d8 100644
--- a/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/EndToEndTestUtils.java
+++ b/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/EndToEndTestUtils.java
@@ -53,7 +53,7 @@ import junit.framework.Assert;
  * @author Youtao Liu
  */
 public class EndToEndTestUtils {
-  public static int emulatorPort = 5556;
+  public static int emulatorPort = 5560;
   // usually 5554.
   private static final int ORIENTATION_PORTRAIT = 1;
   private static final int ORIENTATION_LANDSCAPE = 0;
diff --git a/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/ExportAndImportTest.java b/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/ExportAndImportTest.java
index a18d2229b..0992d747c 100644
--- a/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/ExportAndImportTest.java
+++ b/MyTracksTest_E2E/src/com/google/android/apps/mytracks/endtoendtest/ExportAndImportTest.java
@@ -97,7 +97,7 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
     EndToEndTestUtils.deleteAllTracks();
 
     // Import this tracks.
-    EndToEndTestUtils.findMenuItem(activityMyTracks.getString(R.string.menu_import), true);
+    importTracks(EndToEndTestUtils.GPX);
     EndToEndTestUtils.SOLO.waitForText(getImportSuccessMessage(1, EndToEndTestUtils.GPX));
 
     /*
@@ -158,7 +158,7 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
    * <li>Create two tracks, one of them is empty(Have no Gps data).</li>
    * <li>Tests import when there is no track file.</li>
    * <li>Tests export tracks to Gpx files.</li>
-   * <li>Tests import files to tracks.</li>
+   * <li>Tests import Gpx files to tracks.</li>
    * <li>Tests export tracks to Kml files.</li>
    * </ul>
    */
@@ -177,8 +177,8 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
     // Get track number in current track list of MyTracks.
     int trackNumber = EndToEndTestUtils.SOLO.getCurrentListViews().get(0).getCount();
 
-    // No file to imported.
-    EndToEndTestUtils.findMenuItem(activityMyTracks.getString(R.string.menu_import), true);
+    // No Gpx file to imported.
+    importTracks(EndToEndTestUtils.GPX);
     EndToEndTestUtils.SOLO.waitForText(getImportErrorMessage(0, 0, EndToEndTestUtils.GPX));
     EndToEndTestUtils
         .getButtonOnScreen(activityMyTracks.getString(R.string.generic_ok), true, true);
@@ -192,16 +192,16 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
     EndToEndTestUtils
         .getButtonOnScreen(activityMyTracks.getString(R.string.generic_ok), true, true);
 
-    // Check export file.
+    // Check export Gpx file.
     assertEquals(gpxFilesNumber + trackNumber,
         EndToEndTestUtils.getExportedFiles(EndToEndTestUtils.GPX).length);
     instrumentation.waitForIdleSync();
 
-    // Click to import track.
+    // Click to import Gpx track.
     gpxFilesNumber = EndToEndTestUtils.getExportedFiles(EndToEndTestUtils.GPX).length;
     trackNumber = EndToEndTestUtils.SOLO.getCurrentListViews().get(0).getCount();
 
-    EndToEndTestUtils.findMenuItem(activityMyTracks.getString(R.string.menu_import), true);
+    importTracks(EndToEndTestUtils.GPX);
     EndToEndTestUtils.rotateAllActivities();
 
     /*
@@ -236,6 +236,30 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
     // Check export files.
     assertEquals(gpxFilesNumber, EndToEndTestUtils.getExportedFiles(EndToEndTestUtils.GPX).length);
     assertEquals(trackNumber, EndToEndTestUtils.getExportedFiles(EndToEndTestUtils.KML).length);
+
+    // Click to import KML track.
+    int KMLFilesNumber = EndToEndTestUtils.getExportedFiles(EndToEndTestUtils.KML).length;
+    instrumentation.waitForIdleSync();
+    trackNumber = EndToEndTestUtils.SOLO.getCurrentListViews().get(0).getCount();
+
+    importTracks(EndToEndTestUtils.KML);
+
+    /*
+     * Wait for the prefix of import success string is much faster than wait the
+     * whole string.
+     */
+    EndToEndTestUtils.SOLO.waitForText(activityMyTracks.getString(R.string.import_success).split(
+        "%")[0]);
+
+    /*
+     * Check import KML tracks should be equal with the sum of trackNumber and
+     * KMLFilesNumber;
+     */
+    EndToEndTestUtils
+        .getButtonOnScreen(activityMyTracks.getString(R.string.generic_ok), true, true);
+    instrumentation.waitForIdleSync();
+    assertEquals(trackNumber + KMLFilesNumber, EndToEndTestUtils.SOLO.getCurrentListViews().get(0)
+        .getCount());
   }
 
   @Override
@@ -262,4 +286,14 @@ public class ExportAndImportTest extends ActivityInstrumentationTestCase2<TrackL
     String directoryName = FileUtils.buildExternalDirectoryPath(type);
     return activityMyTracks.getString(R.string.import_error, count, files, directoryName);
   }
+
+  /**
+   * Imports tracks from external storage.
+   * 
+   * @param fileType the file type, can be GPX and KML currently
+   */
+  private void importTracks(String fileType) {
+    EndToEndTestUtils.findMenuItem(activityMyTracks.getString(R.string.menu_import_all), true);
+    EndToEndTestUtils.SOLO.clickOnText(fileType.toUpperCase());
+  }
 }