# -*- coding: utf-8 -*-
"""Tests using pytest_resilient_circuits"""
from __future__ import print_function
import pytest
from mock import patch
from resilient_circuits.util import get_config_data, get_function_definition
from resilient_circuits import SubmitTestFunction, FunctionResult
from mock_artifacts import mocked_amp_client, get_mock_config
PACKAGE_NAME = "fn_cisco_amp4ep"
FUNCTION_NAME = "fn_amp_get_file_lists"
# Mock configuration-data section
config_data = get_mock_config()
# Provide a simulation of the Resilient REST API (uncomment to connect to a real appliance)
resilient_mock = "pytest_resilient_circuits.BasicResilientMock"
def assert_keys_in(json_obj, *keys):
for key in keys:
assert key in json_obj
def call_fn_amp_get_file_lists_function(circuits, function_params, timeout=10):
# Fire a message to the function
evt = SubmitTestFunction("fn_amp_get_file_lists", function_params)
circuits.manager.fire(evt)
event = circuits.watcher.wait("fn_amp_get_file_lists_result",
parent=evt,
def setup_method(self):
global config_data
config_data = get_mock_config("*****@*****.**")
