Enable side-by-side beta and dev channels.

chrome/test/mini_installer/variable_expander.py

 # Copyright 2013 The Chromium 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 base64
 import hashlib
 import os
 import string
 import win32api
 import win32com.client
@@ skipping 43 matching lines @@
 
 
 class VariableExpander:
   """Expands variables in strings."""
 
   def __init__(self, mini_installer_path, next_version_mini_installer_path):
     """Constructor.
 
     The constructor initializes a variable dictionary that maps variables to
     their values. These are the only acceptable variables:
+        * $BRAND: the browser brand (e.g., "Google Chrome" or "Chromium").
         * $CHROME_DIR: the directory of Chrome (or Chromium) from the base
             installation directory.
         * $CHROME_HTML_PROG_ID: 'ChromeHTML' (or 'ChromiumHTM').
         * $CHROME_LONG_NAME: 'Google Chrome' (or 'Chromium').
-        * $CHROME_LONG_NAME_SXS: 'Google Chrome SxS' if $SUPPORTS_SXS.
+        * $CHROME_LONG_NAME_BETA: 'Google Chrome Beta' if $BRAND is 'Google
+        *   Chrome'.
+        * $CHROME_LONG_NAME_DEV: 'Google Chrome Dev' if $BRAND is 'Google
+        *   Chrome'.
+        * $CHROME_LONG_NAME_SXS: 'Google Chrome SxS' if $BRAND is 'Google
+        *   Chrome'.
         * $CHROME_SHORT_NAME: 'Chrome' (or 'Chromium').
-        * $CHROME_SHORT_NAME_SXS: 'ChromeCanary' if $SUPPORTS_SXS.
+        * $CHROME_SHORT_NAME_BETA: 'ChromeBeta' if $BRAND is 'Google Chrome'.
+        * $CHROME_SHORT_NAME_DEV: 'ChromeDev' if $BRAND is 'Google Chrome'.
+        * $CHROME_SHORT_NAME_SXS: 'ChromeCanary' if $BRAND is 'Google Chrome'.
         * $CHROME_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
             key, of Chrome for Google Update.
+        * $CHROME_UPDATE_REGISTRY_SUBKEY_DEV: the registry key, excluding the
+            root key, of Chrome Dev for Google Update.
+        * $CHROME_UPDATE_REGISTRY_SUBKEY_BETA: the registry key, excluding the
+            root key, of Chrome Beta for Google Update.
         * $CHROME_UPDATE_REGISTRY_SUBKEY_SXS: the registry key, excluding the
             root key, of Chrome SxS for Google Update.
         * $LAUNCHER_UPDATE_REGISTRY_SUBKEY: the registry key, excluding the root
-            key, of the app launcher for Google Update if $SUPPORTS_SXS.
+            key, of the app launcher for Google Update if $BRAND is 'Google
+        *   Chrome'.
         * $LOCAL_APPDATA: the unquoted path to the Local Application Data
             folder.
         * $MINI_INSTALLER: the unquoted path to the mini_installer.
         * $MINI_INSTALLER_FILE_VERSION: the file version of $MINI_INSTALLER.
         * $NEXT_VERSION_MINI_INSTALLER: the unquoted path to a mini_installer
             whose version is higher than $MINI_INSTALLER.
         * $NEXT_VERSION_MINI_INSTALLER_FILE_VERSION: the file version of
             $NEXT_VERSION_MINI_INSTALLER.
         * $PROGRAM_FILES: the unquoted path to the Program Files folder.
-        * $SUPPORTS_SXS: a boolean indicating whether or not SxS installs
-            are supported by the mini_installer under test.
         * $USER_SPECIFIC_REGISTRY_SUFFIX: the output from the function
             _GetUserSpecificRegistrySuffix().
         * $VERSION_[XP/SERVER_2003/VISTA/WIN7/WIN8/WIN8_1/WIN10]: a 2-tuple
             representing the version of the corresponding OS.
         * $WINDOWS_VERSION: a 2-tuple representing the current Windows version.
 
     Args:
       mini_installer_path: The path to a mini_installer.
       next_version_mini_installer_path: The path to a mini_installer whose
           version is higher than |mini_installer_path|.
@@ skipping 19 matching lines @@
         'VERSION_WIN7': '(6, 1)',
         'VERSION_WIN8': '(6, 2)',
         'VERSION_WIN8_1': '(6, 3)',
         'VERSION_XP': '(5, 1)',
         'WINDOWS_VERSION': '(%s, %s)' % (windows_major_ver, windows_minor_ver)
     }
 
     mini_installer_product_name = _GetProductName(mini_installer_abspath)
     if mini_installer_product_name == 'Google Chrome Installer':
       self._variable_mapping.update({
+          'BRAND': 'Google Chrome',
           'BINARIES_UPDATE_REGISTRY_SUBKEY': (
             'Software\\Google\\Update\\Clients\\'
             '{4DC8B4CA-1BDA-483e-B5FA-D3C12E15B62D}'),
           'CHROME_DIR': 'Google\\Chrome',
           'CHROME_HTML_PROG_ID': 'ChromeHTML',
           'CHROME_LONG_NAME': 'Google Chrome',
           'CHROME_SHORT_NAME': 'Chrome',
           'CHROME_UPDATE_REGISTRY_SUBKEY': (
             'Software\\Google\\Update\\Clients\\'
             '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
           'CHROME_CLIENT_STATE_KEY': (
             'Software\\Google\\Update\\ClientState\\'
             '{8A69D345-D564-463c-AFF1-A69D9E530F96}'),
-          'SUPPORTS_SXS': True,
+          'CHROME_DIR_BETA': 'Google\\Chrome Beta',
+          'CHROME_DIR_DEV': 'Google\\Chrome Dev',
           'CHROME_DIR_SXS': 'Google\\Chrome SxS',
+          'CHROME_LONG_NAME_BETA': 'Google Chrome Beta',
+          'CHROME_LONG_NAME_DEV': 'Google Chrome Dev',
           'CHROME_LONG_NAME_SXS': 'Google Chrome SxS',
+          'CHROME_SHORT_NAME_BETA': 'ChromeBeta',
+          'CHROME_SHORT_NAME_DEV': 'ChromeDev',
           'CHROME_SHORT_NAME_SXS': 'ChromeCanary',
+          'CHROME_UPDATE_REGISTRY_SUBKEY_BETA': (
+            'Software\\Google\\Update\\Clients\\'
+            '{8237E44A-0054-442C-B6B6-EA0509993955}'),
+          'CHROME_UPDATE_REGISTRY_SUBKEY_DEV': (
+            'Software\\Google\\Update\\Clients\\'
+            '{401C381F-E0DE-4B85-8BD8-3F3F14FBDA57}'),
           'CHROME_UPDATE_REGISTRY_SUBKEY_SXS': (
             'Software\\Google\\Update\\Clients\\'
             '{4ea16ac7-fd5a-47c3-875b-dbf4a2008c20}'),
           'LAUNCHER_UPDATE_REGISTRY_SUBKEY': (
             'Software\\Google\\Update\\Clients\\'
             '{FDA71E6F-AC4C-4a00-8B70-9958A68906BF}')
       })
     elif mini_installer_product_name == 'Chromium Installer':
       self._variable_mapping.update({
+          'BRAND': 'Chromium',
           'BINARIES_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium Binaries',
           'CHROME_DIR': 'Chromium',
           'CHROME_HTML_PROG_ID': 'ChromiumHTM',
           'CHROME_LONG_NAME': 'Chromium',
           'CHROME_SHORT_NAME': 'Chromium',
           'CHROME_UPDATE_REGISTRY_SUBKEY': 'Software\\Chromium',
           'CHROME_CLIENT_STATE_KEY': 'Software\\Chromium',
-          'SUPPORTS_SXS': False
       })
     else:
       raise KeyError("Unknown mini_installer product name '%s'" %
                      mini_installer_product_name)
 
 
   def Expand(self, str):
     """Expands variables in the given string.
 
     This method resolves only variables defined in the constructor. It does not
     resolve environment variables. Any dollar signs that are not part of
     variables must be escaped with $$, otherwise a KeyError or a ValueError will
     be raised.
 
     Args:
       str: A string.
 
     Returns:
       A new string created by replacing variables with their values.
     """
     return string.Template(str).substitute(self._variable_mapping)