Skip to content

Commit 806b256

Browse files
committed
f - make explicit test pubkeys
1 parent 1af5d5e commit 806b256

File tree

1 file changed

+44
-62
lines changed

1 file changed

+44
-62
lines changed

lightning/src/offers/offer.rs

Lines changed: 44 additions & 62 deletions
Original file line numberDiff line numberDiff line change
@@ -427,27 +427,18 @@ mod tests {
427427
use ln::features::OfferFeatures;
428428
use onion_message::{BlindedHop, BlindedPath};
429429

430-
fn pubkey() -> PublicKey {
430+
fn pubkey(byte: u8) -> PublicKey {
431431
let secp_ctx = Secp256k1::new();
432-
PublicKey::from_secret_key(&secp_ctx, &privkey())
432+
PublicKey::from_secret_key(&secp_ctx, &privkey(byte))
433433
}
434434

435-
fn privkey() -> SecretKey {
436-
SecretKey::from_slice(&[42; 32]).unwrap()
437-
}
438-
439-
fn blinded_pubkey(byte: u8) -> PublicKey {
440-
let secp_ctx = Secp256k1::new();
441-
PublicKey::from_secret_key(&secp_ctx, &blinded_privkey(byte))
442-
}
443-
444-
fn blinded_privkey(byte: u8) -> SecretKey {
435+
fn privkey(byte: u8) -> SecretKey {
445436
SecretKey::from_slice(&[byte; 32]).unwrap()
446437
}
447438

448439
#[test]
449440
fn builds_offer_with_defaults() {
450-
let offer = OfferBuilder::new("foo".into(), pubkey()).build();
441+
let offer = OfferBuilder::new("foo".into(), pubkey(42)).build();
451442
let tlv_stream = offer.as_tlv_stream();
452443

453444
assert_eq!(offer.as_bytes(), &offer.to_bytes()[..]);
@@ -463,7 +454,7 @@ mod tests {
463454
assert_eq!(offer.issuer(), None);
464455
assert_eq!(offer.quantity_min(), 1);
465456
assert_eq!(offer.quantity_max(), 1);
466-
assert_eq!(offer.node_id(), pubkey());
457+
assert_eq!(offer.node_id(), pubkey(42));
467458

468459
assert_eq!(tlv_stream.chains, None);
469460
assert_eq!(tlv_stream.metadata, None);
@@ -476,7 +467,7 @@ mod tests {
476467
assert_eq!(tlv_stream.issuer, None);
477468
assert_eq!(tlv_stream.quantity_min, None);
478469
assert_eq!(tlv_stream.quantity_max, None);
479-
assert_eq!(tlv_stream.node_id, Some(&pubkey()));
470+
assert_eq!(tlv_stream.node_id, Some(&pubkey(42)));
480471
}
481472

482473
#[test]
@@ -487,20 +478,20 @@ mod tests {
487478
ChainHash::using_genesis_block(Network::Testnet),
488479
];
489480

490-
let offer = OfferBuilder::new("foo".into(), pubkey())
481+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
491482
.chain(Network::Bitcoin)
492483
.build();
493484
assert_eq!(offer.chains(), vec![chain]);
494485
assert_eq!(offer.as_tlv_stream().chains, Some(&vec![chain]));
495486

496-
let offer = OfferBuilder::new("foo".into(), pubkey())
487+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
497488
.chain(Network::Bitcoin)
498489
.chain(Network::Bitcoin)
499490
.build();
500491
assert_eq!(offer.chains(), vec![chain]);
501492
assert_eq!(offer.as_tlv_stream().chains, Some(&vec![chain]));
502493

503-
let offer = OfferBuilder::new("foo".into(), pubkey())
494+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
504495
.chain(Network::Bitcoin)
505496
.chain(Network::Testnet)
506497
.build();
@@ -510,13 +501,13 @@ mod tests {
510501

511502
#[test]
512503
fn builds_offer_with_metadata() {
513-
let offer = OfferBuilder::new("foo".into(), pubkey())
504+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
514505
.metadata(vec![42; 32])
515506
.build();
516507
assert_eq!(offer.metadata(), Some(&vec![42; 32]));
517508
assert_eq!(offer.as_tlv_stream().metadata, Some(&vec![42; 32]));
518509

519-
let offer = OfferBuilder::new("foo".into(), pubkey())
510+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
520511
.metadata(vec![42; 32])
521512
.metadata(vec![43; 32])
522513
.build();
@@ -529,23 +520,23 @@ mod tests {
529520
let bitcoin_amount = Amount::Bitcoin { amount_msats: 1000 };
530521
let currency_amount = Amount::Currency { iso4217_code: *b"USD", amount: 10 };
531522

532-
let offer = OfferBuilder::new("foo".into(), pubkey())
523+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
533524
.amount(bitcoin_amount.clone())
534525
.build();
535526
let tlv_stream = offer.as_tlv_stream();
536527
assert_eq!(offer.amount(), Some(&bitcoin_amount));
537528
assert_eq!(tlv_stream.amount, Some(1000));
538529
assert_eq!(tlv_stream.currency, None);
539530

540-
let offer = OfferBuilder::new("foo".into(), pubkey())
531+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
541532
.amount(currency_amount.clone())
542533
.build();
543534
let tlv_stream = offer.as_tlv_stream();
544535
assert_eq!(offer.amount(), Some(&currency_amount));
545536
assert_eq!(tlv_stream.amount, Some(10));
546537
assert_eq!(tlv_stream.currency, Some(b"USD"));
547538

548-
let offer = OfferBuilder::new("foo".into(), pubkey())
539+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
549540
.amount(currency_amount.clone())
550541
.amount(bitcoin_amount.clone())
551542
.build();
@@ -556,13 +547,13 @@ mod tests {
556547

557548
#[test]
558549
fn builds_offer_with_features() {
559-
let offer = OfferBuilder::new("foo".into(), pubkey())
550+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
560551
.features(OfferFeatures::empty())
561552
.build();
562553
assert_eq!(offer.features(), Some(&OfferFeatures::empty()));
563554
assert_eq!(offer.as_tlv_stream().features, Some(&OfferFeatures::empty()));
564555

565-
let offer = OfferBuilder::new("foo".into(), pubkey())
556+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
566557
.features(OfferFeatures::unknown())
567558
.features(OfferFeatures::empty())
568559
.build();
@@ -575,15 +566,15 @@ mod tests {
575566
let future_expiry = Duration::from_secs(u64::max_value());
576567
let past_expiry = Duration::from_secs(0);
577568

578-
let offer = OfferBuilder::new("foo".into(), pubkey())
569+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
579570
.absolute_expiry(future_expiry)
580571
.build();
581572
#[cfg(feature = "std")]
582573
assert!(!offer.is_expired());
583574
assert_eq!(offer.absolute_expiry(), Some(future_expiry));
584575
assert_eq!(offer.as_tlv_stream().absolute_expiry, Some(future_expiry.as_secs()));
585576

586-
let offer = OfferBuilder::new("foo".into(), pubkey())
577+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
587578
.absolute_expiry(future_expiry)
588579
.absolute_expiry(past_expiry)
589580
.build();
@@ -595,55 +586,46 @@ mod tests {
595586

596587
#[test]
597588
fn builds_offer_with_paths() {
598-
// TODO: Use more realistic data
599589
let paths = vec![
600590
BlindedPath {
601-
introduction_node_id: pubkey(),
602-
blinding_point: pubkey(),
591+
introduction_node_id: pubkey(40),
592+
blinding_point: pubkey(41),
603593
blinded_hops: vec![
604-
BlindedHop {
605-
blinded_node_id: blinded_pubkey(43), encrypted_payload: vec![0; 43],
606-
},
607-
BlindedHop {
608-
blinded_node_id: blinded_pubkey(44), encrypted_payload: vec![0; 44],
609-
},
594+
BlindedHop { blinded_node_id: pubkey(43), encrypted_payload: vec![0; 43] },
595+
BlindedHop { blinded_node_id: pubkey(44), encrypted_payload: vec![0; 44] },
610596
],
611597
},
612598
BlindedPath {
613-
introduction_node_id: pubkey(),
614-
blinding_point: pubkey(),
599+
introduction_node_id: pubkey(40),
600+
blinding_point: pubkey(41),
615601
blinded_hops: vec![
616-
BlindedHop {
617-
blinded_node_id: blinded_pubkey(45), encrypted_payload: vec![0; 45],
618-
},
619-
BlindedHop {
620-
blinded_node_id: blinded_pubkey(46), encrypted_payload: vec![0; 46],
621-
},
602+
BlindedHop { blinded_node_id: pubkey(45), encrypted_payload: vec![0; 45] },
603+
BlindedHop { blinded_node_id: pubkey(46), encrypted_payload: vec![0; 46] },
622604
],
623605
},
624606
];
625607

626-
let offer = OfferBuilder::new("foo".into(), pubkey())
608+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
627609
.path(paths[0].clone())
628610
.path(paths[1].clone())
629611
.build();
630612
let tlv_stream = offer.as_tlv_stream();
631613
assert_eq!(offer.paths(), Some(&paths));
632-
assert_eq!(offer.node_id(), pubkey());
633-
assert_ne!(pubkey(), blinded_pubkey(44));
614+
assert_eq!(offer.node_id(), pubkey(42));
615+
assert_ne!(pubkey(42), pubkey(44));
634616
assert_eq!(tlv_stream.paths, Some(&paths));
635-
assert_eq!(tlv_stream.node_id, Some(&pubkey()));
617+
assert_eq!(tlv_stream.node_id, Some(&pubkey(42)));
636618
}
637619

638620
#[test]
639621
fn builds_offer_with_issuer() {
640-
let offer = OfferBuilder::new("foo".into(), pubkey())
622+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
641623
.issuer("bar".into())
642624
.build();
643625
assert_eq!(offer.issuer(), Some(&String::from("bar")));
644626
assert_eq!(offer.as_tlv_stream().issuer, Some(&String::from("bar")));
645627

646-
let offer = OfferBuilder::new("foo".into(), pubkey())
628+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
647629
.issuer("bar".into())
648630
.issuer("baz".into())
649631
.build();
@@ -657,7 +639,7 @@ mod tests {
657639
let five = NonZeroU64::new(5).unwrap();
658640
let ten = NonZeroU64::new(10).unwrap();
659641

660-
let offer = OfferBuilder::new("foo".into(), pubkey())
642+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
661643
.quantity_fixed(one)
662644
.build();
663645
let tlv_stream = offer.as_tlv_stream();
@@ -666,7 +648,7 @@ mod tests {
666648
assert_eq!(tlv_stream.quantity_min, None);
667649
assert_eq!(tlv_stream.quantity_max, None);
668650

669-
let offer = OfferBuilder::new("foo".into(), pubkey())
651+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
670652
.quantity_fixed(ten)
671653
.build();
672654
let tlv_stream = offer.as_tlv_stream();
@@ -675,7 +657,7 @@ mod tests {
675657
assert_eq!(tlv_stream.quantity_min, Some(10));
676658
assert_eq!(tlv_stream.quantity_max, Some(10));
677659

678-
let offer = OfferBuilder::new("foo".into(), pubkey())
660+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
679661
.quantity_fixed(ten)
680662
.quantity_fixed(five)
681663
.build();
@@ -685,7 +667,7 @@ mod tests {
685667
assert_eq!(tlv_stream.quantity_min, Some(5));
686668
assert_eq!(tlv_stream.quantity_max, Some(5));
687669

688-
let offer = OfferBuilder::new("foo".into(), pubkey())
670+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
689671
.quantity_range(..ten)
690672
.quantity_fixed(five)
691673
.build();
@@ -702,7 +684,7 @@ mod tests {
702684
let five = NonZeroU64::new(5).unwrap();
703685
let ten = NonZeroU64::new(10).unwrap();
704686

705-
let offer = OfferBuilder::new("foo".into(), pubkey())
687+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
706688
.quantity_range(..)
707689
.build();
708690
let tlv_stream = offer.as_tlv_stream();
@@ -711,7 +693,7 @@ mod tests {
711693
assert_eq!(tlv_stream.quantity_min, Some(1));
712694
assert_eq!(tlv_stream.quantity_max, None);
713695

714-
let offer = OfferBuilder::new("foo".into(), pubkey())
696+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
715697
.quantity_range(..ten)
716698
.build();
717699
let tlv_stream = offer.as_tlv_stream();
@@ -720,7 +702,7 @@ mod tests {
720702
assert_eq!(tlv_stream.quantity_min, None);
721703
assert_eq!(tlv_stream.quantity_max, Some(9));
722704

723-
let offer = OfferBuilder::new("foo".into(), pubkey())
705+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
724706
.quantity_range(one..ten)
725707
.build();
726708
let tlv_stream = offer.as_tlv_stream();
@@ -729,7 +711,7 @@ mod tests {
729711
assert_eq!(tlv_stream.quantity_min, None);
730712
assert_eq!(tlv_stream.quantity_max, Some(9));
731713

732-
let offer = OfferBuilder::new("foo".into(), pubkey())
714+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
733715
.quantity_range(five..=ten)
734716
.build();
735717
let tlv_stream = offer.as_tlv_stream();
@@ -739,7 +721,7 @@ mod tests {
739721
assert_eq!(tlv_stream.quantity_max, Some(10));
740722

741723
let one = NonZeroU64::new(1).unwrap();
742-
let offer = OfferBuilder::new("foo".into(), pubkey())
724+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
743725
.quantity_range(one..=one)
744726
.build();
745727
let tlv_stream = offer.as_tlv_stream();
@@ -748,7 +730,7 @@ mod tests {
748730
assert_eq!(tlv_stream.quantity_min, None);
749731
assert_eq!(tlv_stream.quantity_max, None);
750732

751-
let offer = OfferBuilder::new("foo".into(), pubkey())
733+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
752734
.quantity_range(five..=five)
753735
.build();
754736
let tlv_stream = offer.as_tlv_stream();
@@ -757,7 +739,7 @@ mod tests {
757739
assert_eq!(tlv_stream.quantity_min, Some(5));
758740
assert_eq!(tlv_stream.quantity_max, Some(5));
759741

760-
let offer = OfferBuilder::new("foo".into(), pubkey())
742+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
761743
.quantity_range(ten..five)
762744
.build();
763745
let tlv_stream = offer.as_tlv_stream();
@@ -766,7 +748,7 @@ mod tests {
766748
assert_eq!(tlv_stream.quantity_min, None);
767749
assert_eq!(tlv_stream.quantity_max, None);
768750

769-
let offer = OfferBuilder::new("foo".into(), pubkey())
751+
let offer = OfferBuilder::new("foo".into(), pubkey(42))
770752
.quantity_fixed(five)
771753
.quantity_range(..ten)
772754
.build();

0 commit comments

Comments
 (0)