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)