commit 140b6b4ce2d72e5370753dd6deab284e2a541c12
author herbertxue <herbertxue@google.com> Mon Oct 14 20:24:10 2019 +0800
committer herbertxue <herbertxue@google.com> Thu Oct 24 14:31:59 2019 +0800
tree 2c8611f7475660e020b543568a406140daa66f06
parent 9cc5d7ae8ad30b724d8141caa0efe5a6fad1a7bf

Deprecate the named tuple gcompute.IP.

Replace gcompute.IP by ssh.IP object.

Bug: 142095457
Test: acloud-dev create
Change-Id: Iff25d98f888f9ed4ed818a69f911878fc5d0dac0

public/device_driver_test.py

diff --git a/public/device_driver_test.py b/public/device_driver_test.py
index 42bb5c9..237d3ea 100644
--- a/public/device_driver_test.py
+++ b/public/device_driver_test.py
@@ -29,8 +29,8 @@
 from acloud.internal.lib import android_build_client
 from acloud.internal.lib import android_compute_client
 from acloud.internal.lib import driver_test_lib
-from acloud.internal.lib import gcompute_client
 from acloud.internal.lib import gstorage_client
+from acloud.internal.lib import ssh
 from acloud.public import device_driver
 
 
@@ -81,7 +81,7 @@
         """Test CreateGCETypeAVD."""
         cfg = _CreateCfg()
         fake_gs_url = "fake_gs_url"
-        fake_ip = gcompute_client.IP(external="140.1.1.1", internal="10.1.1.1")
+        fake_ip = ssh.IP(external="140.1.1.1", internal="10.1.1.1")
         fake_instance = "fake-instance"
         fake_image = "fake-image"
         fake_build_target = "fake_target"
@@ -136,7 +136,7 @@
     def testCreateGCETypeAVDInternalIP(self):
         """Test CreateGCETypeAVD with internal IP."""
         cfg = _CreateCfg()
-        fake_ip = gcompute_client.IP(external="140.1.1.1", internal="10.1.1.1")
+        fake_ip = ssh.IP(external="140.1.1.1", internal="10.1.1.1")
         fake_instance = "fake-instance"
         fake_build_target = "fake_target"
         fake_build_id = "12345"