From d09e492bb269f2bf4e9da312963e1ed4bfb57a18 Mon Sep 17 00:00:00 2001
From: JoeGruff <joegruffins@gmail.com>
Date: Tue, 8 Dec 2020 11:30:03 +0900
Subject: [PATCH] rpcserver: Add handleGetMempoolInfo test.

---
 internal/rpcserver/rpcserverhandlers_test.go | 36 ++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/internal/rpcserver/rpcserverhandlers_test.go b/internal/rpcserver/rpcserverhandlers_test.go
index 255d2126..e88e05f2 100644
--- a/internal/rpcserver/rpcserverhandlers_test.go
+++ b/internal/rpcserver/rpcserverhandlers_test.go
@@ -4464,6 +4464,42 @@ func TestHandleGetInfo(t *testing.T) {
 	}})
 }
 
+func TestHandleGetMempoolInfo(t *testing.T) {
+	t.Parallel()
+
+	txDescOne := &mempool.TxDesc{
+		StartingPriority: 0,
+		TxDesc: mining.TxDesc{
+			Tx:   dcrutil.NewTx(block432100.Transactions[0]),
+			Type: stake.TxTypeSSGen,
+			Fee:  1,
+		},
+	}
+	txDescTwo := &mempool.TxDesc{
+		StartingPriority: 0,
+		TxDesc: mining.TxDesc{
+			Tx:   dcrutil.NewTx(block432100.Transactions[1]),
+			Type: stake.TxTypeRegular,
+			Fee:  300000,
+		},
+	}
+
+	testRPCServerHandler(t, []rpcTest{{
+		name:    "handleGetMempoolInfo: ok",
+		handler: handleGetMempoolInfo,
+		mockTxMempooler: func() *testTxMempooler {
+			mp := defaultMockTxMempooler()
+			mp.txDescs = []*mempool.TxDesc{txDescOne, txDescTwo}
+			return mp
+		}(),
+		cmd: &types.GetMempoolInfoCmd{},
+		result: &types.GetMempoolInfoResult{
+			Size:  2,
+			Bytes: 627,
+		},
+	}})
+}
+
 func TestHandleGetNetTotals(t *testing.T) {
 	t.Parallel()