def test_lease_wallet_on_lease_result_empty(): lease_wallet = LeaseWallet() assert None == lease_wallet.get_lease_state(resource='A') lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_OK, _create_lease('A', 'B', [1, 0])) lease_wallet.on_lease_use_result(lease_use_result, resource='A') assert None == lease_wallet.get_lease_state(resource='A')
def test_lease_wallet_on_lease_result_ok(): lease_wallet = LeaseWallet() lease = _create_lease('A', 'epoch', [1]) lease_wallet.add(lease) # Assert that initial state of adding a lease looks fine. lease_state = lease_wallet.get_lease_state(resource='A') assert None != lease_state assert Lease.CompareResult.SAME == lease.compare(lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare(lease_state.lease_current) assert LeaseState.STATUS_SELF_OWNER == lease_state.lease_status # When an "OK" lease_use_result comes, LeaseState should not change. lease_use_result = _create_lease_use_result(LeaseProto.LeaseUseResult.STATUS_OK, lease_state.lease_current, None) lease_wallet.on_lease_use_result(lease_use_result, resource='A') new_lease_state = lease_wallet.get_lease_state(resource='A') assert None != new_lease_state assert Lease.CompareResult.SAME == lease.compare(new_lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare(new_lease_state.lease_current) assert LeaseState.STATUS_SELF_OWNER == new_lease_state.lease_status # When an "OK" lease_use_result arrives for a lease other than current one, LeaseState also should not change lease_wallet.advance(resource='A') lease_wallet.on_lease_use_result(lease_use_result, resource='A') newer_lease_state = lease_wallet.get_lease_state(resource='A') assert None != newer_lease_state assert Lease.CompareResult.SAME == lease.compare(newer_lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare(newer_lease_state.lease_current) assert Lease.CompareResult.OLDER == new_lease_state.lease_current.compare( newer_lease_state.lease_current) assert LeaseState.STATUS_SELF_OWNER == newer_lease_state.lease_status
def test_lease_wallet_on_lease_result_empty(): lease_wallet = LeaseWallet() with pytest.raises(NoSuchLease) as excinfo: lease_wallet.get_lease_state(resource='A') assert 'No lease for resource "A"' == str(excinfo.value) lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_OK, _create_lease('A', 'B', [1, 0])) lease_wallet.on_lease_use_result(lease_use_result, resource='A') with pytest.raises(NoSuchLease) as excinfo: lease_wallet.get_lease_state(resource='A') assert 'No lease for resource "A"' == str(excinfo.value)
def test_lease_wallet_on_lease_result_older(): lease_wallet = LeaseWallet() lease = _create_lease('A', 'epoch', [1]) lease_wallet.add(lease) # Assert that initial state of adding a lease looks fine. lease_state = lease_wallet.get_lease_state(resource='A') assert None != lease_state assert Lease.CompareResult.SAME == lease.compare( lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare( lease_state.lease_current) assert LeaseState.Status.SELF_OWNER == lease_state.lease_status stale_lease = lease_state.lease_current # Advance the Lease so we can compare newer and older leases recent_lease = lease_wallet.advance(resource='A') # When an "OLDER" result comes in for an attempt which is not the current lease, # do not change the current lease state lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_OLDER, stale_lease, None) lease_wallet.on_lease_use_result(lease_use_result, resource='A') new_lease_state = lease_wallet.get_lease_state(resource='A') assert None != new_lease_state assert Lease.CompareResult.SAME == lease.compare( new_lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare( new_lease_state.lease_current) assert LeaseState.Status.SELF_OWNER == new_lease_state.lease_status # When an "OLDER" result comes in for an attempt which is the current lease, # the lease state should change to other owner. lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_OLDER, recent_lease, None) lease_wallet.on_lease_use_result(lease_use_result, resource='A') newer_lease_state = lease_wallet.get_lease_state(resource='A') assert None != newer_lease_state assert None == newer_lease_state.lease_current assert None == newer_lease_state.lease_original assert LeaseState.Status.OTHER_OWNER == newer_lease_state.lease_status with pytest.raises(LeaseNotOwnedByWallet) as excinfo: lease_wallet.get_lease(resource='A') assert 'Lease on "A" has state ({})'.format( LeaseState.Status.OTHER_OWNER) == str(excinfo.value)
def test_lease_wallet_on_lease_result_wrong_epoch(): lease_wallet = LeaseWallet() lease = _create_lease('A', 'epoch', [1]) lease_wallet.add(lease) # Assert that initial state of adding a lease looks fine. lease_state = lease_wallet.get_lease_state(resource='A') assert None != lease_state assert Lease.CompareResult.SAME == lease.compare( lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare( lease_state.lease_current) assert LeaseState.STATUS_SELF_OWNER == lease_state.lease_status stale_lease = lease_state.lease_current # Advance the Lease so we can compare newer and older leases recent_lease = lease_wallet.advance(resource='A') # When a "REVOKED" result comes in for an attempt which is not the current lease, # do not change the current lease state lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_WRONG_EPOCH, stale_lease, None) lease_wallet.on_lease_use_result(lease_use_result, resource='A') new_lease_state = lease_wallet.get_lease_state(resource='A') assert None != new_lease_state assert Lease.CompareResult.SAME == lease.compare( new_lease_state.lease_original) assert Lease.CompareResult.SUPER_LEASE == lease.compare( new_lease_state.lease_current) assert LeaseState.STATUS_SELF_OWNER == new_lease_state.lease_status # When an "REVOKED" result comes in for an attempt which is the current lease, # the lease state should change to revoked status lease_use_result = _create_lease_use_result( LeaseProto.LeaseUseResult.STATUS_WRONG_EPOCH, recent_lease, None) lease_wallet.on_lease_use_result(lease_use_result, resource='A') newer_lease_state = lease_wallet.get_lease_state(resource='A') assert None != newer_lease_state assert None == newer_lease_state.lease_current assert None == newer_lease_state.lease_original assert LeaseState.STATUS_UNOWNED == newer_lease_state.lease_status