OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. | |
3 # | |
4 # Use of this source code is governed by a BSD-style license | |
5 # that can be found in the LICENSE file in the root of the source | |
6 # tree. An additional intellectual property rights grant can be found | |
7 # in the file PATENTS. All contributing project authors may | |
8 # be found in the AUTHORS file in the root of the source tree. | |
9 | |
10 import glob | |
11 import unittest | |
12 from video_analysis import FindUsbPortForV4lDevices | |
13 | |
14 | |
15 class RunVideoAnalysisTest(unittest.TestCase): | |
16 def SetGlobPath(self, path1, path2): | |
17 self.path1 = path1 | |
18 self.path2 = path2 | |
19 | |
20 def setUp(self): | |
21 self.path1 = '' | |
22 self.path2 = '' | |
23 self.request_nbr = 1 | |
24 | |
25 def GlobMock(string): | |
26 # Eat incoming string. | |
27 del string | |
28 if self.request_nbr == 1: | |
29 self.request_nbr += 1 | |
30 return self.path1 | |
31 else: | |
32 self.request_nbr = 1 | |
33 return self.path2 | |
34 | |
35 # Override the glob function with our own that returns a string set by the | |
36 # test. | |
37 glob.glob = GlobMock | |
38 | |
39 # Verifies that the correct USB id is returned. | |
40 def testFindUSBPortForV4lDevices(self): | |
41 short_path1 = ('/sys/bus/usb/devices/usb1/1-1/driver/4-4/4-4:1.0/' | |
42 'video4linux/video0') | |
43 short_path2 = ('/sys/bus/usb/devices/usb1/1-1/driver/4-3/4-3:1.0/' | |
44 'video4linux/video1') | |
45 self.SetGlobPath(short_path1, short_path2) | |
46 short_usb_ids = ['4-4', '4-3'] | |
47 self.assertEqual(FindUsbPortForV4lDevices('video0', 'video1'), | |
48 short_usb_ids) | |
49 | |
50 long_path1 = ('/sys/bus/usb/devices/usb1/1-1/driver/3-3/3-3.1:1.0/' | |
51 'video4linux/video0') | |
52 long_path2 = ('/sys/bus/usb/devices/usb1/1-1/driver/3-2/3-2.1:1.0/' | |
53 'video4linux/video1') | |
54 self.SetGlobPath(long_path1, long_path2) | |
55 long_usb_ids = ['3-3.1', '3-2.1'] | |
56 self.assertEqual(FindUsbPortForV4lDevices('video0', 'video1'), long_usb_ids) | |
57 | |
58 | |
59 def testFindUSBPortForV4lDevicesNoDevice(self): | |
60 no_device_found = ('') | |
61 v4l_device = ('/sys/bus/usb/devices/usb1/1-1/driver/3-2/3-2.1:1.0/' | |
62 'video4linux/video1') | |
63 self.SetGlobPath(no_device_found, v4l_device) | |
64 empty_list = [] | |
65 self.assertEqual(FindUsbPortForV4lDevices('video0', 'video1'), empty_list) | |
66 | |
67 | |
68 if __name__ == "__main__": | |
69 unittest.main() | |
OLD | NEW |