Inside Bitcoin

251        psbt_offline.unloadwallet()
252
253    def do_test(self, comment, pattern, privmap, treefn):
254        nkeys = len(privmap)
255        keys = random.sample(self.keys, nkeys * 4)
256        self.do_test_addr(comment, pattern, privmap, treefn, keys[0:nkeys])
257        self.do_test_sendtoaddress(comment, pattern, privmap, treefn, keys[0:nkeys], keys[nkeys:2*nkeys])
258        self.do_test_psbt(comment, pattern, privmap, treefn, keys[2*nkeys:3*nkeys], keys[3*nkeys:4*nkeys])
259
260    def generate_test_keys(self):
261        xprvs = [ExtendedPrivateKey.generate() for _ in range(0, 13)]
262        return [{
263            "xprv": xprv.to_string(),
264            "xpub": xprv.pubkey().to_string(),
265            "pubs": [compute_xonly_pubkey(xprv.derive_path(f"m/{i}").key.get_bytes())[0].hex() for i in range(0, 4)]
266        } for xprv in xprvs]
267
268    def run_test(self):
269        self.keys = self.generate_test_keys()
270        self.nodes[0].createwallet(wallet_name="boring")
271        self.boring = self.nodes[0].get_wallet_rpc("boring")
272
273        self.log.info("Mining blocks...")
274        gen_addr = self.boring.getnewaddress()
275        self.generatetoaddress(self.nodes[0], 101, gen_addr, sync_fun=self.no_op)
276
277        self.do_test(
278            "tr(XPRV)",
279            "tr($1/*)",
280            [True],
281            lambda k1: (key(k1), [])
282        )
283        self.do_test(
284            "tr(H,XPRV)",
285            "tr($H,pk($1/*))",
286            [True],
287            lambda k1: (key(H_POINT), [pk(k1)])
288        )
289        self.do_test(
290            "wpkh(XPRV)",
291            "wpkh($1/*)",
292            [True],
293            None
294        )
295        self.do_test(
296            "tr(XPRV,{H,{H,XPUB}})",
297            "tr($1/*,{pk($H),{pk($H),pk($2/*)}})",
298            [True, False],
299            lambda k1, k2: (key(k1), [pk(H_POINT), [pk(H_POINT), pk(k2)]])
300        )
301        self.do_test(
302            "wsh(multi(1,XPRV,XPUB))",
303            "wsh(multi(1,$1/*,$2/*))",
304            [True, False],
305            None
306        )
307        self.do_test(
308            "tr(XPRV,{XPUB,XPUB})",
309            "tr($1/*,{pk($2/*),pk($2/*)})",
310            [True, False],
311            lambda k1, k2: (key(k1), [pk(k2), pk(k2)])
312        )
313        self.do_test(
314            "tr(XPRV,{{XPUB,H},{H,XPUB}})",
315            "tr($1/*,{{pk($2/*),pk($H)},{pk($H),pk($2/*)}})",
316            [True, False],
317            lambda k1, k2: (key(k1), [[pk(k2), pk(H_POINT)], [pk(H_POINT), pk(k2)]])
318        )
319        self.do_test(
320            "tr(XPUB,{{H,{H,XPUB}},{H,{H,{H,XPRV}}}})",
321            "tr($1/*,{{pk($H),{pk($H),pk($2/*)}},{pk($H),{pk($H),{pk($H),pk($3/*)}}}})",
322            [False, False, True],
323            lambda k1, k2, k3: (key(k1), [[pk(H_POINT), [pk(H_POINT), pk(k2)]], [pk(H_POINT), [pk(H_POINT), [pk(H_POINT), pk(k3)]]]])
324        )
325        self.do_test(
326            "tr(XPRV,{XPUB,{{XPUB,{H,H}},{{H,H},XPUB}}})",
327            "tr($1/*,{pk($2/*),{{pk($2/*),{pk($H),pk($H)}},{{pk($H),pk($H)},pk($2/*)}}})",
328            [True, False],
329            lambda k1, k2: (key(k1), [pk(k2), [[pk(k2), [pk(H_POINT), pk(H_POINT)]], [[pk(H_POINT), pk(H_POINT)], pk(k2)]]])
330        )
331        self.do_test(
332            "tr(H,multi_a(1,XPRV))",
333            "tr($H,multi_a(1,$1/*))",
334            [True],
335            lambda k1: (key(H_POINT), [multi_a(1, [k1])])
336        )
337        self.do_test(
338            "tr(H,sortedmulti_a(1,XPRV,XPUB))",
339            "tr($H,sortedmulti_a(1,$1/*,$2/*))",
340            [True, False],
341            lambda k1, k2: (key(H_POINT), [multi_a(1, [k1, k2], True)])
342        )
343        self.do_test(
344            "tr(H,{H,multi_a(1,XPUB,XPRV)})",
345            "tr($H,{pk($H),multi_a(1,$1/*,$2/*)})",
346            [False, True],
347            lambda k1, k2: (key(H_POINT), [pk(H_POINT), [multi_a(1, [k1, k2])]])
348        )
349        self.do_test(
350            "tr(H,sortedmulti_a(1,XPUB,XPRV,XPRV))",
351            "tr($H,sortedmulti_a(1,$1/*,$2/*,$3/*))",
352            [False, True, True],
353            lambda k1, k2, k3: (key(H_POINT), [multi_a(1, [k1, k2, k3], True)])
354        )
355        self.do_test(
356            "tr(H,multi_a(2,XPRV,XPUB,XPRV))",
357            "tr($H,multi_a(2,$1/*,$2/*,$3/*))",
358            [True, False, True],
359            lambda k1, k2, k3: (key(H_POINT), [multi_a(2, [k1, k2, k3])])
360        )
361        self.do_test(
362            "tr(XPUB,{{XPUB,{XPUB,sortedmulti_a(2,XPRV,XPUB,XPRV)}})",
363            "tr($2/*,{pk($2/*),{pk($2/*),sortedmulti_a(2,$1/*,$2/*,$3/*)}})",
364            [True, False, True],
365            lambda k1, k2, k3: (key(k2), [pk(k2), [pk(k2), multi_a(2, [k1, k2, k3], True)]])
366        )
367        rnd_pos = random.randrange(MAX_PUBKEYS_PER_MULTI_A)
368        self.do_test(
369            "tr(XPUB,multi_a(1,H...,XPRV,H...))",
370            "tr($2/*,multi_a(1" + (",$H" * rnd_pos) + ",$1/*" + (",$H" * (MAX_PUBKEYS_PER_MULTI_A - 1 - rnd_pos)) + "))",
371            [True, False],
372            lambda k1, k2: (key(k2), [multi_a(1, ([H_POINT] * rnd_pos) + [k1] + ([H_POINT] * (MAX_PUBKEYS_PER_MULTI_A - 1 - rnd_pos)))])
373        )
374        self.do_test(
375            "rawtr(XPRV)",
376            "rawtr($1/*)",
377            [True],
378            lambda k1: key(k1)
379        )
380
381if __name__ == '__main__':
382    WalletTaprootTest(__file__).main()
383