def _get_function(self):
events = [mu.PeriodicEvent(
local_session(self.policy.session_factory),
self.policy.data['mode'],
self.region
)]
return mu.PolicyFunction(self.policy, events=events)
def _get_function(self):
events = [
mu.PubSubSource(local_session(self.policy.session_factory),
{"topic": self._resource_topic()}),
mu.SecurityCenterSubscriber(
local_session(self.policy.session_factory), {
"topic": self._resource_topic(),
"org": self.policy.data["mode"]["org"]
}, self.policy.resource_manager)
]
return mu.PolicyFunction(self.policy, events=events)
def test_policy_context_deps(self):
p = self.load_policy({
'name': 'check',
'resource': 'gcp.instance',
'mode': {
'type': 'gcp-periodic',
'schedule': 'every 2 hours'}},
output_dir='gs://somebucket/some-prefix',
log_group='gcp',
config={'metrics': 'gcp'})
pf = mu.PolicyFunction(p, archive=True)
self.assertEqual(
pf.get_output_deps(),
['google-cloud-monitoring',
'google-cloud-storage',
'google-cloud-logging'])
def _get_function(self):
events = [
mu.ApiSubscriber(local_session(self.policy.session_factory),
self.policy.data['mode'])
]
return mu.PolicyFunction(self.policy, events=events)
