2faffc96de
* Flask Request object does not have a 'body' attribute, changed to 'data' * Making moto 'glaciar' more aws 'glaciar' like. * Making moto 'glacier' more aws 'glacier' like. * Fixing Travis errors? * OK, check if object has proper attribute because HTTPrettyRequest has no data attribute and Python Requests has no body attribute. * Output to match test expectation; sleep for 60 seconds to mimic actual wait time. * Amending test_describe_job to reflect changes. * Shorten time from 1 minute to seconds. * Shorten sleep time in test. Forgot about the test.
91 lines
2.7 KiB
Python
91 lines
2.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import json
|
|
import time
|
|
|
|
from boto.glacier.layer1 import Layer1
|
|
import sure # noqa
|
|
|
|
from moto import mock_glacier_deprecated
|
|
|
|
|
|
@mock_glacier_deprecated
|
|
def test_init_glacier_job():
|
|
conn = Layer1(region_name="us-west-2")
|
|
vault_name = "my_vault"
|
|
conn.create_vault(vault_name)
|
|
archive_id = conn.upload_archive(
|
|
vault_name, "some stuff", "", "", "some description")
|
|
|
|
job_response = conn.initiate_job(vault_name, {
|
|
"ArchiveId": archive_id,
|
|
"Type": "archive-retrieval",
|
|
})
|
|
job_id = job_response['JobId']
|
|
job_response['Location'].should.equal(
|
|
"//vaults/my_vault/jobs/{0}".format(job_id))
|
|
|
|
|
|
@mock_glacier_deprecated
|
|
def test_describe_job():
|
|
conn = Layer1(region_name="us-west-2")
|
|
vault_name = "my_vault"
|
|
conn.create_vault(vault_name)
|
|
archive_id = conn.upload_archive(
|
|
vault_name, "some stuff", "", "", "some description")
|
|
job_response = conn.initiate_job(vault_name, {
|
|
"ArchiveId": archive_id,
|
|
"Type": "archive-retrieval",
|
|
})
|
|
job_id = job_response['JobId']
|
|
|
|
job = conn.describe_job(vault_name, job_id)
|
|
joboutput = json.loads(job.read().decode("utf-8"))
|
|
|
|
joboutput.should.have.key('Tier').which.should.equal('Standard')
|
|
joboutput.should.have.key('StatusCode').which.should.equal('InProgress')
|
|
joboutput.should.have.key('VaultARN').which.should.equal('arn:aws:glacier:RegionInfo:us-west-2:012345678901:vaults/my_vault')
|
|
|
|
|
|
@mock_glacier_deprecated
|
|
def test_list_glacier_jobs():
|
|
conn = Layer1(region_name="us-west-2")
|
|
vault_name = "my_vault"
|
|
conn.create_vault(vault_name)
|
|
archive_id1 = conn.upload_archive(
|
|
vault_name, "some stuff", "", "", "some description")['ArchiveId']
|
|
archive_id2 = conn.upload_archive(
|
|
vault_name, "some other stuff", "", "", "some description")['ArchiveId']
|
|
|
|
conn.initiate_job(vault_name, {
|
|
"ArchiveId": archive_id1,
|
|
"Type": "archive-retrieval",
|
|
})
|
|
conn.initiate_job(vault_name, {
|
|
"ArchiveId": archive_id2,
|
|
"Type": "archive-retrieval",
|
|
})
|
|
|
|
jobs = conn.list_jobs(vault_name)
|
|
len(jobs['JobList']).should.equal(2)
|
|
|
|
|
|
@mock_glacier_deprecated
|
|
def test_get_job_output():
|
|
conn = Layer1(region_name="us-west-2")
|
|
vault_name = "my_vault"
|
|
conn.create_vault(vault_name)
|
|
archive_response = conn.upload_archive(
|
|
vault_name, "some stuff", "", "", "some description")
|
|
archive_id = archive_response['ArchiveId']
|
|
job_response = conn.initiate_job(vault_name, {
|
|
"ArchiveId": archive_id,
|
|
"Type": "archive-retrieval",
|
|
})
|
|
job_id = job_response['JobId']
|
|
|
|
time.sleep(6)
|
|
|
|
output = conn.get_job_output(vault_name, job_id)
|
|
output.read().decode("utf-8").should.equal("some stuff")
|