diff --git a/src/vmc/nessus/tests.py b/src/vmc/nessus/tests.py index 475da71..19829b9 100644 --- a/src/vmc/nessus/tests.py +++ b/src/vmc/nessus/tests.py @@ -22,7 +22,7 @@ from django.test import TestCase -from vmc.assets.models import Asset, Port +from vmc.assets.models import Asset from vmc.nessus.api import Nessus from vmc.nessus.models import Config from vmc.nessus.tasks import get_trash_folder_id, update, update_data diff --git a/src/vmc/ralph/tests.py b/src/vmc/ralph/tests.py index 0bb50b8..743b8a1 100644 --- a/src/vmc/ralph/tests.py +++ b/src/vmc/ralph/tests.py @@ -18,14 +18,14 @@ * """ -from unittest.mock import patch, Mock, MagicMock +import json +from unittest.mock import patch, Mock from django.test import TestCase -from vmc.assets.models import Asset, Port -from vmc.ralph.api import Ralph, AssetIdException +from vmc.assets.models import Asset +from vmc.ralph.api import Ralph from vmc.ralph.models import Config -import json from vmc.common.tests import get_fixture_location from vmc.ralph.tasks import load_all_assets