[WPR] Implement platform dependent recordings for WPR archives.

telemetry/telemetry/wpr/archive_info_unittest.py

 # Copyright 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 json
 import os
 import shutil
 import tempfile
 import unittest
 
 from py_utils import cloud_storage  # pylint: disable=import-error
 
 from telemetry.page import page
 from telemetry.testing import system_stub
 from telemetry.wpr import archive_info
+from telemetry.wpr import archive_info2
 
 
 class MockPage(page.Page):
   def __init__(self, url, name=None):
     super(MockPage, self).__init__(url, None, name=name)
 
 
 page1 = MockPage('http://www.foo.com/', 'Foo')
 page2 = MockPage('http://www.bar.com/', 'Bar')
 page3 = MockPage('http://www.baz.com/')
@@ skipping 195 matching lines @@
     self.assertTrue(os.path.exists(new_recording))
     self.assertFalse(os.path.exists(new_temp_recording))
     self.assertCorrectHashFile(new_recording)
 
     # Check that the archive info was written correctly.
     self.assertTrue(os.path.exists(self.story_set_archive_info_file))
     read_archive_info = archive_info.WprArchiveInfo.FromFile(
         self.story_set_archive_info_file, cloud_storage.PUBLIC_BUCKET)
     self.assertEquals(new_recording,
                       read_archive_info.WprFilePathForStory(page1))
+
+  def testFromFileMultiplatform(self):
+    multiplatform_archive_info_contents = ("""
+        {
+            "platform_specific": true,
+            "archives": {
+                "%s": ["%s", "%s"],
+                "%s": ["%s"]
+            }
+        }
+    """ % (recording1, page1.display_name, page2.display_name, recording2,
+           page3.display_name))
+
+    archive_info_file = os.path.join(
+        self.tmp_dir, 'info2.json')
+    with open(archive_info_file, 'w') as f:
+      f.write(multiplatform_archive_info_contents)
+
+    archive = archive_info.WprArchiveInfo.FromFile(
+        archive_info_file, cloud_storage.PUBLIC_BUCKET)
+    self.assertTrue(isinstance(archive, archive_info2.WprArchiveInfo))