def main(): try: args = docopt(__doc__) pipeline_definition_uri = args["--pipeline-definition-uri"] table_definition_uri = args["--table-definition-uri"] table_name = args["--tablename"] backup_source = args["--backup-source"] region = args["--region"] subnet_id = args["--subnetId"] if args["--subnetId"] else get_first_subnet_id_from_vpc_stack(region) log_dest = args["--log-dest"] if args["--log-dest"] else get_log_dest_from_backup_source(backup_source) mail_address = args["--notify-mail"] data_only = args["--data-only"] desired_write_throughput = args["--write-throughput"] print "Using subnet id: {0}".format(subnet_id) handler = DynamoDbRestoreHandler(data_only, table_name, table_definition_uri, pipeline_definition_uri, backup_source, subnet_id, log_dest, mail_address, desired_write_throughput, region) handler.restore() except ClientError as e: print e print "Your credentials may be expired!" sys.exit(1)
def test_pipeline_exists_return_true_for_existing_pipeline(self): mocked_response = { "pipelineIdList": [ {"id": "df-09204411HHMSRZ11YXDP", "name": "my-pipeline-1"}, {"id": "df-08522353PMJ0KLMYFRUL", "name": "my-pipeline-2"}, {"id": "df-08320941WW3ZVXZXG4D2", "name": "my-pipeline-3 Backup"}, ] } client_mock = Mock() client_mock.list_pipelines.return_value = mocked_response self.assertTrue(DynamoDbRestoreHandler.pipeline_exists(client_mock, "my-pipeline-1"))