[NaCl SDK] update_nacl_manifest updates canary to the latest available bundle.

native_client_sdk/src/build_tools/tests/update_nacl_manifest_test.py

 #!/usr/bin/env python
 # Copyright (c) 2012 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 copy
 import datetime
 import hashlib
 import logging
 import os
@@ skipping 184 matching lines @@
       # add .json manifest snippet, it should look like a normal Bundle, but
       # only has one archive.
       new_bundle = manifest_util.Bundle('')
       new_bundle.CopyFrom(bundle)
       del new_bundle.archives[:]
       new_bundle.AddArchive(archive)
       self[path + '.json'] = new_bundle.GetDataAsString()
 
 
 class TestDelegate(update_nacl_manifest.Delegate):
-  def __init__(self, manifest, history, files, version_mapping):
+  def __init__(self, manifest, history, files):
     self.manifest = manifest
     self.history = history
     self.files = files
-    self.version_mapping = version_mapping
     self.dryrun = 0
     self.called_gsutil_cp = False
     self.called_sendmail = False
 
   def GetRepoManifest(self):
     return self.manifest
 
   def GetHistory(self):
     return self.history
 
-  def GetTrunkRevision(self, version):
-    return self.version_mapping[version]
-
   def GsUtil_ls(self, url):
     path = GetPathFromGsUrl(url)
     result = []
-    for filename, _ in self.files.iteritems():
-      if filename.startswith(path):
-        result.append(MakeGsUrl(filename))
-    return result
+    for filename in self.files.iterkeys():
+      if not filename.startswith(path):
+        continue
+
+      # Find the first slash after the prefix (path).
+      # +1, because if the slash is directly after path, then we want to find
+      # the following slash anyway.
+      slash = filename.find('/', len(path) + 1)
+
+      if slash != -1:
+        filename = filename[:slash]
+
+      result.append(MakeGsUrl(filename))
+
+    # Remove dupes.
+    return list(set(result))
 
   def GsUtil_cat(self, url):
     path = GetPathFromGsUrl(url)
     if path not in self.files:
       raise subprocess.CalledProcessError(1, 'gsutil cat %s' % (url,))
     return self.files[path]
 
   def GsUtil_cp(self, src, dest, stdin=None):
     self.called_gsutil_cp = True
     dest_path = GetPathFromGsUrl(dest)
@@ skipping 44 matching lines @@
 
 
 class TestUpdateManifest(unittest.TestCase):
   def setUp(self):
     self.history = MakeHistory()
     self.files = MakeFiles()
     self.version_mapping = {}
     self.delegate = None
     self.uploaded_manifest = None
     self.manifest = None
-    # Ignore logging warnings, etc.
-    logging.getLogger('update_nacl_manifest').setLevel(logging.INFO)
 
   def _MakeDelegate(self):
     self.delegate = TestDelegate(self.manifest, self.history.history,
-        self.files, self.version_mapping)
+        self.files)
 
   def _Run(self, host_oses, extra_archives=None, fixed_bundle_versions=None):
     update_nacl_manifest.Run(self.delegate, host_oses, extra_archives,
                              fixed_bundle_versions)
 
   def _HasUploadedManifest(self):
     return 'naclsdk_manifest2.json' in self.files
 
   def _ReadUploadedManifest(self):
     self.uploaded_manifest = manifest_util.SDKManifest()
@@ skipping 162 matching lines @@
     self.files.Add(bundle_string_revision)
     self._MakeDelegate()
     self._Run(OS_MLW)
     self._ReadUploadedManifest()
     uploaded_bundle = self.uploaded_manifest.GetBundle(
         bundle_string_revision.name)
     self.assertEqual(uploaded_bundle.revision, 1234)
     self.assertEqual(uploaded_bundle.version, 18)
 
   def testUpdateCanary(self):
-    # Note that the bundle in naclsdk_manifest2.json will be called
-    # CANARY_BUNDLE_NAME, whereas the bundle in the manifest "snippet" will be
-    # called "pepper_21".
-    canary_bundle = copy.deepcopy(BCANARY_NONE)
-    self.manifest = MakeManifest(canary_bundle)
-    self.history.Add(OS_MW, CANARY, V21_0_1145_0)
-    self.files.Add(B21_0_1145_0_MLW)
+    self.manifest = MakeManifest(copy.deepcopy(BCANARY_NONE))
+    self.files.Add(BTRUNK_140819_MLW)
     self._MakeDelegate()
     self._Run(OS_MLW)
     self._ReadUploadedManifest()
-    self._AssertUploadedManifestHasBundle(B21_0_1145_0_MLW, CANARY,
-                                          bundle_name=CANARY_BUNDLE_NAME)
-
-  def testUpdateCanaryUseTrunkArchives(self):
-    canary_bundle = copy.deepcopy(BCANARY_NONE)
-    self.manifest = MakeManifest(canary_bundle)
-    self.history.Add(OS_MW, CANARY, V21_0_1166_0)
-    self.files.Add(B21_0_1166_0_MW)
-    self.files.Add(BTRUNK_140819_MLW)
-    self.version_mapping[V21_0_1166_0] = VTRUNK_140819
-    self._MakeDelegate()
-    self._Run(OS_MLW)
-    self._ReadUploadedManifest()
-
-    test_bundle = copy.deepcopy(B21_0_1166_0_MW)
-    test_bundle.AddArchive(BTRUNK_140819_MLW.GetArchive('linux'))
-    self._AssertUploadedManifestHasBundle(test_bundle, CANARY,
-                                          bundle_name=CANARY_BUNDLE_NAME)
-
-  def testCanaryUseOnlyTrunkArchives(self):
-    self.manifest = MakeManifest(copy.deepcopy(BCANARY_NONE))
-    history = """win,canary,21.0.1163.0,2012-06-04 12:35:44.784446
-mac,canary,21.0.1163.0,2012-06-04 11:54:09.433166"""
-    self._AddCsvHistory(history)
-    self.version_mapping['21.0.1163.0'] = 'trunk.140240'
-    my_bundle = MakePlatformBundle(21, 140240, '21.0.1163.0', OS_MLW)
-    self.files.Add(my_bundle)
-    self._MakeDelegate()
-    self._Run(OS_MLW)
-    self._ReadUploadedManifest()
-    self._AssertUploadedManifestHasBundle(my_bundle, CANARY,
+    self._AssertUploadedManifestHasBundle(BTRUNK_140819_MLW, CANARY,
                                           bundle_name=CANARY_BUNDLE_NAME)
 
   def testCanaryShouldOnlyUseCanaryVersions(self):
     canary_bundle = copy.deepcopy(BCANARY_NONE)
     self.manifest = MakeManifest(canary_bundle)
     self.history.Add(OS_MW, CANARY, V21_0_1166_0)
     self.history.Add(OS_MW, BETA, V19_0_1084_41)
     self.files.Add(B19_0_1084_41_MLW)
     self.version_mapping[V21_0_1166_0] = VTRUNK_140819
     self._MakeDelegate()
     self.assertRaises(Exception, self._Run, OS_MLW)
 
-  def testMissingCanaryFollowedByStableShouldWork(self):
-    history = """win,canary,21.0.1160.0,2012-06-01 19:44:35.936109
-mac,canary,21.0.1160.0,2012-06-01 18:20:02.003123
-mac,stable,19.0.1084.52,2012-06-01 17:59:21.559710
-win,canary,21.0.1159.2,2012-06-01 02:31:43.877688
-mac,stable,19.0.1084.53,2012-06-01 01:39:57.549149
-win,canary,21.0.1158.0,2012-05-31 20:16:55.615236
-win,canary,21.0.1157.0,2012-05-31 17:41:29.516013
-mac,canary,21.0.1158.0,2012-05-31 17:41:27.591354
-mac,beta,20.0.1132.21,2012-05-30 23:45:38.535586
-linux,beta,20.0.1132.21,2012-05-30 23:45:37.025015
-cf,beta,20.0.1132.21,2012-05-30 23:45:36.767529
-win,beta,20.0.1132.21,2012-05-30 23:44:56.675123
-win,canary,21.0.1156.1,2012-05-30 22:28:01.872056
-mac,canary,21.0.1156.1,2012-05-30 21:20:29.920390
-win,canary,21.0.1156.0,2012-05-30 12:46:48.046627
-mac,canary,21.0.1156.0,2012-05-30 12:14:21.305090"""
-    self.manifest = MakeManifest(copy.deepcopy(BCANARY_NONE))
-    self._AddCsvHistory(history)
-    self.version_mapping = {
-        '21.0.1160.0': 'trunk.139984',
-        '21.0.1159.2': 'trunk.139890',
-        '21.0.1158.0': 'trunk.139740',
-        '21.0.1157.0': 'unknown',
-        '21.0.1156.1': 'trunk.139576',
-        '21.0.1156.0': 'trunk.139984'}
-    self.files.Add(MakePlatformBundle(21, 139890, '21.0.1159.2', OS_MLW))
-    self.files.Add(MakePlatformBundle(21, 0, '21.0.1157.1', ('linux', 'win')))
-    my_bundle = MakePlatformBundle(21, 139576, '21.0.1156.1', OS_MLW)
-    self.files.Add(my_bundle)
-    self._MakeDelegate()
-    self._Run(OS_MLW)
-    self._ReadUploadedManifest()
-    self._AssertUploadedManifestHasBundle(my_bundle, CANARY,
-                                          bundle_name=CANARY_BUNDLE_NAME)
-
   def testExtensionWorksAsBz2(self):
     # Allow old bundles with just .bz2 extension to work
     self.manifest = MakeManifest(B18_NONE)
     self.history.Add(OS_MLW, BETA, V18_0_1025_163)
     bundle = copy.deepcopy(B18_0_1025_163_MLW)
     archive_url = bundle.GetArchive('mac').url
     bundle.GetArchive('mac').url = archive_url.replace('.tar', '')
     self.files.Add(bundle)
     self._MakeDelegate()
     self._Run(OS_MLW)
@@ skipping 146 matching lines @@
 
     uploaded_bundle = self.uploaded_manifest.GetBundle('pepper_18')
     self.assertEqual(uploaded_bundle, B18_0_1025_163_MLW)
 
   def testBundleWithoutHistoryOrOnlineRaises(self):
     self.manifest = MakeManifest(B18_NONE)
     self._MakeDelegate()
     self.assertRaises(update_nacl_manifest.UnknownLockedBundleException,
                       self._Run, OS_MLW)
 
-  def testIgnoreLastDigitOnCanary(self):
-    # The final number in a canary build does not include any different
-    # changes, it is just a different experiment (e.g. using ASAN, or using
-    # aura). We should not compare these versions differently.
-    #
-    # Note that the version mapping will show that 31.0.1608.0 is different
-    # from 31.0.1608.1 -- this is because 31.0.1608.1 is built from the branch,
-    # not from trunk. Inspecting the branch would show that there are no
-    # changes (why would there be? No one has any reason to merge changes to a
-    # canary branch.)
-    self.manifest = MakeManifest(copy.deepcopy(BCANARY_NONE))
-    history = """win,canary,31.0.1608.1,2013-08-22 09:33:24.469760
-mac,canary,31.0.1608.0,2013-08-22 07:18:09.762600"""
-    self._AddCsvHistory(history)
-    self.version_mapping['31.0.1608.1'] = 'trunk.218914'
-    self.version_mapping['31.0.1608.0'] = 'trunk.218872'
-    my_bundle = MakePlatformBundle(31, 218872, '31.0.1608.0', OS_MLW)
-    self.files.Add(my_bundle)
-    self._MakeDelegate()
-    self._Run(OS_MLW)
-    self._ReadUploadedManifest()
-    self._AssertUploadedManifestHasBundle(my_bundle, CANARY,
-                                          bundle_name=CANARY_BUNDLE_NAME)
-
-  def testDontIgnoreLastDigitForNonCanary(self):
-    self.manifest = MakeManifest(B26_NONE)
-    self.history.Add(OS_M, BETA, V26_0_1386_1)  # Only Mac
-    self.history.Add(OS_LW, BETA, V26_0_1386_0)  # Only Linux, Windows.
-    self.files.Add(B26_0_1386_0_MLW)
-
-    self._MakeDelegate()
-    # This raises because pepper_26 is not found in the history, and therefore
-    # "locked", but it also doesn't have an online version, therefore there is
-    # no good version number to upload.
-    #
-    # Basically we're asserting that 26.0.1386.1 != 26.0.1386.0, which would be
-    # true if it were canary.
-    self.assertRaises(update_nacl_manifest.UnknownLockedBundleException,
-                      self._Run, OS_MLW)
-
   def testUpdateBionic(self):
     bionic_bundle = copy.deepcopy(BBIONIC_NONE)
     self.manifest = MakeManifest(bionic_bundle)
     self.history.Add(OS_MW, CANARY, V37_0_2054_0)
     self.files.Add(BBIONIC_TRUNK_277776)
     self.version_mapping[V37_0_2054_0] = VTRUNK_277776
     self._MakeDelegate()
     self._Run(OS_MLW)
     self._ReadUploadedManifest()
     self._AssertUploadedManifestHasBundle(BBIONIC_TRUNK_277776, CANARY,
@@ skipping 31 matching lines @@
     self.assertTrue('size' not in archive)
     self.assertTrue('checksum' not in archive)
     self.assertRaises(manifest_util.Error, manifest.Validate)
 
     manifest.Validate(add_missing_info=True)
 
     self.assertEqual(archive['size'], self.data_len)
     self.assertEqual(archive['checksum']['sha1'], self.sha1)
 
 
-class TestRealDelegate(unittest.TestCase):
-  def setUp(self):
-    self.delegate = update_nacl_manifest.RealDelegate()
+if __name__ == '__main__':
+  logging.basicConfig(level=logging.CRITICAL)
+  # Uncomment the following line to enable more debugging info.
+  # logging.getLogger('update_nacl_manifest').setLevel(logging.INFO)
 
-  def testGetTrunkRevision(self):
-    revision_dict = {
-      '21.0.1180.80': '151582',
-      '23.0.1271.89': '167132',
-      '24.0.1305.4': '164971',
-    }
-    for version, revision in revision_dict.iteritems():
-      self.assertEqual('trunk.%s' % revision,
-                       self.delegate.GetTrunkRevision(version))
-
-
-if __name__ == '__main__':
-  logging.basicConfig(level=logging.INFO)
   sys.exit(unittest.main())