commit 0942b1df911f6d78b0bc4277456de01ebbf9e5ec
author Dan Shi <dshi@chromium.org> Tue Mar 31 11:07:00 2015 -0700
committer ChromeOS Commit Bot <chromeos-commit-bot@chromium.org> Tue Mar 31 21:50:01 2015 +0000
tree ca8f14eaaf25325bf47e73d64c50edef623a3e36
parent 36f638e1e0b3ec6613827e88724d7e45957d83f0

[autotest] Add get_release_version method to servo_host

get_build_id method in autoupdater was removed. Add the new method for
servo_host to use.

BUG=chromium:472135
TEST=local servo test, confirm verify servo works

Change-Id: Ic255dce3b7bfa361204bd6dd7bb03b8f86ec6c5e
Reviewed-on: https://chromium-review.googlesource.com/263245
Trybot-Ready: Dan Shi <dshi@chromium.org>
Tested-by: Dan Shi <dshi@chromium.org>
Reviewed-by: Fang Deng <fdeng@chromium.org>
Commit-Queue: Dan Shi <dshi@chromium.org>

server/hosts/servo_host.py

diff --git a/server/hosts/servo_host.py b/server/hosts/servo_host.py
index cbe6099..65d2af8 100644
--- a/server/hosts/servo_host.py
+++ b/server/hosts/servo_host.py
@@ -18,11 +18,13 @@
 from autotest_lib.client.bin import utils
 from autotest_lib.client.common_lib import error
 from autotest_lib.client.common_lib import global_config
+from autotest_lib.client.common_lib import lsbrelease_utils
 from autotest_lib.client.common_lib.cros import autoupdater
 from autotest_lib.client.common_lib.cros import dev_server
 from autotest_lib.client.common_lib.cros import retry
 from autotest_lib.client.common_lib.cros.graphite import autotest_stats
 from autotest_lib.client.common_lib.cros.network import ping_runner
+from autotest_lib.client.cros import constants as client_constants
 from autotest_lib.server import site_utils as server_site_utils
 from autotest_lib.server.cros.servo import servo
 from autotest_lib.server.cros.dynamic_suite import frontend_wrappers
@@ -374,6 +376,18 @@
                         (self.hostname, e))
 
 
+    def get_release_version(self):
+        """Get the value of attribute CHROMEOS_RELEASE_VERSION from lsb-release.
+
+        @returns The version string in lsb-release, under attribute
+                 CHROMEOS_RELEASE_VERSION.
+        """
+        lsb_release_content = self.run(
+                    'cat "%s"' % client_constants.LSB_RELEASE).stdout.strip()
+        return lsbrelease_utils.get_chromeos_release_version(
+                    lsb_release_content=lsb_release_content)
+
+
     @_timer.decorate
     def _update_image(self):
         """Update the image on the servo host, if needed.
@@ -421,7 +435,7 @@
             url = ds.get_update_url(latest_build)
 
         updater = autoupdater.ChromiumOSUpdater(update_url=url, host=self)
-        current_build_number = updater.get_build_id()
+        current_build_number = self.get_release_version()
         status = updater.check_update_status()
 
         if status == autoupdater.UPDATER_NEED_REBOOT:
@@ -443,7 +457,7 @@
             # raised, so test requires servo should fail.
             self.reboot(**kwargs)
             if self.wait_up(timeout=120):
-                current_build_number = updater.get_build_id()
+                current_build_number = self.get_release_version()
                 logging.info('servo host %s back from reboot, with build %s',
                              self.hostname, current_build_number)
             else: