Support dev builds.

client/common_lib/cros/autoupdater.py

 # Copyright (c) 2010 The Chromium OS Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 import httplib
 import logging
 import re
 import socket
 import urlparse
 
@@ skipping 116 matching lines @@
             # the update as failed, and keep the same rootfs after
             # reboot.
             self.revert_boot_partition()
             raise ChromiumOSError('stateful_update failed on %s.' %
                                   self.host.hostname)
         return True
 
 
     def check_version(self):
         booted_version = self.get_build_id()
+        if not booted_version:
+            booted_version = self.get_dev_build_id()
         if not booted_version in self.update_version:
             logging.error('Expected Chromium OS version: %s.'
                           'Found Chromium OS %s',
                           self.update_version, booted_version)
             raise ChromiumOSError('Updater failed on host %s' %
                                   self.host.hostname)
         else:
             return True
 
 
     def get_build_id(self):
         """Turns the CHROMEOS_RELEASE_DESCRIPTION into a string that
         matches the build ID."""
-        # TODO(seano): handle dev build naming schemes.
         version = self._run('grep CHROMEOS_RELEASE_DESCRIPTION'
                             ' /etc/lsb-release').stdout
         build_re = (r'CHROMEOS_RELEASE_DESCRIPTION='
                     '(\d+\.\d+\.\d+\.\d+) \(\w+ \w+ (\w+)(.*)\)')
         version_match = re.match(build_re, version)
-        if not version_match:
-            raise ChromiumOSError('Unable to get build ID from %s. Found "%s"',
-                                  self.host.hostname, version)
-        version, build_id, builder = version_match.groups()
-        build_match = re.match(r'.*: (\d+)', builder)
-        if build_match:
-            builder_num = '-b%s' % build_match.group(1)
-        else:
-            builder_num = ''
-        return '%s-r%s%s' % (version, build_id, builder_num)
+        if version_match:
+            version, build_id, builder = version_match.groups()
+            build_match = re.match(r'.*: (\d+)', builder)
+            if build_match:
+                builder_num = '-b%s' % build_match.group(1)
+            else:
+                builder_num = ''
+            return '%s-r%s%s' % (version, build_id, builder_num)
+
+
+    def get_dev_build_id(self):
+        """Pulls the CHROMEOS_RELEASE_VERSION string from /etc/lsb-release."""
+        return self._run('grep CHROMEOS_RELEASE_VERSION'
+                         ' /etc/lsb-release').stdout.split('=')[1].strip()