diff --git a/gns3server/db/models/__init__.py b/gns3server/db/models/__init__.py index ee4a11122..a5501eef0 100644 --- a/gns3server/db/models/__init__.py +++ b/gns3server/db/models/__init__.py @@ -51,6 +51,7 @@ __all__ = [ "EthernetSwitchTemplate", "IOUTemplate", "Image", + "ImageSyncJob", "LLMModelConfig", "Privilege", "QemuTemplate", diff --git a/gns3server/schemas/__init__.py b/gns3server/schemas/__init__.py index e88623cde..cdc1d67ce 100644 --- a/gns3server/schemas/__init__.py +++ b/gns3server/schemas/__init__.py @@ -230,6 +230,8 @@ __all__ = [ "IOUTemplateUpdate", "IOUUpdate", "Image", + "ImageSyncJob", + "ImageSyncRequest", "ImageType", "LLMModelConfigCreate", "LLMModelConfigData", diff --git a/tests/services/test_image_reconciliation.py b/tests/services/test_image_reconciliation.py index c0351ffe0..8ee12856d 100644 --- a/tests/services/test_image_reconciliation.py +++ b/tests/services/test_image_reconciliation.py @@ -18,7 +18,7 @@ from watchdog.events import FileDeletedEvent, FileModifiedEvent, FileMovedEvent from gns3server.db.models import Base, Image, ImageSyncJob, Template from gns3server.db.models.images import image_template_map from gns3server.db.repositories.images import ImagesRepository -from gns3server.services.image_reconciliation import ImageReconciliationService, InventoryEvents, enumerate_root +from gns3server.services.image_reconciliation import ImageReconciliationService, InventoryEvents from gns3server.utils.image_inventory import ImageLockBusy, fingerprint, image_lock from gns3server.utils.images import InvalidImageError, inspect_image_file, md5sum, write_image