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