1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
#!/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(os.pardir, os.pardir, 'data', 'media', 'csv',
'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), 240,
'total number of elements in the returned list is wrong')
# Get all media data.
media_test_matrix_list = matrix.GenerateAllMediaInfosInCompactForm(False)
self.assertEqual(len(media_test_matrix_list), 449,
'total number of elements in the returned list is wrong')
mobile_test_info = MediaTestMatrix.LookForMediaInfoInCompactFormByNickName(
media_test_matrix_list, 'mobile0.webm')
self.assertEqual(mobile_test_info[2], 'mobile0.webm',
'the nickname of the returned element is wrong')
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), 240,
'total number of elements in the returned list is wrong')
# Get all media data.
media_test_matrix_list = matrix.GenerateAllMediaInfos(False)
self.assertEqual(len(media_test_matrix_list), 449,
'total number of elements in the returned list is wrong')
(info, url, link, tag, is_video, nickname) = (
matrix.GenerateRandomMediaInfo())
self.assertTrue(info is not None,
'randomly generated media info does not have spec')
self.assertTrue(url is not None,
'randomly generated media info does not have url')
self.assertTrue(link is not None,
'randomly generated media info does not have link')
self.assertTrue(tag is not None,
'randomly generated media info does not have tag')
self.assertTrue(nickname is not None,
'randomly generated media info does not have nickname')
if __name__ == '__main__':
unittest.main()
|
