Panther features a simple framework to allow basic mocking for unit tests.
When writing a detection that requires an external API call, mocks can be utilized to mimic the server response in the unit tests without having to actually execute an API call.
Mocks are defined by a given
Mock Name and
Return Value which respectively denote the name of the object to patch and the
string to be returned when the patched object is invoked.
Added mocks are defined on the unit test level allowing users to define different mocks for each unit test. The section to add mocks to a unit test are located directly underneath the unit test resource definition
Mocks are allowed on the global and built-in python namespaces. This includes:
Imported Modules and Functions
Built-in Python Functions
Python provides two distinct forms of importing which can be mocked as such:
from package import module
The detection utilizes a global helper function
resource_lookup function from
panther_oss_helpers which queries the
resources-api and returns the resource attributes.
However, the unit test should be able to be performed without any external API calls.
This test fails as there is no corresponding resource mapping to the generic example data.
# --- Snipped ---from panther_oss_helpers import resource_lookup# --- Snipped ---def policy(resource):# --- Snipped ---for recorder_name in resource.get("Recorders", ):recorder = resource_lookup(recorder_name)resource_records_global_resources = bool(deep_get(recorder, "RecordingGroup", "IncludeGlobalResourceTypes")and deep_get(recorder, "Status", "Recording"))if resource_records_global_resources:return Truereturn False# --- Snipped ---
The detection uses the
from panther_oss_helpers import resource_lookup convention which means the mock should be defined for the
Mocks provide a way to leverage real world data to test the detection logic.
While this resource should be compliant, the unit test fails.
Detections that do not expect a
string to be returned require a small tweak for mocks.
In order to get this unit test working as expected, the following modifications need to be made:
# --- Snipped ---import json# Another option is to use: from ast import literal_eval# --- Snipped ---def policy(resource):# --- Snipped ---recorder = resource_lookup(recorder_name)if isinstance(recorder, str):recorder = json.loads(recorder)# --- Snipped ---
Once this modification is added, you can now test the detection logic with real data!