diff --git a/subiquity/server/autoinstall.py b/subiquity/server/autoinstall.py new file mode 100644 index 000000000..0abc3dd37 --- /dev/null +++ b/subiquity/server/autoinstall.py @@ -0,0 +1,31 @@ +# Copyright 2024 Canonical, Ltd. +# +# This program is free software: you can redistribute it and/or modify +# it under the terms of the GNU Affero General Public License as +# published by the Free Software Foundation, either version 3 of the +# License, or (at your option) any later version. +# +# This program is distributed in the hope that it will be useful, +# but WITHOUT ANY WARRANTY; without even the implied warranty of +# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the +# GNU Affero General Public License for more details. +# +# You should have received a copy of the GNU Affero General Public License +# along with this program. If not, see . +import logging + +log = logging.getLogger("subiquity.server.autoinstall") + + +class AutoinstallError(Exception): + pass + + +class AutoinstallValidationError(AutoinstallError): + def __init__( + self, + owner: str, + ): + self.message = f"Malformed autoinstall in {owner!r} section" + self.owner = owner + super().__init__(self.message) diff --git a/subiquity/server/server.py b/subiquity/server/server.py index 1aa949be1..75831878d 100644 --- a/subiquity/server/server.py +++ b/subiquity/server/server.py @@ -30,7 +30,7 @@ from subiquity.cloudinit import get_host_combined_cloud_config from subiquity.common.api.server import bind, controller_for_request from subiquity.common.apidef import API -from subiquity.common.errorreport import ErrorReporter, ErrorReportKind +from subiquity.common.errorreport import ErrorReport, ErrorReporter, ErrorReportKind from subiquity.common.serialize import to_json from subiquity.common.types import ( ApplicationState, @@ -41,7 +41,7 @@ PasswordKind, ) from subiquity.models.subiquity import ModelNames, SubiquityModel -from subiquity.server.autoinstall import AutoinstallValidationError +from subiquity.server.autoinstall import AutoinstallError, AutoinstallValidationError from subiquity.server.controller import SubiquityController from subiquity.server.dryrun import DRConfig from subiquity.server.errors import ErrorController @@ -403,8 +403,9 @@ def note_data_for_apport(self, key, value): def make_apport_report(self, kind, thing, *, wait=False, **kw): return self.error_reporter.make_apport_report(kind, thing, wait=wait, **kw) - async def _run_error_cmds(self, report): - await report._info_task + async def _run_error_cmds(self, report: Optional[ErrorReport] = None) -> None: + if report is not None and report._info_task is not None: + await report._info_task Error = getattr(self.controllers, "Error", None) if Error is not None and Error.cmds: try: @@ -421,7 +422,7 @@ def _exception_handler(self, loop, context): return report = self.error_reporter.report_for_exc(exc) log.error("top level error", exc_info=exc) - if not report: + if not isinstance(exc, AutoinstallError) and not report: report = self.make_apport_report( ErrorReportKind.UNKNOWN, "unknown error", exc=exc ) diff --git a/subiquity/server/tests/test_controller.py b/subiquity/server/tests/test_controller.py index bf73c4536..89592aa75 100644 --- a/subiquity/server/tests/test_controller.py +++ b/subiquity/server/tests/test_controller.py @@ -61,7 +61,7 @@ def test_setup_autoinstall(self, mock_load): mock_load.assert_called_once_with("default-data") def test_autoinstall_validation(self): - """Test validation error type""" + """Test validation error type and no apport reporting""" self.controller.autoinstall_schema = { "type": "object", @@ -84,3 +84,8 @@ def test_autoinstall_validation(self): # Assert error section is based on autoinstall_key self.assertEquals(exception.owner, "some-key") + + # Assert apport report is not created + # This only checks that controllers do not manually create an apport + # report on validation. Should also be tested in Server + self.controller.app.make_apport_report.assert_not_called() diff --git a/subiquity/server/tests/test_server.py b/subiquity/server/tests/test_server.py index ae380bf2a..ce1eb1b09 100644 --- a/subiquity/server/tests/test_server.py +++ b/subiquity/server/tests/test_server.py @@ -160,6 +160,7 @@ async def asyncSetUp(self): }, }, } + self.server.make_apport_report = Mock() def test_valid_schema(self): """Test that the expected autoinstall JSON schema is valid""" @@ -179,6 +180,20 @@ def test_autoinstall_validation__error_type(self): with self.assertRaises(AutoinstallValidationError): self.server.validate_autoinstall() + async def test_autoinstall_validation__no_error_report(self): + """Test no apport reporting""" + + exception = AutoinstallValidationError("Mock") + + loop = Mock() + context = {"exception": exception} + + with patch("subiquity.server.server.log"): + with patch.object(self.server, "_run_error_cmds"): + self.server._exception_handler(loop, context) + + self.server.make_apport_report.assert_not_called() + class TestMetaController(SubiTestCase): async def test_interactive_sections_not_present(self): diff --git a/subiquitycore/tests/mocks.py b/subiquitycore/tests/mocks.py index 89456ac6d..748b97b55 100644 --- a/subiquitycore/tests/mocks.py +++ b/subiquitycore/tests/mocks.py @@ -50,5 +50,6 @@ def make_app(model=None): app.log_syslog_id = None app.report_start_event = mock.Mock() app.report_finish_event = mock.Mock() + app.make_apport_report = mock.Mock() return app