uitest-automator: Use ddmlib for device communications.

Change-Id: I2983399832a8f64b8ac0447587233baba4774311

1 files changed, 12 insertions, 2 deletions

diff --git a/ddms/libs/ddmlib/src/com/android/ddmlib/CollectingOutputReceiver.java b/ddms/libs/ddmlib/src/com/android/ddmlib/CollectingOutputReceiver.java
index cb4612f..80aa8e1 100644
--- a/ddms/libs/ddmlib/src/com/android/ddmlib/CollectingOutputReceiver.java
+++ b/ddms/libs/ddmlib/src/com/android/ddmlib/CollectingOutputReceiver.java
@@ -17,16 +17,24 @@ package com.android.ddmlib;
 
 
 import java.io.UnsupportedEncodingException;
+import java.util.concurrent.CountDownLatch;
 
 /**
  * A {@link IShellOutputReceiver} which collects the whole shell output into one
  * {@link String}.
  */
 public class CollectingOutputReceiver implements IShellOutputReceiver {
-
+    private CountDownLatch mCompletionLatch;
     private StringBuffer mOutputBuffer = new StringBuffer();
     private boolean mIsCanceled = false;
 
+    public CollectingOutputReceiver() {
+    }
+
+    public CollectingOutputReceiver(CountDownLatch commandCompleteLatch) {
+        mCompletionLatch = commandCompleteLatch;
+    }
+
     public String getOutput() {
         return mOutputBuffer.toString();
     }
@@ -68,6 +76,8 @@ public class CollectingOutputReceiver implements IShellOutputReceiver {
      */
     @Override
     public void flush() {
-        // ignore
+        if (mCompletionLatch != null) {
+            mCompletionLatch.countDown();
+        }
     }
 }