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