slight improvement of BlockTools

Author: arvidn

chia/simulator/block_tools.py

@@ -796,6 +796,8 @@ def get_consecutive_blocks(
                 for coin in b.get_included_reward_coins():
                     if coin.puzzle_hash == self.farmer_ph:
                         available_coins.append(coin)
+                    if coin.puzzle_hash == self.pool_ph:
+                        available_coins.append(coin)
             print(
                 f"found {len(available_coins)} reward coins in existing chain."
                 "for simplicity, we assume the rewards are all unspent in the original chain"
@@ -1026,6 +1028,8 @@ def get_consecutive_blocks(
                             for coin in full_block.get_included_reward_coins():
                                 if coin.puzzle_hash == self.farmer_ph:
                                     pending_rewards.append(coin)
+                                if coin.puzzle_hash == self.pool_ph:
+                                    pending_rewards.append(coin)
                             if full_block.is_transaction_block():
                                 available_coins.extend(pending_rewards)
                                 pending_rewards = []
@@ -1333,6 +1337,8 @@ def get_consecutive_blocks(
                             for coin in full_block.get_included_reward_coins():
                                 if coin.puzzle_hash == self.farmer_ph:
                                     pending_rewards.append(coin)
+                                if coin.puzzle_hash == self.pool_ph:
+                                    pending_rewards.append(coin)
                             if full_block.is_transaction_block():
                                 available_coins.extend(pending_rewards)
                                 pending_rewards = []