Initial checkin of media test matrix class, which will be used for media perf test later.

chrome/test/functional/media_test_matrix_unittest.py

+#!/usr/bin/python
+
+# Copyright (c) 2011 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 os
+import random
+import unittest
+from media_test_matrix import MediaTestMatrix
+
+
+class MediaTestMatrixTest(unittest.TestCase):
+  """Unit test for MediaTestMatrix class."""
+
+  def __ReadCSVFile(self):
+    """Reads CSV file for test and returns a media test matrix object.
+
+    Returns:
+      a media test matrix.
+    """
+    matrix = MediaTestMatrix()
+    # Pyauto.DataDir() method is not used to avoid dependency to pyauto.
+    data_file = os.path.join('..', 'data', 'media', 'media_matrix_data.csv')
+    matrix.ReadData(data_file)
+    return matrix
+
+  def testGenerateAllMediaInfosInCompactForm(self):
+    matrix = self.__readCSVFile()
+    # Get video data only case.
+    media_test_matrix_list = matrix.GenerateAllMediaInfosInCompactForm(True)
+    self.assertEqual(len(media_test_matrix_list), 249)

[dennis_jeffrey, 2011/03/18 23:51:41]

In all assert function calls, I believe there's an optional parameter called
"msg" which you can specify to give an informative error message in the event
that the assertion fails.  You may want to consider adding that to all asserts
in this file, so that more informative error messages can be printed when there
are failures.

[imasaki1, 2011/03/19 01:00:38]

Done.

+    media_test_matrix_list = matrix.GenerateAllMediaInfosInCompactForm(False)
+    # Get all media data.

[dennis_jeffrey, 2011/03/18 23:51:41]

Move this comment right before line 33 above.

[imasaki1, 2011/03/19 01:00:38]

Done.

+    self.assertEqual(len(media_test_matrix_list), 466)
+    mobile_test_info = MediaTestMatrix.LookForMediaInfoInCompactFormByNickName(
+        media_test_matrix_list, 'mobile0.webm')
+    self.assertEqual(len(media_test_matrix_list), 1)
+    self.assertEqual(mobile_test_info[1], 'mobile0.webm')
+
+  def testGenerateAllMediaInfos(self):
+    matrix = self.__readCSVFile()
+    # Get video data only case.
+    media_test_matrix_list = matrix.GenerateAllMediaInfos(True)
+    self.assertEqual(len(media_test_matrix_list), 249)
+    # Get all media data.
+    media_test_matrix_list = matrix.GenerateAllMediaInfos(False)
+    self.assertEqual(len(media_test_matrix_list), 466)
+    (info, url, link, tag, is_video, nickname) = (
+        matrix.GenerateRandomMediaInfo())
+    self.assertTrue(info is not None)

[dennis_jeffrey, 2011/03/18 23:51:41]

Should we assert on any of the other return values?  (url, link, tag, is_video,
nickname)?

[imasaki1, 2011/03/19 01:00:38]

Done.

+
+if __name__ == '__main__':
+    unittest.main()