rpc_sort_multisig.py raw

   1  #!/usr/bin/env python3
   2  # Copyright (c) 2014-2016 The Limenka developers
   3  # Distributed under the MIT software license, see the accompanying
   4  # file COPYING or http://www.opensource.org/licenses/mit-license.php.
   5  
   6  # Exercise the createmultisig API
   7  
   8  from test_framework.test_framework import LimenkaTestFramework
   9  from test_framework.util import (
  10      assert_equal,
  11      assert_raises_rpc_error,
  12  )
  13  
  14  class SortMultisigTest(LimenkaTestFramework):
  15      def set_test_params(self):
  16          self.num_nodes = 1
  17          self.extra_args = [[]]
  18          self.setup_clean_chain = True
  19  
  20      def run_simple_test(self):
  21          pub1 = "022df8750480ad5b26950b25c7ba79d3e37d75f640f8e5d9bcd5b150a0f85014da"
  22          pub2 = "03e3818b65bcc73a7d64064106a859cc1a5a728c4345ff0b641209fba0d90de6e9"
  23          pub3 = "021f2f6e1e50cb6a953935c3601284925decd3fd21bc445712576873fb8c6ebc18"
  24  
  25          pubs = [pub1,pub2,pub3]
  26  
  27          default = self.nodes[0].createmultisig(2, pubs)
  28          unsorted_ms = self.nodes[0].createmultisig(2, pubs, {"sort": False})
  29          assert_equal(unsorted_ms, self.nodes[0].createmultisig(2, pubs, options={"sort": False}))
  30          assert_equal(unsorted_ms, self.nodes[0].createmultisig(2, pubs, sort=False))
  31  
  32          assert_equal("2N2BchzwfyuqJep7sKmFfBucfopHZQuPnpt", unsorted_ms["address"])
  33          assert_equal("5221022df8750480ad5b26950b25c7ba79d3e37d75f640f8e5d9bcd5b150a0f85014da2103e3818b65bcc73a7d64064106a859cc1a5a728c4345ff0b641209fba0d90de6e921021f2f6e1e50cb6a953935c3601284925decd3fd21bc445712576873fb8c6ebc1853ae", unsorted_ms["redeemScript"])
  34          assert_equal(default["address"], unsorted_ms["address"])
  35          assert_equal(default["redeemScript"], unsorted_ms["redeemScript"])
  36  
  37          sorted_ms = self.nodes[0].createmultisig(2, pubs, {"sort": True})
  38          assert_equal(sorted_ms, self.nodes[0].createmultisig(2, pubs, options={"sort": True}))
  39          assert_equal(sorted_ms, self.nodes[0].createmultisig(2, pubs, sort=True))
  40          assert_equal("2NFd5JqpwmQNz3gevZJ3rz9ofuHvqaP9Cye", sorted_ms["address"])
  41          assert_equal("5221021f2f6e1e50cb6a953935c3601284925decd3fd21bc445712576873fb8c6ebc1821022df8750480ad5b26950b25c7ba79d3e37d75f640f8e5d9bcd5b150a0f85014da2103e3818b65bcc73a7d64064106a859cc1a5a728c4345ff0b641209fba0d90de6e953ae", sorted_ms["redeemScript"])
  42  
  43      def run_demonstrate_sorting(self):
  44          pub1 = "022df8750480ad5b26950b25c7ba79d3e37d75f640f8e5d9bcd5b150a0f85014da"
  45          pub2 = "03e3818b65bcc73a7d64064106a859cc1a5a728c4345ff0b641209fba0d90de6e9"
  46          pub3 = "021f2f6e1e50cb6a953935c3601284925decd3fd21bc445712576873fb8c6ebc18"
  47  
  48          sorted_ms = self.nodes[0].createmultisig(2, [pub3,pub1,pub2,])
  49  
  50          self.test_if_result_matches(2, [pub1,pub2,pub3], True, sorted_ms["address"])
  51          self.test_if_result_matches(2, [pub1,pub3,pub2], True, sorted_ms["address"])
  52          self.test_if_result_matches(2, [pub2,pub3,pub1], True, sorted_ms["address"])
  53          self.test_if_result_matches(2, [pub2,pub1,pub3], True, sorted_ms["address"])
  54          self.test_if_result_matches(2, [pub3,pub1,pub2], True, sorted_ms["address"])
  55          self.test_if_result_matches(2, [pub3,pub2,pub1], True, sorted_ms["address"])
  56  
  57          self.test_if_result_matches(2, [pub1,pub2,pub3], False, sorted_ms["address"])
  58          self.test_if_result_matches(2, [pub1,pub3,pub2], False, sorted_ms["address"])
  59          self.test_if_result_matches(2, [pub2,pub3,pub1], False, sorted_ms["address"])
  60          self.test_if_result_matches(2, [pub2,pub1,pub3], False, sorted_ms["address"])
  61          self.test_if_result_matches(2, [pub3,pub2,pub1], False, sorted_ms["address"])
  62  
  63      def test_if_result_matches(self, m, keys, sort, against):
  64          result = self.nodes[0].createmultisig(m, keys, {"sort": sort})
  65          assert_equal(sort, result["address"] == against)
  66  
  67      def test_compressed_keys_forbidden(self):
  68          pub1 = "02fdf7e1b65a477a7815effde996a03a7d94cbc46f7d14c05ef38425156fc92e22"
  69          pub2 = "04823336da95f0b4cf745839dff26992cef239ad2f08f494e5b57c209e4f3602d5526bc251d480e3284d129f736441560e17f3a7eb7ed665fdf0158f44550b926c"
  70          rs = "522102fdf7e1b65a477a7815effde996a03a7d94cbc46f7d14c05ef38425156fc92e224104823336da95f0b4cf745839dff26992cef239ad2f08f494e5b57c209e4f3602d5526bc251d480e3284d129f736441560e17f3a7eb7ed665fdf0158f44550b926c52ae"
  71          pubs = [pub1,pub2]
  72  
  73          default = self.nodes[0].createmultisig(2, pubs)
  74          assert_equal(rs, default["redeemScript"])
  75  
  76          unsorted_ms = self.nodes[0].createmultisig(2, pubs, {"sort": False})
  77          assert_equal(rs, unsorted_ms["redeemScript"])
  78          assert_equal(default["address"], unsorted_ms["address"])
  79          assert_equal(default["redeemScript"], unsorted_ms["redeemScript"])
  80  
  81          assert_raises_rpc_error(-1, "Compressed key required for BIP67: 04823336da95f0b4cf745839dff26992cef239ad2f08f494e5b57c209e4f3602d5526bc251d480e3284d129f736441560e17f3a7eb7ed665fdf0158f44550b926c", self.nodes[0].createmultisig, 2, pubs, {"sort": True})
  82  
  83      def run_test(self):
  84          self.run_simple_test()
  85          self.run_demonstrate_sorting()
  86          self.test_compressed_keys_forbidden()
  87  
  88  if __name__ == '__main__':
  89      SortMultisigTest(__file__).main()
  90  
  91