Add 'composite' path option in testpoolswap

DeFiCh · Mar 2, 2022 · 8c38863 · 8c38863
1 parent 20f7ed2
commit 8c38863
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 2 deletions.
diff --git a/src/masternodes/rpc_poolpair.cpp b/src/masternodes/rpc_poolpair.cpp
@@ -997,6 +997,7 @@ UniValue testpoolswap(const JSONRPCRequest& request) {
                        "One of auto/direct (default = direct)\n"
                        "auto - automatically use composite swap or direct swap as needed.\n"
                        "direct - uses direct path only or fails.\n"
+                       "composite - uses composite path only or fails.\n"
                        "Note: The default will be switched to auto in the upcoming versions."
                    },
                    {
@@ -1051,7 +1052,7 @@ UniValue testpoolswap(const JSONRPCRequest& request) {
         int targetHeight = ::ChainActive().Height() + 1;
 
         auto poolPair = mnview_dummy.GetPoolPair(poolSwapMsg.idTokenFrom, poolSwapMsg.idTokenTo);
-        if (poolPair) path = "direct";
+        if (poolPair && path == "auto") path = "direct";
 
         // If no direct swap found search for composite swap
         if (path == "direct") {
@@ -1089,7 +1090,7 @@ UniValue testpoolswap(const JSONRPCRequest& request) {
             auto compositeSwap = CPoolSwap(poolSwapMsg, targetHeight);
 
             std::vector<DCT_ID> poolIds;
-            if (path == "auto") {
+            if (path == "auto" || path == "composite") {
                 poolIds = compositeSwap.CalculateSwaps(mnview_dummy, true);
             } else {
                 path = "custom";

diff --git a/test/functional/feature_poolswap_composite.py b/test/functional/feature_poolswap_composite.py
@@ -320,6 +320,30 @@ def run_test(self):
         assert_equal(estimateCompositePathsResAuto['path'], "direct")
         assert_equal(estimateCompositePathsResDirect, estimateCompositePathsResAuto)
 
+        estimateCompositePathsResComposite = self.nodes[0].testpoolswap({
+            "from": source,
+            "tokenFrom": symbolLTC,
+            "amountFrom": 1,
+            "to": destination,
+            "tokenTo": symbolDFI,
+        }, "composite", True)
+
+        assert_equal(estimateCompositePathsResComposite['path'], "composite")
+
+        poolLTC_USDC = list(self.nodes[0].getpoolpair("LTC-USDC").keys())[0]
+        poolDOGE_USDC = list(self.nodes[0].getpoolpair("DOGE-USDC").keys())[0]
+        poolDOGE_DFI = list(self.nodes[0].getpoolpair("DOGE-DFI").keys())[0]
+        assert_equal(estimateCompositePathsResComposite['pools'], [poolLTC_USDC, poolDOGE_USDC, poolDOGE_DFI])
+
+        assert_raises_rpc_error(-32600, "Cannot find usable pool pair.", self.nodes[0].testpoolswap,
+        {
+            "from": source,
+            "tokenFrom": symbolDUSD,
+            "amountFrom": 100,
+            "to": destination,
+            "tokenTo": symbolDFI,
+        }, "composite")
+
         assert_raises_rpc_error(-32600, "Custom pool path is invalid.", self.nodes[0].testpoolswap,
         {
             "from": source,