OLD | NEW |
---|---|
(Empty) | |
1 #!/usr/bin/env python | |
kjellander_webrtc
2017/04/04 14:44:23
Rename this file to run_video_analysis_test.py
It
| |
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 unittest | |
11 from run_video_analysis import FindUsbPortForV4lDevices | |
12 import glob | |
kjellander_webrtc
2017/04/04 14:44:23
sort "import glob" before "import unittest"
| |
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.requestNbr = 1 | |
24 | |
25 def glob_mock(): | |
26 if self.requestNbr == 1: | |
27 self.requestNbr += 1 | |
28 return self.path1 | |
29 else: | |
30 self.requestNbr = 1 | |
31 return self.path2 | |
32 | |
33 # Override the glob function with our own that returns a string set by the | |
34 # test. | |
35 glob.glob = glob_mock | |
36 | |
37 # Verifies that the correct USB id is returned. | |
38 def testFindUSBPortForV4lDevices(self): | |
39 short_path1 = ('/sys/bus/usb/devices/usb1/1-1/driver/4-4/4-4:1.0/' | |
40 'video4linux/video0') | |
41 short_path2 = ('/sys/bus/usb/devices/usb1/1-1/driver/4-3/4-3:1.0/' | |
42 'video4linux/video1') | |
43 self.setGlobPath(short_path1, short_path2) | |
44 short_usb_ids = ['4-4', '4-3'] | |
45 self.assertEqual(FindUsbPortForV4lDevices('video0', 'video1'), | |
46 short_usb_ids) | |
47 | |
48 long_path1 = ('/sys/bus/usb/devices/usb1/1-1/driver/3-3/3-3.1:1.0/' | |
49 'video4linux/video0') | |
50 long_path2 = ('/sys/bus/usb/devices/usb1/1-1/driver/3-2/3-2.1:1.0/' | |
51 'video4linux/video1') | |
52 self.setGlobPath(long_path1, long_path2) | |
53 long_usb_ids = ['3-3.1', '3-2.1'] | |
54 self.assertEqual(FindUsbPortForV4lDevices('video0', 'video1'), long_usb_ids) | |
55 | |
56 | |
57 if __name__ == "__main__": | |
58 unittest.main() | |
OLD | NEW |