commit 8edbb65cf2d736560c1fafbdb429778237626b8c
author Girish Moturu <gmoturu@google.com> Fri Sep 04 13:22:53 2020 -0700
committer Girish Moturu <gmoturu@google.com> Fri Sep 04 13:23:52 2020 -0700
tree cee767c996f11a8c152d4713680be84d0ef8cf45
parent a25a0b1cc53b0a5fb18a13b62805698b5593620e

[IpSecTest] Add missing import time

Cherry picked from aosp

Bug: 163818569
Test: None
Merged-In: Ic234fbaf79a2bd626252189b0891d55085a93861
Change-Id: I07fb2e28b161addce8d071d754fbb11b52a7ad9b

acts/tests/google/net/IpSecTest.py

diff --git a/acts/tests/google/net/IpSecTest.py b/acts/tests/google/net/IpSecTest.py
index 5173110..1b4a144 100644
--- a/acts/tests/google/net/IpSecTest.py
+++ b/acts/tests/google/net/IpSecTest.py
@@ -25,6 +25,7 @@
 from acts.test_utils.wifi import wifi_test_utils as wutils
 
 import random
+import time
 
 WLAN = "wlan0"
 WAIT_FOR_IP = 15
@@ -44,10 +45,17 @@
         wutils.connect_to_wifi_network(self.dut_b, self.wifi_network)
         time.sleep(WAIT_FOR_IP)
 
-        self.ipv4_dut_a = self.dut_a.droid.connectivityGetIPv4Addresses(WLAN)[0]
-        self.ipv4_dut_b = self.dut_b.droid.connectivityGetIPv4Addresses(WLAN)[0]
-        self.ipv6_dut_a = self.dut_a.droid.connectivityGetIPv6Addresses(WLAN)[0]
-        self.ipv6_dut_b = self.dut_b.droid.connectivityGetIPv6Addresses(WLAN)[0]
+        try:
+            self.ipv4_dut_a = self.dut_a.droid.connectivityGetIPv4Addresses(
+                WLAN)[0]
+            self.ipv4_dut_b = self.dut_b.droid.connectivityGetIPv4Addresses(
+                WLAN)[0]
+            self.ipv6_dut_a = self.dut_a.droid.connectivityGetIPv6Addresses(
+                WLAN)[0]
+            self.ipv6_dut_b = self.dut_b.droid.connectivityGetIPv6Addresses(
+                WLAN)[0]
+        except Exception as e:
+            asserts.abort_class("Failed to get IPv4/IPv6 address: %s" % e)
 
         self.crypt_auth_combos = iutils.generate_random_crypt_auth_combo()