def test_stake_part_set_unstake_update(self):
part = StakePart()
part.set_complete(True)
stake = 100
block_height = 10
part.add_stake(100)
unstake = stake
part.set_unstake(block_height, unstake)
self.assertEqual(0, part.stake)
self.assertEqual(stake, part.unstake)
self.assertEqual(block_height, part.unstake_block_height)
self.assertTrue(
part.is_set(BasePartState.DIRTY | BasePartState.COMPLETE))
block_height += block_height
unstake = 10
part.set_unstake(block_height, unstake)
self.assertEqual(stake - unstake, part.stake)
self.assertEqual(unstake, part.unstake)
self.assertEqual(block_height, part.unstake_block_height)
self.assertTrue(
part.is_set(BasePartState.DIRTY | BasePartState.COMPLETE))
refund_unstake = part.normalize(block_height + 1, Revision.IISS.value)
self.assertEqual(unstake, refund_unstake)
def test_stake_part_add_stake(self):
part = StakePart()
part.set_complete(True)
stake = 100
part.add_stake(100)
self.assertEqual(stake, part.stake)
self.assertTrue(
part.is_set(BasePartState.DIRTY | BasePartState.COMPLETE))
def test_delegation_part_equal(self):
part1 = StakePart()
part1.normalize(0, Revision.IISS.value)
part2 = StakePart()
part2.normalize(1, Revision.IISS.value)
self.assertEqual(part1, part2)
offset = 100
part1.add_stake(offset)
part3 = StakePart(stake=offset)
part3.normalize(100, Revision.IISS.value)
self.assertEqual(part1, part3)
def test_delegation_part_equal(self):
part1 = StakePart()
part1.normalize(0)
part2 = StakePart()
part2.normalize(1)
self.assertEqual(part1, part2)
offset = 100
part1.add_stake(offset)
part3 = StakePart(stake=offset)
part3.normalize(100)
self.assertEqual(part1, part3)
