def test_delete_all_requests_dberror(self): """ Tests db error deleting all requests from the request_status table """ exp_err = 'Database Error. Internal database error, please contact LP DAAC User Services' try: mock_secretsmanager_get_parameter(1) database.single_query = Mock(side_effect=[DbError(exp_err)]) requests_db.delete_all_requests() self.fail("expected DatabaseError") except requests_db.DatabaseError as err: self.assertEqual(exp_err, str(err)) database.single_query.assert_called_once()
def task(event, context): #pylint: disable-msg=unused-argument """ Task called by the handler to perform the work. This task will parse the input, removing the granuleId and file keys for a granule. Args: event (dict): passed through from the handler context (Object): passed through from the handler Returns: dict: dict containing granuleId and keys. See handler for detail. Raises: ExtractFilePathsError: An error occurred parsing the input. """ result = {} _LOG.debug(f"event: '{event}' ") try: function = event['function'] except KeyError: raise BadRequestError("Missing 'function' in input data") if function == "query": result = query_requests(event) if function == "add": result = add_request(event) if function == "clear": result = requests_db.delete_all_requests() return result
def tearDown(self): boto3.client = Mock() # todo: Why? mock_secretsmanager_get_parameter(1) try: requests_db.delete_all_requests() except requests_db.NotFound: pass except requests_db.DatabaseError: pass requests_db.request_id_generator = self.mock_generator CumulusLogger.error = self.mock_error CumulusLogger.info = self.mock_info boto3.client = self.mock_boto3_client os.environ.pop('PREFIX', None) del os.environ['RESTORE_EXPIRE_DAYS'] del os.environ['RESTORE_REQUEST_RETRIES'] db_config.del_env() del os.environ["RESTORE_RETRIEVAL_TYPE"]
def tearDown(self): boto3.client = Mock() mock_ssm_get_parameter(1) requests_db.request_id_generator = self.mock_request_group_id requests_db.get_utc_now_iso = self.mock_utcnow try: requests_db.delete_all_requests() except requests_db.NotFound: pass except requests_db.DatabaseError: pass boto3.client = self.mock_boto3 del os.environ["PREFIX"] del os.environ["DATABASE_HOST"] del os.environ["DATABASE_NAME"] del os.environ["DATABASE_USER"] del os.environ["DATABASE_PW"] del os.environ["PLATFORM"]
def tearDown(self): boto3.client = Mock() mock_ssm_get_parameter(1) try: requests_db.delete_all_requests() except requests_db.NotFound: pass except requests_db.DatabaseError: pass requests_db.get_utc_now_iso = self.mock_utcnow boto3.client = self.mock_boto3_client try: del os.environ['COPY_RETRY_SLEEP_SECS'] del os.environ['COPY_RETRIES'] del os.environ["DATABASE_HOST"] del os.environ["DATABASE_NAME"] del os.environ["DATABASE_USER"] del os.environ["DATABASE_PW"] except KeyError: pass
def tearDown(self): boto3.client = Mock() mock_ssm_get_parameter(1) try: requests_db.delete_all_requests() except requests_db.NotFound: pass except requests_db.DatabaseError: pass requests_db.request_id_generator = self.mock_generator CumulusLogger.error = self.mock_error CumulusLogger.info = self.mock_info boto3.client = self.mock_boto3_client del os.environ['RESTORE_EXPIRE_DAYS'] del os.environ['RESTORE_REQUEST_RETRIES'] del os.environ["DATABASE_HOST"] del os.environ["DATABASE_NAME"] del os.environ["DATABASE_USER"] del os.environ["DATABASE_PW"] del os.environ["DATABASE_PORT"] del os.environ["RESTORE_RETRIEVAL_TYPE"]
def test_delete_all_requests(self): """ Tests deleting all requests from the request_status table """ try: self.create_test_requests() boto3.client = Mock() mock_ssm_get_parameter(1) result = requests_db.delete_all_requests() self.assertEqual([], result) except requests_db.DatabaseError as err: self.fail(f"delete_all_requests. {str(err)}")
def tearDown(self): boto3.client = Mock() mock_secretsmanager_get_parameter(1) try: requests_db.delete_all_requests() except requests_db.NotFound: pass except requests_db.DatabaseError: pass requests_db.get_utc_now_iso = self.mock_utcnow boto3.client = self.mock_boto3_client try: os.environ.pop('PREFIX') del os.environ['COPY_RETRY_SLEEP_SECS'] del os.environ['COPY_RETRIES'] del os.environ['DATABASE_HOST'] del os.environ['DATABASE_NAME'] del os.environ['DATABASE_USER'] del os.environ['DATABASE_PW'] except KeyError: pass
def test_delete_all_requests(self): """ Tests deleting all requests from the request_status table """ exp_request_ids = [REQUEST_ID1, REQUEST_ID2, REQUEST_ID3, REQUEST_ID4, REQUEST_ID5, REQUEST_ID6, REQUEST_ID7, REQUEST_ID8, REQUEST_ID9, REQUEST_ID10, REQUEST_ID11] try: create_select_requests(exp_request_ids) empty_result = [] mock_ssm_get_parameter(1) database.single_query = Mock( side_effect=[empty_result]) result = requests_db.delete_all_requests() database.single_query.assert_called() self.assertEqual(empty_result, result) except requests_db.DatabaseError as err: self.fail(f"delete_all_requests. {str(err)}")