/testkit-core/src/main/scala/org/bitcoins/testkitcore/gen/AddressGenerator.scala

https://github.com/bitcoin-s/bitcoin-s-core · Scala · 45 lines · 34 code · 9 blank · 2 comment · 5 complexity · d051796f83139e5baad6f76958d4de51 MD5 · raw file 

    

  1. package org.bitcoins.testkitcore.gen
    2. 

  2. 

  3. import org.bitcoins.core.protocol._
    4. 

  4. import org.bitcoins.core.protocol.script.WitnessVersion0
    5. 

  5. import org.scalacheck.Gen
    6. 

  6. 

  7. /** Created by chris on 6/12/17.
    8. 

  8.   */
    9. 

  9. sealed trait AddressGenerator {
    10. 

  10. 

  11.   def p2pkhAddress: Gen[P2PKHAddress] =
    12. 

  12.     for {
    13. 

  13.       hash <- CryptoGenerators.sha256Hash160Digest
    14. 

  14.       network <- ChainParamsGenerator.networkParams
    15. 

  15.       addr = P2PKHAddress(hash, network)
    16. 

  16.     } yield addr
    17. 

  17. 

  18.   def p2shAddress: Gen[P2SHAddress] =
    19. 

  19.     for {
    20. 

  20.       hash <- CryptoGenerators.sha256Hash160Digest
    21. 

  21.       network <- ChainParamsGenerator.networkParams
    22. 

  22.       addr = P2SHAddress(hash, network)
    23. 

  23.     } yield addr
    24. 

  24. 

  25.   def bech32Address: Gen[Bech32Address] =
    26. 

  26.     for {
    27. 

  27.       (witSPKV0, _) <- ScriptGenerators.witnessScriptPubKeyV0
    28. 

  28.       network <- ChainParamsGenerator.networkParams
    29. 

  29.       addr = Bech32Address(witSPKV0, network)
    30. 

  30.     } yield addr
    31. 

  31. 

  32.   def bech32mAddress: Gen[Bech32mAddress] =
    33. 

  33.     for {
    34. 

  34.       (witSPK, _) <- ScriptGenerators.witnessScriptPubKey.suchThat(
    35. 

  35.         _._1.witnessVersion != WitnessVersion0)
    36. 

  36.       network <- ChainParamsGenerator.networkParams
    37. 

  37.     } yield Bech32mAddress(witSPK, network)
    38. 

  38. 

  39.   def bitcoinAddress: Gen[BitcoinAddress] =
    40. 

  40.     Gen.oneOf(p2pkhAddress, p2shAddress, bech32Address, bech32mAddress)
    41. 

  41. 

  42.   def address: Gen[Address] = bitcoinAddress
    43. 

  43. }
    44. 

  44. 

  45. object AddressGenerator extends AddressGenerator
    46.