diff --git a/subiquity/server/controllers/tests/test_snaplist.py b/subiquity/server/controllers/tests/test_snaplist.py index 78f2af49..ba12aed7 100644 --- a/subiquity/server/controllers/tests/test_snaplist.py +++ b/subiquity/server/controllers/tests/test_snaplist.py @@ -14,7 +14,7 @@ # along with this program. If not, see . import unittest -from unittest.mock import AsyncMock, Mock +from unittest.mock import AsyncMock import requests diff --git a/subiquitycore/tests/mocks.py b/subiquitycore/tests/mocks.py index 74f2f66b..663b02b8 100644 --- a/subiquitycore/tests/mocks.py +++ b/subiquitycore/tests/mocks.py @@ -43,4 +43,9 @@ def make_app(model=None): app.opts = mock.Mock() app.opts.dry_run = True app.scale_factor = 1000 + app.echo_syslog_id = None + app.log_syslog_id = None + app.report_start_event = mock.Mock() + app.report_finish_event = mock.Mock() + return app