def test_precmds_to_two_duts(self, mock_ec, mock_rp):
# pylint: disable=no-self-use
bench = Bench()
bench._resource_provider = mock.Mock()
bench.args = mock.MagicMock()
bench.args.my_duts = mock.MagicMock(return_value=False)
bench.execute_command = mock.MagicMock()
mock_resconf = mock.Mock()
mock_resconf.get_dut_configuration = mock.MagicMock(return_value=[{
"pre_cmds": ["first", "second"]
}, {
"pre_cmds": ["first2", "second2"]
}])
bench.resource_configuration = mock_resconf
mock_resconf.count_duts = mock.MagicMock(return_value=2)
# Call using mangled name of __send_pre_commands method
bench.send_pre_commands()
mock_ec.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2")
])
bench.execute_command.reset_mock()
# Test again with argument cmds
bench.send_pre_commands("somecommand")
mock_ec.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2"),
mock.call("*", "somecommand")
])
def test_postcmds_to_two_duts(self, mock_ec, mock_rp):
tc = Bench()
tc._resource_provider = mock.Mock()
tc.execute_command = mock.MagicMock()
mock_resconf = mock.Mock()
mock_resconf.get_dut_configuration = mock.MagicMock(return_value=[{
"post_cmds": ["first", "second"]
}, {
"post_cmds": ["first2", "second2"]
}])
tc.resource_configuration = mock_resconf
# Call using mangled name of __send_pre_commands method
tc._Bench__send_post_commands()
tc.execute_command.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2")
])
tc.execute_command.reset_mock()
# Test again with argument cmds
tc._Bench__send_post_commands("somecommand")
tc.execute_command.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2"),
mock.call("*", "somecommand")
])
def test_postcmds_to_two_duts(self, mock_ec, mock_rp):
# pylint: disable=no-self-use
bench = Bench()
bench._resource_provider = mock.MagicMock()
bench.execute_command = mock.MagicMock()
bench.args = mock.MagicMock()
bench.logger = mock.MagicMock()
type(bench.args).my_duts = mock.PropertyMock(return_value=False)
type(bench.args).pause_when_external_dut = mock.MagicMock(
return_value=False)
bench._resources.init(mock.MagicMock())
bench._commands.init()
mock_dut1 = mock.MagicMock()
type(mock_dut1).index = mock.PropertyMock(return_value=1)
mock_dut2 = mock.MagicMock()
type(mock_dut2).index = mock.PropertyMock(return_value=2)
bench._resources.duts = [mock_dut1, mock_dut2]
mock_resconf = mock.Mock()
mock_resconf.get_dut_configuration = mock.MagicMock(return_value=[{
"post_cmds": ["first", "second"]
}, {
"post_cmds": ["first2", "second2"]
}])
bench.resource_configuration = mock_resconf
mock_resconf.count_duts = mock.MagicMock(return_value=2)
bench.send_post_commands()
mock_ec.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2")
])
mock_ec.reset_mock()
# Test again with argument cmds
bench.send_post_commands("somecommand")
mock_ec.assert_has_calls([
mock.call(1, "first"),
mock.call(1, "second"),
mock.call(2, "first2"),
mock.call(2, "second2"),
mock.call("*", "somecommand")
])
