add unit tests for find-affordable-networks

status-im · Apr 3, 2024 · 12fbba8 · 12fbba8
1 parent dabdce6
commit 12fbba8
Showing 1 changed file with 41 additions and 0 deletions.
diff --git a/src/status_im/contexts/wallet/common/utils/send_test.cljs b/src/status_im/contexts/wallet/common/utils/send_test.cljs
@@ -20,3 +20,44 @@
             expected-eip1559-disabled-result (money/bignumber 0.000760406)]
         (is (money/equal-to (utils/calculate-gas-fee data-eip1559-disabled)
                             expected-eip1559-disabled-result))))))
+
+(deftest test-find-affordable-networks
+  (testing "All networks affordable and selected, none disabled"
+    (let [balances-per-chain {"1" {:balance "50.0" :chain-id "1"}
+                              "2" {:balance "40.0" :chain-id "2"}}
+          input-value        20
+          selected-networks  ["1" "2"]
+          disabled-chain-ids []
+          expected           ["1" "2"]]
+      (is (= (set (utils/find-affordable-networks {:balances-per-chain balances-per-chain
+                                                   :input-value        input-value
+                                                   :selected-networks  selected-networks
+                                                   :disabled-chain-ids disabled-chain-ids}))
+             (set expected)))))
+
+  (testing "No networks affordable"
+    (let [balances-per-chain {"1" {:balance "5.0" :chain-id "1"}
+                              "2" {:balance "1.0" :chain-id "2"}}
+          input-value        10
+          selected-networks  ["1" "2"]
+          disabled-chain-ids []
+          expected           []]
+      (is (= (set (utils/find-affordable-networks {:balances-per-chain balances-per-chain
+                                                   :input-value        input-value
+                                                   :selected-networks  selected-networks
+                                                   :disabled-chain-ids disabled-chain-ids}))
+             (set expected)))))
+
+  (testing "Selected networks subset, with some disabled"
+    (let [balances-per-chain {"1" {:balance "100.0" :chain-id "1"}
+                              "2" {:balance "50.0" :chain-id "2"}
+                              "3" {:balance "20.0" :chain-id "3"}}
+          input-value        15
+          selected-networks  ["1" "2" "3"]
+          disabled-chain-ids ["2"]
+          expected           ["1" "3"]]
+      (is (= (set (utils/find-affordable-networks {:balances-per-chain balances-per-chain
+                                                   :input-value        input-value
+                                                   :selected-networks  selected-networks
+                                                   :disabled-chain-ids disabled-chain-ids}))
+             (set expected))))))