Fix imports for unit test files

This commit is contained in:
Shreya 2021-04-16 13:20:04 +05:30
parent f8936f61bb
commit a2115e4254
4 changed files with 5 additions and 7 deletions

View File

@ -1,7 +1,7 @@
import json import json
from unittest import TestCase from unittest import TestCase
from .aws_service import filter_instance_data_from_aws_response from common.cloud.aws.aws_service import filter_instance_data_from_aws_response
__author__ = "shay.nehmad" __author__ = "shay.nehmad"

View File

@ -1,6 +1,6 @@
from unittest import TestCase from unittest import TestCase
from .payload_parsing import LimitedSizePayload, Payload from infection_monkey.exploit.tools.payload_parsing import LimitedSizePayload, Payload
class TestPayload(TestCase): class TestPayload(TestCase):

View File

@ -5,9 +5,8 @@ from time import sleep
import pytest import pytest
from monkey_island.cc.models.monkey import Monkey, MonkeyNotFoundError from monkey_island.cc.models.monkey import Monkey, MonkeyNotFoundError
from monkey_island.cc.models.monkey_ttl import MonkeyTtl
from ..test_common.fixtures import FixtureEnum from monkey_island.cc.test_common.fixtures import FixtureEnum
from .monkey_ttl import MonkeyTtl
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)

View File

@ -1,8 +1,7 @@
from unittest.mock import Mock from unittest.mock import Mock
import monkey_island.cc.services.telemetry.processing.post_breach as post_breach import monkey_island.cc.services.telemetry.processing.post_breach as post_breach
from monkey_island.cc.services.telemetry.processing.post_breach import EXECUTION_WITHOUT_OUTPUT
from .post_breach import EXECUTION_WITHOUT_OUTPUT
original_telem_multiple_results = { original_telem_multiple_results = {
"data": { "data": {