def setup(self): """Setup before each method""" self._dispatcher = PagerDutyOutput(REGION, FUNCTION_NAME, CONFIG) remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def setup(self): """Setup before each method""" self._dispatcher = SlackOutput(REGION, ACCOUNT_ID, FUNCTION_NAME, None) remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def setup(self): """Setup before each method""" self._dispatcher = PagerDutyIncidentOutput(REGION, ACCOUNT_ID, FUNCTION_NAME, CONFIG) self._dispatcher._base_url = self.CREDS['api'] remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def setup(self): """Setup before each method""" self._mock_s3 = mock_s3() self._mock_s3.start() self._mock_kms = mock_kms() self._mock_kms.start() self._dispatcher = PagerDutyOutputV2(REGION, ACCOUNT_ID, FUNCTION_NAME, None) remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def setup(self): """Setup before each method""" self._mock_s3 = mock_s3() self._mock_s3.start() self._mock_kms = mock_kms() self._mock_kms.start() self._dispatcher = SlackOutput(None) remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def setup(self): """Setup before each method""" self._mock_s3 = mock_s3() self._mock_s3.start() self._mock_kms = mock_kms() self._mock_kms.start() self._dispatcher = PagerDutyIncidentOutput(None) self._dispatcher._base_url = self.CREDS['api'] remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self.DESCRIPTOR) put_mock_creds(output_name, self.CREDS, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS)
def _setup_dispatch(self): """Helper for setting up SlackOutput dispatch""" remove_temp_secrets() output_name = self.__dispatcher.output_cred_name(self.__descriptor) creds = {'url': 'https://api.slack.com/web-hook-key'} put_mock_creds(output_name, creds, self.__dispatcher.secrets_bucket, REGION, KMS_ALIAS) return get_alert()
def _setup_dispatch(self, url): """Helper for setting up PhantomOutput dispatch""" remove_temp_secrets() output_name = self.__dispatcher.output_cred_name(self.__descriptor) creds = {'url': url, 'ph_auth_token': 'mocked_auth_token'} put_mock_creds(output_name, creds, self.__dispatcher.secrets_bucket, REGION, KMS_ALIAS) return get_alert()
def test_load_creds(self): """OutputDispatcher - Load Credentials""" remove_temp_secrets() output_name = self._dispatcher.output_cred_name(self._descriptor) creds = {'url': 'http://www.foo.bar/test', 'token': 'token_to_encrypt'} put_mock_creds(output_name, creds, self._dispatcher.secrets_bucket, REGION, KMS_ALIAS) loaded_creds = self._dispatcher._load_creds(self._descriptor) assert_is_not_none(loaded_creds) assert_equal(len(loaded_creds), 2) assert_equal(loaded_creds['url'], u'http://www.foo.bar/test') assert_equal(loaded_creds['token'], u'token_to_encrypt')
def _setup_dispatch(self): """Helper for setting up PagerDutyOutput dispatch""" remove_temp_secrets() # Cache the _get_default_properties and set it to return None self.__backup_method = self.__dispatcher._get_default_properties self.__dispatcher._get_default_properties = lambda: None output_name = self.__dispatcher.output_cred_name(self.__descriptor) creds = {'url': 'http://pagerduty.foo.bar/create_event.json', 'service_key': 'mocked_service_key'} put_mock_creds(output_name, creds, self.__dispatcher.secrets_bucket, REGION, KMS_ALIAS) return get_alert()
def test_load_credentials(self): """S3Driver - With File Driver - Load Credentials - Pulls into LocalFileStore Here we use the S3Driver's caching ability to yank stuff into a local driver.""" remove_temp_secrets() creds = {'my_secret': 'i ate two portions of biscuits and gravy'} input_credentials = Credentials(creds, is_encrypted=False, region=REGION) descriptor = 'test_descriptor' # Annoyingly, moto needs us to create the bucket first # We put a random unrelated object into the bucket and this will set up the bucket for us put_mock_s3_object(self._s3_driver.get_s3_secrets_bucket(), 'aaa', 'bbb', REGION) # First, check if the Local driver can find the credentials (we don't expect it to) assert_false(self._fs_driver.has_credentials(descriptor)) # Save the credentials using S3 driver result = self._s3_driver.save_credentials_into_s3( descriptor, input_credentials, KMS_ALIAS) assert_true(result) # We still don't expect the Local driver to find the credentials assert_false(self._fs_driver.has_credentials(descriptor)) # Use S3Driver to warm up the Local driver self._s3_driver.load_credentials(descriptor) # Now we should be able to get the credentials from the local fs assert_true(self._fs_driver.has_credentials(descriptor)) credentials = self._fs_driver.load_credentials(descriptor) assert_is_not_none(credentials) assert_true(credentials.is_encrypted()) loaded_creds = json.loads(credentials.get_data_kms_decrypted()) assert_equal(loaded_creds, creds) remove_temp_secrets()
def test_load_creds(self): """OutputDispatcher - Load Credentials""" remove_temp_secrets() key = get_formatted_output_credentials_name( 'test_service', self._descriptor ) creds = {'url': 'http://www.foo.bar/test', 'token': 'token_to_encrypt'} put_mock_creds(key, creds, self._dispatcher._credentials_provider._core_driver._bucket, REGION, KMS_ALIAS) loaded_creds = self._dispatcher._load_creds(self._descriptor) assert_is_not_none(loaded_creds) assert_equal(len(loaded_creds), 2) assert_equal(loaded_creds['url'], u'http://www.foo.bar/test') assert_equal(loaded_creds['token'], u'token_to_encrypt')