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
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"

View File

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

View File

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

View File

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