#!/usr/bin/env python3
# Copyright (c) 2014-2019 The Bitcoin Core developers
# Distributed under the MIT software license, see the accompanying
# file COPYING or http://www.opensource.org/licenses/mit-license.php.
"""Test mempool limiting together/eviction with the wallet."""

from decimal import Decimal

from test_framework.blocktools import COINBASE_MATURITY
from test_framework.test_framework import BitcoinTestFramework
from test_framework.util import assert_equal, assert_greater_than, assert_raises_rpc_error, gen_return_txouts
from test_framework.wallet import MiniWallet


class MempoolLimitTest(BitcoinTestFramework):
    def set_test_params(self):
        self.setup_clean_chain = True
        self.num_nodes = 1
        self.extra_args = [[
            "-acceptnonstdtxn=1",
            "-maxmempool=5",
            "-spendzeroconfchange=0",
        ]]
        self.supports_cli = False

    def send_large_txs(self, miniwallet, txouts, fee_rate, tx_batch_size):
        for _ in range(tx_batch_size):
            tx = miniwallet.create_self_transfer(from_node=self.nodes[0], fee_rate=fee_rate)['tx']
            for txout in txouts:
                tx.vout.append(txout)
            miniwallet.sendrawtransaction(from_node=self.nodes[0], tx_hex=tx.serialize().hex())

    def run_test(self):
        txouts = gen_return_txouts()
        miniwallet = MiniWallet(self.nodes[0])
        relayfee = self.nodes[0].getnetworkinfo()['relayfee']

        self.log.info('Check that mempoolminfee is minrelytxfee')
        assert_equal(self.nodes[0].getmempoolinfo()['minrelaytxfee'], Decimal('0.00001000'))
        assert_equal(self.nodes[0].getmempoolinfo()['mempoolminfee'], Decimal('0.00001000'))

        tx_batch_size = 25
        num_of_batches = 3
        # Generate UTXOs to flood the mempool
        # 1 to create a tx initially that will be evicted from the mempool later
        # 3 batches of multiple transactions with a fee rate much higher than the previous UTXO
        # And 1 more to verify that this tx does not get added to the mempool with a fee rate less than the mempoolminfee
        self.generate(miniwallet, 1 + (num_of_batches * tx_batch_size) + 1)

        # Mine 99 blocks so that the UTXOs are allowed to be spent
        self.generate(self.nodes[0], COINBASE_MATURITY - 1)

        self.log.info('Create a mempool tx that will be evicted')
        tx_to_be_evicted_id = miniwallet.send_self_transfer(from_node=self.nodes[0], fee_rate=relayfee)["txid"]

        # Increase the tx fee rate massively to give the subsequent transactions a higher priority in the mempool
        base_fee = relayfee * 1000

        self.log.info("Fill up the mempool with txs with higher fee rate")
        for batch_of_txid in range(num_of_batches):
            fee_rate=(batch_of_txid + 1) * base_fee
            self.send_large_txs(miniwallet, txouts, fee_rate, tx_batch_size)

        self.log.info('The tx should be evicted by now')
        # The number of transactions created should be greater than the ones present in the mempool
        assert_greater_than(tx_batch_size * num_of_batches, len(self.nodes[0].getrawmempool()))
        # Initial tx created should not be present in the mempool anymore as it had a lower fee rate
        assert tx_to_be_evicted_id not in self.nodes[0].getrawmempool()

        self.log.info('Check that mempoolminfee is larger than minrelytxfee')
        assert_equal(self.nodes[0].getmempoolinfo()['minrelaytxfee'], Decimal('0.00001000'))
        assert_greater_than(self.nodes[0].getmempoolinfo()['mempoolminfee'], Decimal('0.00001000'))

        # Deliberately try to create a tx with a fee less than the minimum mempool fee to assert that it does not get added to the mempool
        self.log.info('Create a mempool tx that will not pass mempoolminfee')
        assert_raises_rpc_error(-26, "mempool min fee not met", miniwallet.send_self_transfer, from_node=self.nodes[0], fee_rate=relayfee, mempool_valid=False)


if __name__ == '__main__':
    MempoolLimitTest().main()

@@ -922,16 +922,22 @@
 xs(self,
+ node,
  miniwal
@@ -1069,37 +1069,28 @@
 r(from_node=
-self.nodes%5B0%5D
+node
 , fee_rate=f
@@ -1227,29 +1227,20 @@
 om_node=
-self.nodes%5B0%5D
+node
 , tx_hex
@@ -1324,16 +1324,43 @@
 xouts()%0A
+        node=self.nodes%5B0%5D%0A
         
@@ -1383,29 +1383,20 @@
 iWallet(
-self.nodes%5B0%5D
+node
 )%0A      
@@ -1408,29 +1408,20 @@
 ayfee = 
-self.nodes%5B0%5D
+node
 .getnetw
@@ -1526,37 +1526,28 @@
 ssert_equal(
-self.nodes%5B0%5D
+node
 .getmempooli
@@ -1610,37 +1610,28 @@
 ssert_equal(
-self.nodes%5B0%5D
+node
 .getmempooli
@@ -2243,29 +2243,20 @@
 enerate(
-self.nodes%5B0%5D
+node
 , COINBA
@@ -2405,37 +2405,28 @@
 r(from_node=
-self.nodes%5B0%5D
+node
 , fee_rate=r
@@ -2804,16 +2804,22 @@
 rge_txs(
+node, 
 miniwall
@@ -3077,29 +3077,20 @@
 es, len(
-self.nodes%5B0%5D
+node
 .getrawm
@@ -3243,29 +3243,20 @@
  not in 
-self.nodes%5B0%5D
+node
 .getrawm
@@ -3364,29 +3364,20 @@
 t_equal(
-self.nodes%5B0%5D
+node
 .getmemp
@@ -3455,29 +3455,20 @@
 er_than(
-self.nodes%5B0%5D
+node
 .getmemp
@@ -3841,29 +3841,20 @@
 om_node=
-self.nodes%5B0%5D
+node
 , fee_ra

self.nodes[0] with node