From 1a93aa4b793586803bd67374c2afc7eeb634316a Mon Sep 17 00:00:00 2001
From: moneromooo-monero <moneromooo-monero@users.noreply.github.com>
Date: Wed, 24 Apr 2019 13:25:21 +0000
Subject: [PATCH] functional_tests: add get_fee_estimate to blockchain test

---
 tests/functional_tests/blockchain.py | 12 ++++++++++++
 utils/python-rpc/framework/daemon.py | 11 +++++++++++
 2 files changed, 23 insertions(+)

diff --git a/tests/functional_tests/blockchain.py b/tests/functional_tests/blockchain.py
index 56164600d..bdf72b72c 100755
--- a/tests/functional_tests/blockchain.py
+++ b/tests/functional_tests/blockchain.py
@@ -70,6 +70,12 @@ class BlockchainTest():
         except: ok = True
         assert ok
 
+        res = daemon.get_fee_estimate()
+        assert res.fee == 234562
+        assert res.quantization_mask == 10000
+        res = daemon.get_fee_estimate(10)
+        assert res.fee <= 234562
+
         # generate blocks
         res_generateblocks = daemon.generateblocks('42ey1afDFnn4886T7196doS9GPMzexD9gXpsZJDwVjeRVdFCSoHnv7KPbBeGpzJBzHRCAs9UxqeoyFQMYbqSWYTfJJQAWDm', blocks)
 
@@ -221,6 +227,12 @@ class BlockchainTest():
             assert res.histogram[i].unlocked_instances == 0
             assert res.histogram[i].recent_instances == 0
 
+        res = daemon.get_fee_estimate()
+        assert res.fee == 234560
+        assert res.quantization_mask == 10000
+        res = daemon.get_fee_estimate(10)
+        assert res.fee <= 234560
+
     def _test_alt_chains(self):
         print('Testing alt chains')
         daemon = Daemon()
diff --git a/utils/python-rpc/framework/daemon.py b/utils/python-rpc/framework/daemon.py
index 04fc5b5cf..bbac31980 100644
--- a/utils/python-rpc/framework/daemon.py
+++ b/utils/python-rpc/framework/daemon.py
@@ -330,3 +330,14 @@ class Daemon(object):
             'id': '0'
         }
         return self.rpc.send_json_rpc_request(get_alternate_chains)
+
+    def get_fee_estimate(self, grace_blocks = 0):
+        get_fee_estimate = {
+            'method': 'get_fee_estimate',
+            'params': {
+                'grace_blocks': grace_blocks,
+            },
+            'jsonrpc': '2.0',
+            'id': '0'
+        }
+        return self.rpc.send_json_rpc_request(get_fee_estimate)