forked from p15670423/monkey
Fix imports for unit test files
This commit is contained in:
parent
f8936f61bb
commit
a2115e4254
|
@ -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"
|
||||||
|
|
||||||
|
|
|
@ -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):
|
||||||
|
|
|
@ -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__)
|
||||||
|
|
||||||
|
|
|
@ -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": {
|
||||||
|
|
Loading…
Reference in New Issue