def test_config_arg():
class Fake():
pass
snap = snapshot.EBSSnapshot('no-region-1')
obj = snap.config(config=Fake())
assert (isinstance(obj, Fake))
def test_connection_arg():
class Fake:
def __init__(self):
pass
snap = snapshot.EBSSnapshot(region='no-region-1')
obj = snap.connection(conn=Fake())
assert (isinstance(obj, Fake))
def test_connection():
snap = snapshot.EBSSnapshot(region='no-region-1')
playback = Playback(region_name='no-region-1',
data_path=os.path.join(PLACEBO_PATH, 'session'))
playback.start()
snap.session(playback.session)
obj = snap.connection()
assert (str(obj).startswith('<botocore.client.EC2'))
def test_volumes_list_filters():
fakepaginator = FakePagenator(group='Volumes')
fakeconnection = FakeConnection(paginatorobj=fakepaginator)
ec2 = snapshot.EBSSnapshot('no-region-1')
ec2.connection(fakeconnection)
try:
for vol in ec2.volumes(filters={'Tag': 'blah'}):
print(vol)
except Exception:
pytest.fail()
def test_snapshots_list():
fakepaginator = FakePagenator(group='Snapshots')
fakeconnection = FakeConnection(paginatorobj=fakepaginator)
ec2 = snapshot.EBSSnapshot(region='no-region-1')
ec2.connection(fakeconnection)
try:
for snap in ec2.snapshots():
print(snap)
except Exception:
pytest.fail()
def test_session(caplog):
playback = Playback(region_name='no-region-1',
data_path=PLACEBO_PATH + '/session')
sess = playback.session
playback.start()
snap = snapshot.EBSSnapshot(role='arn:aws:iam::123456789:role/EBSSnapshot',
region='no-region-1')
obj = snap.session(sess)
for record in caplog.records:
assert record.levelname != 'ERROR'
def test_connection_exception():
playback = Playback(region_name='no-region-1',
data_path=os.path.join(PLACEBO_PATH, 'session'))
playback.start()
snap = snapshot.EBSSnapshot(region='no-region-1')
snap.session(playback.session)
try:
snap.connection()
except Exception:
pytest.fail("Unhandled exception")
def test_create_snapshot(caplog):
playback = Playback(region_name='no-region-1',
data_path=PLACEBO_PATH + '/create_snapshot')
sess = playback.session
playback.start()
try:
obj = snapshot.EBSSnapshot(region='no-region-1')
obj.session(sess)
obj.create_snapshot(fixture_vol(Tags=fixture_tags()))
except Exception as ex:
pytest.fail('Unhandle exception')
for record in caplog.records:
assert record.levelname != 'ERROR'
def test_filter_inlife_snapshot_lt_fail():
date = datetime.datetime.now(tz=tzutc())
date = date + timedelta(days=-2)
snap = fixture_snap(StartTime=date)
ec2 = snapshot.EBSSnapshot('no-region-1')
assert not ec2.filter_inlife_snapshot(snap, lt=-3)
def test_filter_inlife_snapshot_gt():
date = datetime.datetime.now(tz=tzutc())
date = date + timedelta(days=-2)
snap = fixture_snap(StartTime=date, Tags=fixture_tags())
ec2 = snapshot.EBSSnapshot(region='no-region-1')
assert ec2.filter_inlife_snapshot(snap, gt=-3)
def test_config():
snap = snapshot.EBSSnapshot('no-region-1')
obj = snap.config()
assert (str(obj).startswith('<botocore.config.Config'))
