commit acc404bf7dba954cbf9d321f3b925b768892ef3c
author chojoyce <chojoyce@google.com> Tue Jun 25 16:06:45 2019 +0800
committer chojoyce <chojoyce@google.com> Wed Jul 10 13:51:37 2019 +0800
tree a00f2591f59d0beb76d673f4414869d6fd9ca89c
parent d09787afb225bc110ff3b9f66df31dca78960681

Fix error for creating gce AVD

Bug: 135962730
Test: acloud-dev create --avd-type gce --build-id 5108589 --branch git_master

Change-Id: I6d54d87093ee432d2b461e1b87cae6a3583f2494

public/device_driver_test.py

diff --git a/public/device_driver_test.py b/public/device_driver_test.py
index 60b0fb1..d561bb2 100644
--- a/public/device_driver_test.py
+++ b/public/device_driver_test.py
@@ -74,8 +74,8 @@
             return_value=self.compute_client)
         self.Patch(auth, "CreateCredentials", return_value=mock.MagicMock())
 
-    def testCreateAndroidVirtualDevices(self):
-        """Test CreateAndroidVirtualDevices."""
+    def testCreateGCETypeAVD(self):
+        """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")
@@ -98,7 +98,7 @@
         self.compute_client.GetDataDiskName.return_value = disk_name
 
         # Verify
-        report = device_driver.CreateAndroidVirtualDevices(
+        report = device_driver.CreateGCETypeAVD(
             cfg, fake_build_target, fake_build_id)
         self.build_client.CopyTo.assert_called_with(
             fake_build_target, fake_build_id, artifact_name=cfg.disk_image_name,
@@ -130,8 +130,8 @@
         self.assertEquals(report.status, "SUCCESS")
 
     # pylint: disable=invalid-name
-    def testCreateAndroidVirtualDevicesInternalIP(self):
-        """Test CreateAndroidVirtualDevices with internal IP."""
+    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_instance = "fake-instance"
@@ -141,7 +141,7 @@
         self.compute_client.GetInstanceIP.return_value = fake_ip
         self.compute_client.GenerateInstanceName.return_value = fake_instance
 
-        report = device_driver.CreateAndroidVirtualDevices(
+        report = device_driver.CreateGCETypeAVD(
             cfg, fake_build_target, fake_build_id, report_internal_ip=True)
 
         self.assertEquals(