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"))
