| import unittest |
| import mock |
| |
| from autotest_lib.client.common_lib.cros.cfm import cras_node_collector |
| |
| # pylint: disable=missing-docstring |
| |
| class CrasNodeCollectorTest(unittest.TestCase): |
| """Unit tests for cras_node_collector.""" |
| |
| def test_collect_output(self): |
| output_devices = ( |
| 'Output Devices:\n' |
| 'ID Name\n' |
| '13 HDA Intel PCH: ALC283 Analog:3,0\n' |
| '12 HDA Intel HDMI: HDMI 2:2,8\n' |
| '11 HDA Intel HDMI: HDMI 1:2,7\n' |
| '10 HDA Intel HDMI: HDMI 0:2,3\n' |
| '8 Jabra SPEAK 410 USB: USB Audio:1,0\n' |
| '6 Jabra SPEAK 410 USB: USB Audio:0,0\n' |
| 'Output Nodes:') |
| |
| output_nodes = ( |
| 'Output Nodes:\n' |
| 'Stable Id ID Vol Plugged L/R swapped Time Hotword ' |
| 'Type Name\n' |
| '(d5da8553) 13:0 100 no no 0 ' |
| 'HEADPHONE Headphone Jack\n' |
| '(5280b26d) 12:0 100 no no 0 ' |
| 'HDMI HDMI/DP,pcm=8 Jack\n' |
| '(78297d51) 11:0 100 no no 0 ' |
| 'HDMI HDMI/DP,pcm=7 Jack\n' |
| '(f2070728) 10:0 100 no no 1511264933 ' |
| 'HDMI HDMI/DP,pcm=3 Jack\n' |
| '(2fb77ef4) 10:1 100 no no 0 ' |
| 'HDMI HDMI\n' |
| '(051d1a79) 8:0 75 yes no 1511264933 ' |
| 'USB *(default)\n' |
| '(051d1a79) 6:0 100 yes no 1511264933 ' |
| 'USB (default)\n' |
| 'Input Devices:') |
| |
| mock_host = mock.Mock() |
| mock_host.run_output.side_effect = [output_devices, output_nodes] |
| collector = cras_node_collector.CrasNodeCollector(mock_host) |
| |
| nodes = collector.get_output_nodes() |
| self.assertEquals(7, len(nodes)) |
| node_ids = set([node.node_id for node in nodes]) |
| node_types = set([node.node_type for node in nodes]) |
| device_names = set([node.device_name for node in nodes]) |
| self.assertEquals(node_ids, set( |
| ['13:0', '12:0', '11:0', '10:0', '10:1', '8:0', '6:0'])) |
| self.assertEquals(node_types, set(['HEADPHONE', 'USB', 'HDMI' ])) |
| |
| def test_collect_input(self): |
| input_devices = ( |
| 'Input Devices:\n' |
| 'ID Name\n' |
| '14 HDA Intel PCH: ALC283 Analog:3,0\n' |
| '9 Jabra SPEAK 410 USB: USB Audio:1,0\n' |
| '7 Jabra SPEAK 410 USB: USB Audio:0,0\n' |
| '5 Post DSP Loopback\n' |
| '4 Post Mix Pre DSP Loopback\n' |
| 'Input Nodes:' |
| ) |
| input_nodes = ( |
| 'Input Nodes:\n' |
| 'Stable Id ID Gain Plugged L/R swapped Time Hotword ' |
| 'Type Name\n' |
| '(ca711b33) 14:0 0 no no 0 ' |
| 'MIC Mic Jack\n' |
| '(4b5d44d7) 9:0 0 yes no 1511264933 ' |
| 'USB Jabra SPEAK 410 USB: USB Audio:1,0: Mic\n' |
| '(ca24f0a3) 7:0 0 yes no 1511264933 ' |
| 'USB *Jabra SPEAK 410 USB: USB Audio:0,0: Mic\n' |
| '(8c74f766) 5:0 0 yes no 0 ' |
| 'POST_DSP_LOOPBACK Post DSP Loopback\n' |
| '(8a04af91) 4:0 0 yes no 0 ' |
| 'POST_MIX_LOOPBACK Post Mix Pre DSP Loopback\n' |
| 'Attached clients:' |
| ) |
| |
| mock_host = mock.Mock() |
| mock_host.run_output.side_effect = [input_devices, input_nodes] |
| collector = cras_node_collector.CrasNodeCollector(mock_host) |
| |
| nodes = collector.get_input_nodes() |
| self.assertEquals(5, len(nodes)) |
| node_ids = set([node.node_id for node in nodes]) |
| node_types = set([node.node_type for node in nodes]) |
| device_names = set([node.device_name for node in nodes]) |
| self.assertEquals(node_ids, set( |
| ['4:0', '5:0', '7:0', '9:0', '14:0'])) |
| self.assertEquals(node_types, set( |
| ['MIC', 'USB', 'POST_DSP_LOOPBACK', 'POST_MIX_LOOPBACK'])) |
| self.assertEquals(device_names, set( |
| ['HDA Intel PCH: ALC283 Analog:3,0', |
| 'Jabra SPEAK 410 USB: USB Audio:1,0', |
| 'Jabra SPEAK 410 USB: USB Audio:0,0', |
| 'Post DSP Loopback', |
| 'Post Mix Pre DSP Loopback',])) |
| |
| if __name__ == "__main__": |
| unittest.main() |
