def test_board_is_checked(self): """Test that we're happy if the board responds to an sver.""" controller = mock.Mock(spec_set=MachineController) controller.get_software_version.return_value = CoreInfo( (0, 0), 0, 0, 1.33, 128, 0, "Test") machine_control.test_and_boot(controller, "spam", 2, 2)
def test_board_was_not_booted(self, width, height): """Test that the board is booted if it wasn't.""" # Create a controller which times out on the first get_software_version # but which works on the second. controller = mock.Mock(spec_set=MachineController) controller.get_software_version.side_effect = [ TimeoutError, CoreInfo((0, 0), 0, 0, 1.33, 128, 0, "Test") ] machine_control.test_and_boot(controller, "spam", width, height) # Check that the boot code was called correctly controller.boot.assert_called_once_with(width, height)
def test_board_not_booted_fails_to_boot(self, width, height): """Test that we raise an exception if booting the board is unsuccessful. """ # Create a controller which times out on get_software_version. controller = mock.Mock(spec_set=MachineController) controller.get_software_version.side_effect = TimeoutError with pytest.raises(Exception) as excinfo: machine_control.test_and_boot(controller, "spam", width, height) assert "failed to boot" in str(excinfo.value) # Check that the boot code was called correctly controller.boot.assert_called_once_with(width, height)