commit e30011cdf0a42972b8ac2288f042379b623bcc09
author Luke Huang <huangluke@google.com> Mon Mar 18 23:00:59 2019 -0700
committer android-build-merger <android-build-merger@google.com> Mon Mar 18 23:00:59 2019 -0700
tree 0887bdcf9935906da43ba946b093a18f801482a3
parent 9ba36c075d66cf8d11aec32d91962d9fd891e283
parent 24c8d2ab38ae4a78d1223e320ec5597c8d784d76

Merge "Alter CTS tests for change of async DNS API"
am: 54661bb3c1

Change-Id: Ia54bc057099c3767600dcc37830dbdeed76da11d

tests/cts/net/src/android/net/cts/DnsResolverTest.java

diff --git a/tests/cts/net/src/android/net/cts/DnsResolverTest.java b/tests/cts/net/src/android/net/cts/DnsResolverTest.java
index 643d542..6fbe586 100644
--- a/tests/cts/net/src/android/net/cts/DnsResolverTest.java
+++ b/tests/cts/net/src/android/net/cts/DnsResolverTest.java
@@ -40,6 +40,7 @@
 import java.util.ArrayList;
 import java.util.List;
 import java.util.concurrent.CountDownLatch;
+import java.util.concurrent.Executor;
 import java.util.concurrent.TimeUnit;
 import java.util.concurrent.atomic.AtomicReference;
 
@@ -51,14 +52,14 @@
     static final int TIMEOUT_MS = 12_000;
 
     private ConnectivityManager mCM;
-    private Handler mHandler;
+    private Executor mExecutor;
     private DnsResolver mDns;
 
     protected void setUp() throws Exception {
         super.setUp();
         mCM = (ConnectivityManager) getContext().getSystemService(Context.CONNECTIVITY_SERVICE);
-        mHandler = new Handler(Looper.getMainLooper());
         mDns = DnsResolver.getInstance();
+        mExecutor = new Handler(Looper.getMainLooper())::post;
     }
 
     private static String byteArrayToHexString(byte[] bytes) {
@@ -116,7 +117,8 @@
                     fail(msg + e.getMessage());
                 }
             };
-            mDns.query(network, dname, CLASS_IN, TYPE_A, FLAG_NO_CACHE_LOOKUP, mHandler, callback);
+            mDns.query(network, dname, CLASS_IN, TYPE_A, FLAG_NO_CACHE_LOOKUP,
+                    mExecutor, callback);
             try {
                 assertTrue(msg + " but no valid answer after " + TIMEOUT_MS + "ms.",
                         latch.await(TIMEOUT_MS, TimeUnit.MILLISECONDS));
@@ -208,7 +210,7 @@
         for (Network network : getTestableNetworks()) {
             final RawAnswerCallbackImpl callback = new RawAnswerCallbackImpl(msg);
             mDns.query(network, dname, CLASS_IN, TYPE_AAAA, FLAG_NO_CACHE_LOOKUP,
-                    mHandler, callback);
+                    mExecutor, callback);
             try {
                 assertTrue(msg + " but no answer after " + TIMEOUT_MS + "ms.",
                         callback.waitForAnswer());
@@ -236,7 +238,7 @@
         final String msg = "Query with RawAnswerCallback " + byteArrayToHexString(blob);
         for (Network network : getTestableNetworks()) {
             final RawAnswerCallbackImpl callback = new RawAnswerCallbackImpl(msg);
-            mDns.query(network, blob, FLAG_NO_CACHE_LOOKUP, mHandler, callback);
+            mDns.query(network, blob, FLAG_NO_CACHE_LOOKUP, mExecutor, callback);
             try {
                 assertTrue(msg + " but no answer after " + TIMEOUT_MS + "ms.",
                         callback.waitForAnswer());
@@ -274,7 +276,7 @@
                 }
             };
             mDns.query(network, dname, CLASS_IN, TYPE_AAAA, FLAG_NO_CACHE_LOOKUP,
-                    mHandler, callback);
+                    mExecutor, callback);
             try {
                 assertTrue(msg + "but no answer after " + TIMEOUT_MS + "ms.",
                         latch.await(TIMEOUT_MS, TimeUnit.MILLISECONDS));
@@ -311,7 +313,7 @@
                 }
             };
             mDns.query(network, dname, CLASS_IN, TYPE_AAAA, FLAG_NO_CACHE_LOOKUP,
-                    mHandler, callback);
+                    mExecutor, callback);
             try {
                 assertTrue(msg + " but no answer after " + TIMEOUT_MS + "ms.",
                         latch.await(TIMEOUT_MS, TimeUnit.MILLISECONDS));