chia-blockchain 2.5.1rc1__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (1042) hide show
  1. chia/__init__.py +10 -0
  2. chia/__main__.py +5 -0
  3. chia/_tests/README.md +53 -0
  4. chia/_tests/__init__.py +0 -0
  5. chia/_tests/blockchain/__init__.py +0 -0
  6. chia/_tests/blockchain/blockchain_test_utils.py +195 -0
  7. chia/_tests/blockchain/config.py +4 -0
  8. chia/_tests/blockchain/test_augmented_chain.py +145 -0
  9. chia/_tests/blockchain/test_blockchain.py +4202 -0
  10. chia/_tests/blockchain/test_blockchain_transactions.py +1031 -0
  11. chia/_tests/blockchain/test_build_chains.py +59 -0
  12. chia/_tests/blockchain/test_get_block_generator.py +72 -0
  13. chia/_tests/blockchain/test_lookup_fork_chain.py +194 -0
  14. chia/_tests/build-init-files.py +92 -0
  15. chia/_tests/build-job-matrix.py +204 -0
  16. chia/_tests/check_pytest_monitor_output.py +34 -0
  17. chia/_tests/check_sql_statements.py +72 -0
  18. chia/_tests/chia-start-sim +42 -0
  19. chia/_tests/clvm/__init__.py +0 -0
  20. chia/_tests/clvm/benchmark_costs.py +23 -0
  21. chia/_tests/clvm/coin_store.py +149 -0
  22. chia/_tests/clvm/test_chialisp_deserialization.py +101 -0
  23. chia/_tests/clvm/test_clvm_step.py +37 -0
  24. chia/_tests/clvm/test_condition_codes.py +13 -0
  25. chia/_tests/clvm/test_curry_and_treehash.py +55 -0
  26. chia/_tests/clvm/test_message_conditions.py +184 -0
  27. chia/_tests/clvm/test_program.py +150 -0
  28. chia/_tests/clvm/test_puzzle_compression.py +143 -0
  29. chia/_tests/clvm/test_puzzle_drivers.py +45 -0
  30. chia/_tests/clvm/test_puzzles.py +242 -0
  31. chia/_tests/clvm/test_singletons.py +540 -0
  32. chia/_tests/clvm/test_spend_sim.py +181 -0
  33. chia/_tests/cmds/__init__.py +0 -0
  34. chia/_tests/cmds/cmd_test_utils.py +469 -0
  35. chia/_tests/cmds/config.py +3 -0
  36. chia/_tests/cmds/conftest.py +23 -0
  37. chia/_tests/cmds/test_click_types.py +200 -0
  38. chia/_tests/cmds/test_cmd_framework.py +620 -0
  39. chia/_tests/cmds/test_cmds_util.py +97 -0
  40. chia/_tests/cmds/test_daemon.py +92 -0
  41. chia/_tests/cmds/test_dev_gh.py +131 -0
  42. chia/_tests/cmds/test_farm_cmd.py +66 -0
  43. chia/_tests/cmds/test_show.py +116 -0
  44. chia/_tests/cmds/test_sim.py +207 -0
  45. chia/_tests/cmds/test_timelock_args.py +75 -0
  46. chia/_tests/cmds/test_tx_config_args.py +154 -0
  47. chia/_tests/cmds/testing_classes.py +59 -0
  48. chia/_tests/cmds/wallet/__init__.py +0 -0
  49. chia/_tests/cmds/wallet/test_consts.py +47 -0
  50. chia/_tests/cmds/wallet/test_dao.py +565 -0
  51. chia/_tests/cmds/wallet/test_did.py +403 -0
  52. chia/_tests/cmds/wallet/test_nft.py +471 -0
  53. chia/_tests/cmds/wallet/test_notifications.py +124 -0
  54. chia/_tests/cmds/wallet/test_offer.toffer +1 -0
  55. chia/_tests/cmds/wallet/test_tx_decorators.py +27 -0
  56. chia/_tests/cmds/wallet/test_vcs.py +400 -0
  57. chia/_tests/cmds/wallet/test_wallet.py +1125 -0
  58. chia/_tests/cmds/wallet/test_wallet_check.py +109 -0
  59. chia/_tests/conftest.py +1419 -0
  60. chia/_tests/connection_utils.py +125 -0
  61. chia/_tests/core/__init__.py +0 -0
  62. chia/_tests/core/cmds/__init__.py +0 -0
  63. chia/_tests/core/cmds/test_beta.py +382 -0
  64. chia/_tests/core/cmds/test_keys.py +1734 -0
  65. chia/_tests/core/cmds/test_wallet.py +126 -0
  66. chia/_tests/core/config.py +3 -0
  67. chia/_tests/core/consensus/__init__.py +0 -0
  68. chia/_tests/core/consensus/test_block_creation.py +54 -0
  69. chia/_tests/core/consensus/test_pot_iterations.py +117 -0
  70. chia/_tests/core/custom_types/__init__.py +0 -0
  71. chia/_tests/core/custom_types/test_coin.py +107 -0
  72. chia/_tests/core/custom_types/test_proof_of_space.py +144 -0
  73. chia/_tests/core/custom_types/test_spend_bundle.py +70 -0
  74. chia/_tests/core/daemon/__init__.py +0 -0
  75. chia/_tests/core/daemon/config.py +4 -0
  76. chia/_tests/core/daemon/test_daemon.py +2128 -0
  77. chia/_tests/core/daemon/test_daemon_register.py +109 -0
  78. chia/_tests/core/daemon/test_keychain_proxy.py +101 -0
  79. chia/_tests/core/data_layer/__init__.py +0 -0
  80. chia/_tests/core/data_layer/config.py +5 -0
  81. chia/_tests/core/data_layer/conftest.py +106 -0
  82. chia/_tests/core/data_layer/test_data_cli.py +56 -0
  83. chia/_tests/core/data_layer/test_data_layer.py +83 -0
  84. chia/_tests/core/data_layer/test_data_layer_util.py +218 -0
  85. chia/_tests/core/data_layer/test_data_rpc.py +3847 -0
  86. chia/_tests/core/data_layer/test_data_store.py +2424 -0
  87. chia/_tests/core/data_layer/test_data_store_schema.py +381 -0
  88. chia/_tests/core/data_layer/test_plugin.py +91 -0
  89. chia/_tests/core/data_layer/util.py +233 -0
  90. chia/_tests/core/farmer/__init__.py +0 -0
  91. chia/_tests/core/farmer/config.py +3 -0
  92. chia/_tests/core/farmer/test_farmer_api.py +103 -0
  93. chia/_tests/core/full_node/__init__.py +0 -0
  94. chia/_tests/core/full_node/config.py +4 -0
  95. chia/_tests/core/full_node/dos/__init__.py +0 -0
  96. chia/_tests/core/full_node/dos/config.py +3 -0
  97. chia/_tests/core/full_node/full_sync/__init__.py +0 -0
  98. chia/_tests/core/full_node/full_sync/config.py +4 -0
  99. chia/_tests/core/full_node/full_sync/test_full_sync.py +443 -0
  100. chia/_tests/core/full_node/ram_db.py +27 -0
  101. chia/_tests/core/full_node/stores/__init__.py +0 -0
  102. chia/_tests/core/full_node/stores/config.py +4 -0
  103. chia/_tests/core/full_node/stores/test_block_store.py +590 -0
  104. chia/_tests/core/full_node/stores/test_coin_store.py +897 -0
  105. chia/_tests/core/full_node/stores/test_full_node_store.py +1219 -0
  106. chia/_tests/core/full_node/stores/test_hint_store.py +229 -0
  107. chia/_tests/core/full_node/stores/test_sync_store.py +135 -0
  108. chia/_tests/core/full_node/test_address_manager.py +588 -0
  109. chia/_tests/core/full_node/test_block_height_map.py +556 -0
  110. chia/_tests/core/full_node/test_conditions.py +556 -0
  111. chia/_tests/core/full_node/test_full_node.py +2700 -0
  112. chia/_tests/core/full_node/test_generator_tools.py +82 -0
  113. chia/_tests/core/full_node/test_hint_management.py +104 -0
  114. chia/_tests/core/full_node/test_node_load.py +34 -0
  115. chia/_tests/core/full_node/test_performance.py +179 -0
  116. chia/_tests/core/full_node/test_subscriptions.py +492 -0
  117. chia/_tests/core/full_node/test_transactions.py +203 -0
  118. chia/_tests/core/full_node/test_tx_processing_queue.py +155 -0
  119. chia/_tests/core/large_block.py +2388 -0
  120. chia/_tests/core/make_block_generator.py +70 -0
  121. chia/_tests/core/mempool/__init__.py +0 -0
  122. chia/_tests/core/mempool/config.py +4 -0
  123. chia/_tests/core/mempool/test_mempool.py +3255 -0
  124. chia/_tests/core/mempool/test_mempool_fee_estimator.py +104 -0
  125. chia/_tests/core/mempool/test_mempool_fee_protocol.py +55 -0
  126. chia/_tests/core/mempool/test_mempool_item_queries.py +190 -0
  127. chia/_tests/core/mempool/test_mempool_manager.py +2084 -0
  128. chia/_tests/core/mempool/test_mempool_performance.py +64 -0
  129. chia/_tests/core/mempool/test_singleton_fast_forward.py +567 -0
  130. chia/_tests/core/node_height.py +28 -0
  131. chia/_tests/core/server/__init__.py +0 -0
  132. chia/_tests/core/server/config.py +3 -0
  133. chia/_tests/core/server/flood.py +84 -0
  134. chia/_tests/core/server/serve.py +135 -0
  135. chia/_tests/core/server/test_api_protocol.py +21 -0
  136. chia/_tests/core/server/test_capabilities.py +66 -0
  137. chia/_tests/core/server/test_dos.py +319 -0
  138. chia/_tests/core/server/test_event_loop.py +109 -0
  139. chia/_tests/core/server/test_loop.py +294 -0
  140. chia/_tests/core/server/test_node_discovery.py +73 -0
  141. chia/_tests/core/server/test_rate_limits.py +482 -0
  142. chia/_tests/core/server/test_server.py +226 -0
  143. chia/_tests/core/server/test_upnp.py +8 -0
  144. chia/_tests/core/services/__init__.py +0 -0
  145. chia/_tests/core/services/config.py +3 -0
  146. chia/_tests/core/services/test_services.py +188 -0
  147. chia/_tests/core/ssl/__init__.py +0 -0
  148. chia/_tests/core/ssl/config.py +3 -0
  149. chia/_tests/core/ssl/test_ssl.py +202 -0
  150. chia/_tests/core/test_coins.py +33 -0
  151. chia/_tests/core/test_cost_calculation.py +313 -0
  152. chia/_tests/core/test_crawler.py +175 -0
  153. chia/_tests/core/test_crawler_rpc.py +53 -0
  154. chia/_tests/core/test_daemon_rpc.py +24 -0
  155. chia/_tests/core/test_db_conversion.py +130 -0
  156. chia/_tests/core/test_db_validation.py +162 -0
  157. chia/_tests/core/test_farmer_harvester_rpc.py +505 -0
  158. chia/_tests/core/test_filter.py +35 -0
  159. chia/_tests/core/test_full_node_rpc.py +768 -0
  160. chia/_tests/core/test_merkle_set.py +343 -0
  161. chia/_tests/core/test_program.py +47 -0
  162. chia/_tests/core/test_rpc_util.py +86 -0
  163. chia/_tests/core/test_seeder.py +420 -0
  164. chia/_tests/core/test_setproctitle.py +13 -0
  165. chia/_tests/core/util/__init__.py +0 -0
  166. chia/_tests/core/util/config.py +4 -0
  167. chia/_tests/core/util/test_block_cache.py +44 -0
  168. chia/_tests/core/util/test_cached_bls.py +57 -0
  169. chia/_tests/core/util/test_config.py +337 -0
  170. chia/_tests/core/util/test_file_keyring_synchronization.py +105 -0
  171. chia/_tests/core/util/test_files.py +391 -0
  172. chia/_tests/core/util/test_jsonify.py +146 -0
  173. chia/_tests/core/util/test_keychain.py +522 -0
  174. chia/_tests/core/util/test_keyring_wrapper.py +491 -0
  175. chia/_tests/core/util/test_lockfile.py +380 -0
  176. chia/_tests/core/util/test_log_exceptions.py +187 -0
  177. chia/_tests/core/util/test_lru_cache.py +56 -0
  178. chia/_tests/core/util/test_significant_bits.py +40 -0
  179. chia/_tests/core/util/test_streamable.py +883 -0
  180. chia/_tests/db/__init__.py +0 -0
  181. chia/_tests/db/test_db_wrapper.py +566 -0
  182. chia/_tests/environments/__init__.py +0 -0
  183. chia/_tests/environments/common.py +35 -0
  184. chia/_tests/environments/full_node.py +47 -0
  185. chia/_tests/environments/wallet.py +429 -0
  186. chia/_tests/ether.py +19 -0
  187. chia/_tests/farmer_harvester/__init__.py +0 -0
  188. chia/_tests/farmer_harvester/config.py +3 -0
  189. chia/_tests/farmer_harvester/test_farmer.py +1264 -0
  190. chia/_tests/farmer_harvester/test_farmer_harvester.py +292 -0
  191. chia/_tests/farmer_harvester/test_filter_prefix_bits.py +131 -0
  192. chia/_tests/farmer_harvester/test_third_party_harvesters.py +528 -0
  193. chia/_tests/farmer_harvester/test_third_party_harvesters_data.json +29 -0
  194. chia/_tests/fee_estimation/__init__.py +0 -0
  195. chia/_tests/fee_estimation/config.py +3 -0
  196. chia/_tests/fee_estimation/test_fee_estimation_integration.py +262 -0
  197. chia/_tests/fee_estimation/test_fee_estimation_rpc.py +287 -0
  198. chia/_tests/fee_estimation/test_fee_estimation_unit_tests.py +144 -0
  199. chia/_tests/fee_estimation/test_mempoolitem_height_added.py +146 -0
  200. chia/_tests/generator/__init__.py +0 -0
  201. chia/_tests/generator/puzzles/__init__.py +0 -0
  202. chia/_tests/generator/puzzles/test_generator_deserialize.clsp +3 -0
  203. chia/_tests/generator/puzzles/test_generator_deserialize.clsp.hex +1 -0
  204. chia/_tests/generator/puzzles/test_multiple_generator_input_arguments.clsp +19 -0
  205. chia/_tests/generator/puzzles/test_multiple_generator_input_arguments.clsp.hex +1 -0
  206. chia/_tests/generator/test_compression.py +201 -0
  207. chia/_tests/generator/test_generator_types.py +44 -0
  208. chia/_tests/generator/test_rom.py +180 -0
  209. chia/_tests/plot_sync/__init__.py +0 -0
  210. chia/_tests/plot_sync/config.py +3 -0
  211. chia/_tests/plot_sync/test_delta.py +101 -0
  212. chia/_tests/plot_sync/test_plot_sync.py +618 -0
  213. chia/_tests/plot_sync/test_receiver.py +451 -0
  214. chia/_tests/plot_sync/test_sender.py +116 -0
  215. chia/_tests/plot_sync/test_sync_simulated.py +451 -0
  216. chia/_tests/plot_sync/util.py +68 -0
  217. chia/_tests/plotting/__init__.py +0 -0
  218. chia/_tests/plotting/config.py +3 -0
  219. chia/_tests/plotting/test_plot_manager.py +781 -0
  220. chia/_tests/plotting/util.py +12 -0
  221. chia/_tests/pools/__init__.py +0 -0
  222. chia/_tests/pools/config.py +5 -0
  223. chia/_tests/pools/test_pool_cli_parsing.py +128 -0
  224. chia/_tests/pools/test_pool_cmdline.py +1001 -0
  225. chia/_tests/pools/test_pool_config.py +42 -0
  226. chia/_tests/pools/test_pool_puzzles_lifecycle.py +397 -0
  227. chia/_tests/pools/test_pool_rpc.py +1123 -0
  228. chia/_tests/pools/test_pool_wallet.py +205 -0
  229. chia/_tests/pools/test_wallet_pool_store.py +161 -0
  230. chia/_tests/process_junit.py +348 -0
  231. chia/_tests/rpc/__init__.py +0 -0
  232. chia/_tests/rpc/test_rpc_client.py +138 -0
  233. chia/_tests/rpc/test_rpc_server.py +183 -0
  234. chia/_tests/simulation/__init__.py +0 -0
  235. chia/_tests/simulation/config.py +6 -0
  236. chia/_tests/simulation/test_simulation.py +501 -0
  237. chia/_tests/simulation/test_simulator.py +232 -0
  238. chia/_tests/simulation/test_start_simulator.py +107 -0
  239. chia/_tests/testconfig.py +13 -0
  240. chia/_tests/timelord/__init__.py +0 -0
  241. chia/_tests/timelord/config.py +3 -0
  242. chia/_tests/timelord/test_new_peak.py +437 -0
  243. chia/_tests/timelord/test_timelord.py +11 -0
  244. chia/_tests/tools/1315537.json +170 -0
  245. chia/_tests/tools/1315544.json +160 -0
  246. chia/_tests/tools/1315630.json +150 -0
  247. chia/_tests/tools/300000.json +105 -0
  248. chia/_tests/tools/442734.json +140 -0
  249. chia/_tests/tools/466212.json +130 -0
  250. chia/_tests/tools/__init__.py +0 -0
  251. chia/_tests/tools/config.py +5 -0
  252. chia/_tests/tools/test-blockchain-db.sqlite +0 -0
  253. chia/_tests/tools/test_full_sync.py +30 -0
  254. chia/_tests/tools/test_legacy_keyring.py +82 -0
  255. chia/_tests/tools/test_run_block.py +128 -0
  256. chia/_tests/tools/test_virtual_project.py +591 -0
  257. chia/_tests/util/__init__.py +0 -0
  258. chia/_tests/util/benchmark_cost.py +170 -0
  259. chia/_tests/util/benchmarks.py +153 -0
  260. chia/_tests/util/bip39_test_vectors.json +148 -0
  261. chia/_tests/util/blockchain.py +134 -0
  262. chia/_tests/util/blockchain_mock.py +132 -0
  263. chia/_tests/util/build_network_protocol_files.py +302 -0
  264. chia/_tests/util/clvm_generator.bin +0 -0
  265. chia/_tests/util/config.py +3 -0
  266. chia/_tests/util/constants.py +20 -0
  267. chia/_tests/util/db_connection.py +37 -0
  268. chia/_tests/util/full_sync.py +253 -0
  269. chia/_tests/util/gen_ssl_certs.py +114 -0
  270. chia/_tests/util/generator_tools_testing.py +45 -0
  271. chia/_tests/util/get_name_puzzle_conditions.py +52 -0
  272. chia/_tests/util/key_tool.py +36 -0
  273. chia/_tests/util/misc.py +675 -0
  274. chia/_tests/util/network_protocol_data.py +1072 -0
  275. chia/_tests/util/protocol_messages_bytes-v1.0 +0 -0
  276. chia/_tests/util/protocol_messages_json.py +2701 -0
  277. chia/_tests/util/rpc.py +26 -0
  278. chia/_tests/util/run_block.py +163 -0
  279. chia/_tests/util/setup_nodes.py +481 -0
  280. chia/_tests/util/spend_sim.py +492 -0
  281. chia/_tests/util/split_managers.py +102 -0
  282. chia/_tests/util/temp_file.py +14 -0
  283. chia/_tests/util/test_action_scope.py +144 -0
  284. chia/_tests/util/test_async_pool.py +366 -0
  285. chia/_tests/util/test_build_job_matrix.py +42 -0
  286. chia/_tests/util/test_build_network_protocol_files.py +7 -0
  287. chia/_tests/util/test_chia_version.py +50 -0
  288. chia/_tests/util/test_collection.py +11 -0
  289. chia/_tests/util/test_condition_tools.py +229 -0
  290. chia/_tests/util/test_config.py +426 -0
  291. chia/_tests/util/test_dump_keyring.py +60 -0
  292. chia/_tests/util/test_errors.py +10 -0
  293. chia/_tests/util/test_full_block_utils.py +279 -0
  294. chia/_tests/util/test_installed.py +20 -0
  295. chia/_tests/util/test_limited_semaphore.py +53 -0
  296. chia/_tests/util/test_logging_filter.py +42 -0
  297. chia/_tests/util/test_misc.py +445 -0
  298. chia/_tests/util/test_network.py +73 -0
  299. chia/_tests/util/test_network_protocol_files.py +578 -0
  300. chia/_tests/util/test_network_protocol_json.py +267 -0
  301. chia/_tests/util/test_network_protocol_test.py +256 -0
  302. chia/_tests/util/test_paginator.py +71 -0
  303. chia/_tests/util/test_pprint.py +17 -0
  304. chia/_tests/util/test_priority_mutex.py +488 -0
  305. chia/_tests/util/test_recursive_replace.py +116 -0
  306. chia/_tests/util/test_replace_str_to_bytes.py +137 -0
  307. chia/_tests/util/test_service_groups.py +15 -0
  308. chia/_tests/util/test_ssl_check.py +31 -0
  309. chia/_tests/util/test_testnet_overrides.py +19 -0
  310. chia/_tests/util/test_tests_misc.py +38 -0
  311. chia/_tests/util/test_timing.py +37 -0
  312. chia/_tests/util/test_trusted_peer.py +51 -0
  313. chia/_tests/util/time_out_assert.py +191 -0
  314. chia/_tests/wallet/__init__.py +0 -0
  315. chia/_tests/wallet/cat_wallet/__init__.py +0 -0
  316. chia/_tests/wallet/cat_wallet/config.py +4 -0
  317. chia/_tests/wallet/cat_wallet/test_cat_lifecycle.py +468 -0
  318. chia/_tests/wallet/cat_wallet/test_cat_outer_puzzle.py +69 -0
  319. chia/_tests/wallet/cat_wallet/test_cat_wallet.py +1826 -0
  320. chia/_tests/wallet/cat_wallet/test_offer_lifecycle.py +291 -0
  321. chia/_tests/wallet/cat_wallet/test_trades.py +2600 -0
  322. chia/_tests/wallet/clawback/__init__.py +0 -0
  323. chia/_tests/wallet/clawback/config.py +3 -0
  324. chia/_tests/wallet/clawback/test_clawback_decorator.py +78 -0
  325. chia/_tests/wallet/clawback/test_clawback_lifecycle.py +292 -0
  326. chia/_tests/wallet/clawback/test_clawback_metadata.py +50 -0
  327. chia/_tests/wallet/config.py +4 -0
  328. chia/_tests/wallet/conftest.py +278 -0
  329. chia/_tests/wallet/dao_wallet/__init__.py +0 -0
  330. chia/_tests/wallet/dao_wallet/config.py +3 -0
  331. chia/_tests/wallet/dao_wallet/test_dao_clvm.py +1330 -0
  332. chia/_tests/wallet/dao_wallet/test_dao_wallets.py +3488 -0
  333. chia/_tests/wallet/db_wallet/__init__.py +0 -0
  334. chia/_tests/wallet/db_wallet/config.py +3 -0
  335. chia/_tests/wallet/db_wallet/test_db_graftroot.py +141 -0
  336. chia/_tests/wallet/db_wallet/test_dl_offers.py +491 -0
  337. chia/_tests/wallet/db_wallet/test_dl_wallet.py +823 -0
  338. chia/_tests/wallet/did_wallet/__init__.py +0 -0
  339. chia/_tests/wallet/did_wallet/config.py +4 -0
  340. chia/_tests/wallet/did_wallet/test_did.py +2284 -0
  341. chia/_tests/wallet/nft_wallet/__init__.py +0 -0
  342. chia/_tests/wallet/nft_wallet/config.py +4 -0
  343. chia/_tests/wallet/nft_wallet/test_nft_1_offers.py +1493 -0
  344. chia/_tests/wallet/nft_wallet/test_nft_bulk_mint.py +1024 -0
  345. chia/_tests/wallet/nft_wallet/test_nft_lifecycle.py +375 -0
  346. chia/_tests/wallet/nft_wallet/test_nft_offers.py +1209 -0
  347. chia/_tests/wallet/nft_wallet/test_nft_puzzles.py +172 -0
  348. chia/_tests/wallet/nft_wallet/test_nft_wallet.py +2584 -0
  349. chia/_tests/wallet/nft_wallet/test_ownership_outer_puzzle.py +70 -0
  350. chia/_tests/wallet/rpc/__init__.py +0 -0
  351. chia/_tests/wallet/rpc/config.py +4 -0
  352. chia/_tests/wallet/rpc/test_dl_wallet_rpc.py +285 -0
  353. chia/_tests/wallet/rpc/test_wallet_rpc.py +3153 -0
  354. chia/_tests/wallet/simple_sync/__init__.py +0 -0
  355. chia/_tests/wallet/simple_sync/config.py +3 -0
  356. chia/_tests/wallet/simple_sync/test_simple_sync_protocol.py +718 -0
  357. chia/_tests/wallet/sync/__init__.py +0 -0
  358. chia/_tests/wallet/sync/config.py +4 -0
  359. chia/_tests/wallet/sync/test_wallet_sync.py +1692 -0
  360. chia/_tests/wallet/test_address_type.py +189 -0
  361. chia/_tests/wallet/test_bech32m.py +45 -0
  362. chia/_tests/wallet/test_clvm_streamable.py +244 -0
  363. chia/_tests/wallet/test_coin_management.py +354 -0
  364. chia/_tests/wallet/test_coin_selection.py +588 -0
  365. chia/_tests/wallet/test_conditions.py +400 -0
  366. chia/_tests/wallet/test_debug_spend_bundle.py +218 -0
  367. chia/_tests/wallet/test_new_wallet_protocol.py +1174 -0
  368. chia/_tests/wallet/test_nft_store.py +192 -0
  369. chia/_tests/wallet/test_notifications.py +196 -0
  370. chia/_tests/wallet/test_offer_parsing_performance.py +48 -0
  371. chia/_tests/wallet/test_puzzle_store.py +132 -0
  372. chia/_tests/wallet/test_sign_coin_spends.py +159 -0
  373. chia/_tests/wallet/test_signer_protocol.py +947 -0
  374. chia/_tests/wallet/test_singleton.py +122 -0
  375. chia/_tests/wallet/test_singleton_lifecycle_fast.py +772 -0
  376. chia/_tests/wallet/test_singleton_store.py +152 -0
  377. chia/_tests/wallet/test_taproot.py +19 -0
  378. chia/_tests/wallet/test_transaction_store.py +945 -0
  379. chia/_tests/wallet/test_util.py +185 -0
  380. chia/_tests/wallet/test_wallet.py +2139 -0
  381. chia/_tests/wallet/test_wallet_action_scope.py +85 -0
  382. chia/_tests/wallet/test_wallet_blockchain.py +111 -0
  383. chia/_tests/wallet/test_wallet_coin_store.py +1002 -0
  384. chia/_tests/wallet/test_wallet_interested_store.py +43 -0
  385. chia/_tests/wallet/test_wallet_key_val_store.py +40 -0
  386. chia/_tests/wallet/test_wallet_node.py +780 -0
  387. chia/_tests/wallet/test_wallet_retry.py +95 -0
  388. chia/_tests/wallet/test_wallet_state_manager.py +259 -0
  389. chia/_tests/wallet/test_wallet_test_framework.py +275 -0
  390. chia/_tests/wallet/test_wallet_trade_store.py +218 -0
  391. chia/_tests/wallet/test_wallet_user_store.py +34 -0
  392. chia/_tests/wallet/test_wallet_utils.py +156 -0
  393. chia/_tests/wallet/vc_wallet/__init__.py +0 -0
  394. chia/_tests/wallet/vc_wallet/config.py +3 -0
  395. chia/_tests/wallet/vc_wallet/test_cr_outer_puzzle.py +70 -0
  396. chia/_tests/wallet/vc_wallet/test_vc_lifecycle.py +883 -0
  397. chia/_tests/wallet/vc_wallet/test_vc_wallet.py +830 -0
  398. chia/_tests/wallet/wallet_block_tools.py +327 -0
  399. chia/_tests/weight_proof/__init__.py +0 -0
  400. chia/_tests/weight_proof/config.py +3 -0
  401. chia/_tests/weight_proof/test_weight_proof.py +528 -0
  402. chia/apis.py +19 -0
  403. chia/clvm/__init__.py +0 -0
  404. chia/cmds/__init__.py +0 -0
  405. chia/cmds/beta.py +184 -0
  406. chia/cmds/beta_funcs.py +137 -0
  407. chia/cmds/check_wallet_db.py +420 -0
  408. chia/cmds/chia.py +151 -0
  409. chia/cmds/cmd_classes.py +323 -0
  410. chia/cmds/cmd_helpers.py +242 -0
  411. chia/cmds/cmds_util.py +488 -0
  412. chia/cmds/coin_funcs.py +275 -0
  413. chia/cmds/coins.py +182 -0
  414. chia/cmds/completion.py +49 -0
  415. chia/cmds/configure.py +332 -0
  416. chia/cmds/dao.py +1064 -0
  417. chia/cmds/dao_funcs.py +598 -0
  418. chia/cmds/data.py +708 -0
  419. chia/cmds/data_funcs.py +385 -0
  420. chia/cmds/db.py +87 -0
  421. chia/cmds/db_backup_func.py +77 -0
  422. chia/cmds/db_upgrade_func.py +452 -0
  423. chia/cmds/db_validate_func.py +184 -0
  424. chia/cmds/dev.py +18 -0
  425. chia/cmds/farm.py +100 -0
  426. chia/cmds/farm_funcs.py +200 -0
  427. chia/cmds/gh.py +275 -0
  428. chia/cmds/init.py +63 -0
  429. chia/cmds/init_funcs.py +367 -0
  430. chia/cmds/installers.py +131 -0
  431. chia/cmds/keys.py +527 -0
  432. chia/cmds/keys_funcs.py +863 -0
  433. chia/cmds/netspace.py +50 -0
  434. chia/cmds/netspace_funcs.py +54 -0
  435. chia/cmds/options.py +32 -0
  436. chia/cmds/param_types.py +238 -0
  437. chia/cmds/passphrase.py +131 -0
  438. chia/cmds/passphrase_funcs.py +292 -0
  439. chia/cmds/peer.py +51 -0
  440. chia/cmds/peer_funcs.py +129 -0
  441. chia/cmds/plotnft.py +260 -0
  442. chia/cmds/plotnft_funcs.py +405 -0
  443. chia/cmds/plots.py +230 -0
  444. chia/cmds/plotters.py +18 -0
  445. chia/cmds/rpc.py +208 -0
  446. chia/cmds/show.py +72 -0
  447. chia/cmds/show_funcs.py +215 -0
  448. chia/cmds/signer.py +296 -0
  449. chia/cmds/sim.py +225 -0
  450. chia/cmds/sim_funcs.py +509 -0
  451. chia/cmds/start.py +24 -0
  452. chia/cmds/start_funcs.py +109 -0
  453. chia/cmds/stop.py +62 -0
  454. chia/cmds/units.py +9 -0
  455. chia/cmds/wallet.py +1901 -0
  456. chia/cmds/wallet_funcs.py +1874 -0
  457. chia/consensus/__init__.py +0 -0
  458. chia/consensus/block_body_validation.py +562 -0
  459. chia/consensus/block_creation.py +546 -0
  460. chia/consensus/block_header_validation.py +1059 -0
  461. chia/consensus/block_record.py +31 -0
  462. chia/consensus/block_rewards.py +53 -0
  463. chia/consensus/blockchain.py +1087 -0
  464. chia/consensus/blockchain_interface.py +56 -0
  465. chia/consensus/coinbase.py +30 -0
  466. chia/consensus/condition_costs.py +9 -0
  467. chia/consensus/constants.py +49 -0
  468. chia/consensus/cost_calculator.py +15 -0
  469. chia/consensus/default_constants.py +89 -0
  470. chia/consensus/deficit.py +55 -0
  471. chia/consensus/difficulty_adjustment.py +412 -0
  472. chia/consensus/find_fork_point.py +111 -0
  473. chia/consensus/full_block_to_block_record.py +167 -0
  474. chia/consensus/get_block_challenge.py +106 -0
  475. chia/consensus/get_block_generator.py +27 -0
  476. chia/consensus/make_sub_epoch_summary.py +210 -0
  477. chia/consensus/multiprocess_validation.py +268 -0
  478. chia/consensus/pos_quality.py +19 -0
  479. chia/consensus/pot_iterations.py +67 -0
  480. chia/consensus/puzzles/__init__.py +0 -0
  481. chia/consensus/puzzles/chialisp_deserialisation.clsp +69 -0
  482. chia/consensus/puzzles/chialisp_deserialisation.clsp.hex +1 -0
  483. chia/consensus/puzzles/rom_bootstrap_generator.clsp +37 -0
  484. chia/consensus/puzzles/rom_bootstrap_generator.clsp.hex +1 -0
  485. chia/consensus/vdf_info_computation.py +156 -0
  486. chia/daemon/__init__.py +0 -0
  487. chia/daemon/client.py +252 -0
  488. chia/daemon/keychain_proxy.py +502 -0
  489. chia/daemon/keychain_server.py +365 -0
  490. chia/daemon/server.py +1606 -0
  491. chia/daemon/windows_signal.py +56 -0
  492. chia/data_layer/__init__.py +0 -0
  493. chia/data_layer/data_layer.py +1291 -0
  494. chia/data_layer/data_layer_api.py +33 -0
  495. chia/data_layer/data_layer_errors.py +50 -0
  496. chia/data_layer/data_layer_server.py +170 -0
  497. chia/data_layer/data_layer_util.py +985 -0
  498. chia/data_layer/data_layer_wallet.py +1311 -0
  499. chia/data_layer/data_store.py +2267 -0
  500. chia/data_layer/dl_wallet_store.py +407 -0
  501. chia/data_layer/download_data.py +389 -0
  502. chia/data_layer/puzzles/__init__.py +0 -0
  503. chia/data_layer/puzzles/graftroot_dl_offers.clsp +100 -0
  504. chia/data_layer/puzzles/graftroot_dl_offers.clsp.hex +1 -0
  505. chia/data_layer/s3_plugin_config.yml +33 -0
  506. chia/data_layer/s3_plugin_service.py +468 -0
  507. chia/data_layer/util/__init__.py +0 -0
  508. chia/data_layer/util/benchmark.py +107 -0
  509. chia/data_layer/util/plugin.py +40 -0
  510. chia/farmer/__init__.py +0 -0
  511. chia/farmer/farmer.py +923 -0
  512. chia/farmer/farmer_api.py +820 -0
  513. chia/full_node/__init__.py +0 -0
  514. chia/full_node/bitcoin_fee_estimator.py +85 -0
  515. chia/full_node/block_height_map.py +271 -0
  516. chia/full_node/block_store.py +576 -0
  517. chia/full_node/bundle_tools.py +19 -0
  518. chia/full_node/coin_store.py +647 -0
  519. chia/full_node/fee_estimate.py +54 -0
  520. chia/full_node/fee_estimate_store.py +24 -0
  521. chia/full_node/fee_estimation.py +92 -0
  522. chia/full_node/fee_estimator.py +90 -0
  523. chia/full_node/fee_estimator_constants.py +38 -0
  524. chia/full_node/fee_estimator_interface.py +42 -0
  525. chia/full_node/fee_history.py +25 -0
  526. chia/full_node/fee_tracker.py +564 -0
  527. chia/full_node/full_node.py +3327 -0
  528. chia/full_node/full_node_api.py +2025 -0
  529. chia/full_node/full_node_store.py +1033 -0
  530. chia/full_node/hint_management.py +56 -0
  531. chia/full_node/hint_store.py +93 -0
  532. chia/full_node/mempool.py +589 -0
  533. chia/full_node/mempool_check_conditions.py +146 -0
  534. chia/full_node/mempool_manager.py +853 -0
  535. chia/full_node/pending_tx_cache.py +112 -0
  536. chia/full_node/puzzles/__init__.py +0 -0
  537. chia/full_node/puzzles/block_program_zero.clsp +14 -0
  538. chia/full_node/puzzles/block_program_zero.clsp.hex +1 -0
  539. chia/full_node/puzzles/decompress_coin_spend_entry.clsp +5 -0
  540. chia/full_node/puzzles/decompress_coin_spend_entry.clsp.hex +1 -0
  541. chia/full_node/puzzles/decompress_coin_spend_entry_with_prefix.clsp +7 -0
  542. chia/full_node/puzzles/decompress_coin_spend_entry_with_prefix.clsp.hex +1 -0
  543. chia/full_node/puzzles/decompress_puzzle.clsp +6 -0
  544. chia/full_node/puzzles/decompress_puzzle.clsp.hex +1 -0
  545. chia/full_node/signage_point.py +16 -0
  546. chia/full_node/subscriptions.py +247 -0
  547. chia/full_node/sync_store.py +146 -0
  548. chia/full_node/tx_processing_queue.py +78 -0
  549. chia/full_node/util/__init__.py +0 -0
  550. chia/full_node/weight_proof.py +1720 -0
  551. chia/harvester/__init__.py +0 -0
  552. chia/harvester/harvester.py +272 -0
  553. chia/harvester/harvester_api.py +380 -0
  554. chia/introducer/__init__.py +0 -0
  555. chia/introducer/introducer.py +122 -0
  556. chia/introducer/introducer_api.py +70 -0
  557. chia/legacy/__init__.py +0 -0
  558. chia/legacy/keyring.py +155 -0
  559. chia/plot_sync/__init__.py +0 -0
  560. chia/plot_sync/delta.py +61 -0
  561. chia/plot_sync/exceptions.py +56 -0
  562. chia/plot_sync/receiver.py +386 -0
  563. chia/plot_sync/sender.py +340 -0
  564. chia/plot_sync/util.py +43 -0
  565. chia/plotters/__init__.py +0 -0
  566. chia/plotters/bladebit.py +388 -0
  567. chia/plotters/chiapos.py +63 -0
  568. chia/plotters/madmax.py +224 -0
  569. chia/plotters/plotters.py +577 -0
  570. chia/plotters/plotters_util.py +133 -0
  571. chia/plotting/__init__.py +0 -0
  572. chia/plotting/cache.py +213 -0
  573. chia/plotting/check_plots.py +283 -0
  574. chia/plotting/create_plots.py +278 -0
  575. chia/plotting/manager.py +436 -0
  576. chia/plotting/util.py +336 -0
  577. chia/pools/__init__.py +0 -0
  578. chia/pools/pool_config.py +110 -0
  579. chia/pools/pool_puzzles.py +459 -0
  580. chia/pools/pool_wallet.py +933 -0
  581. chia/pools/pool_wallet_info.py +118 -0
  582. chia/pools/puzzles/__init__.py +0 -0
  583. chia/pools/puzzles/pool_member_innerpuz.clsp +70 -0
  584. chia/pools/puzzles/pool_member_innerpuz.clsp.hex +1 -0
  585. chia/pools/puzzles/pool_waitingroom_innerpuz.clsp +69 -0
  586. chia/pools/puzzles/pool_waitingroom_innerpuz.clsp.hex +1 -0
  587. chia/protocols/__init__.py +0 -0
  588. chia/protocols/farmer_protocol.py +102 -0
  589. chia/protocols/full_node_protocol.py +219 -0
  590. chia/protocols/harvester_protocol.py +216 -0
  591. chia/protocols/introducer_protocol.py +25 -0
  592. chia/protocols/pool_protocol.py +177 -0
  593. chia/protocols/protocol_message_types.py +139 -0
  594. chia/protocols/protocol_state_machine.py +87 -0
  595. chia/protocols/protocol_timing.py +8 -0
  596. chia/protocols/shared_protocol.py +86 -0
  597. chia/protocols/timelord_protocol.py +93 -0
  598. chia/protocols/wallet_protocol.py +401 -0
  599. chia/py.typed +0 -0
  600. chia/rpc/__init__.py +0 -0
  601. chia/rpc/crawler_rpc_api.py +80 -0
  602. chia/rpc/data_layer_rpc_api.py +644 -0
  603. chia/rpc/data_layer_rpc_client.py +188 -0
  604. chia/rpc/data_layer_rpc_util.py +58 -0
  605. chia/rpc/farmer_rpc_api.py +365 -0
  606. chia/rpc/farmer_rpc_client.py +86 -0
  607. chia/rpc/full_node_rpc_api.py +959 -0
  608. chia/rpc/full_node_rpc_client.py +292 -0
  609. chia/rpc/harvester_rpc_api.py +141 -0
  610. chia/rpc/harvester_rpc_client.py +54 -0
  611. chia/rpc/rpc_client.py +164 -0
  612. chia/rpc/rpc_server.py +521 -0
  613. chia/rpc/timelord_rpc_api.py +32 -0
  614. chia/rpc/util.py +93 -0
  615. chia/rpc/wallet_request_types.py +904 -0
  616. chia/rpc/wallet_rpc_api.py +4943 -0
  617. chia/rpc/wallet_rpc_client.py +1814 -0
  618. chia/seeder/__init__.py +0 -0
  619. chia/seeder/crawl_store.py +425 -0
  620. chia/seeder/crawler.py +410 -0
  621. chia/seeder/crawler_api.py +135 -0
  622. chia/seeder/dns_server.py +593 -0
  623. chia/seeder/peer_record.py +146 -0
  624. chia/seeder/start_crawler.py +92 -0
  625. chia/server/__init__.py +0 -0
  626. chia/server/address_manager.py +658 -0
  627. chia/server/address_manager_store.py +237 -0
  628. chia/server/api_protocol.py +116 -0
  629. chia/server/capabilities.py +24 -0
  630. chia/server/chia_policy.py +346 -0
  631. chia/server/introducer_peers.py +76 -0
  632. chia/server/node_discovery.py +714 -0
  633. chia/server/outbound_message.py +33 -0
  634. chia/server/rate_limit_numbers.py +214 -0
  635. chia/server/rate_limits.py +153 -0
  636. chia/server/server.py +741 -0
  637. chia/server/signal_handlers.py +120 -0
  638. chia/server/ssl_context.py +32 -0
  639. chia/server/start_data_layer.py +151 -0
  640. chia/server/start_farmer.py +98 -0
  641. chia/server/start_full_node.py +112 -0
  642. chia/server/start_harvester.py +93 -0
  643. chia/server/start_introducer.py +81 -0
  644. chia/server/start_service.py +316 -0
  645. chia/server/start_timelord.py +89 -0
  646. chia/server/start_wallet.py +113 -0
  647. chia/server/upnp.py +118 -0
  648. chia/server/ws_connection.py +766 -0
  649. chia/simulator/__init__.py +0 -0
  650. chia/simulator/add_blocks_in_batches.py +54 -0
  651. chia/simulator/block_tools.py +2054 -0
  652. chia/simulator/full_node_simulator.py +794 -0
  653. chia/simulator/keyring.py +128 -0
  654. chia/simulator/setup_services.py +506 -0
  655. chia/simulator/simulator_constants.py +13 -0
  656. chia/simulator/simulator_full_node_rpc_api.py +99 -0
  657. chia/simulator/simulator_full_node_rpc_client.py +60 -0
  658. chia/simulator/simulator_protocol.py +29 -0
  659. chia/simulator/simulator_test_tools.py +164 -0
  660. chia/simulator/socket.py +24 -0
  661. chia/simulator/ssl_certs.py +114 -0
  662. chia/simulator/ssl_certs_1.py +697 -0
  663. chia/simulator/ssl_certs_10.py +697 -0
  664. chia/simulator/ssl_certs_2.py +697 -0
  665. chia/simulator/ssl_certs_3.py +697 -0
  666. chia/simulator/ssl_certs_4.py +697 -0
  667. chia/simulator/ssl_certs_5.py +697 -0
  668. chia/simulator/ssl_certs_6.py +697 -0
  669. chia/simulator/ssl_certs_7.py +697 -0
  670. chia/simulator/ssl_certs_8.py +697 -0
  671. chia/simulator/ssl_certs_9.py +697 -0
  672. chia/simulator/start_simulator.py +143 -0
  673. chia/simulator/wallet_tools.py +246 -0
  674. chia/ssl/__init__.py +0 -0
  675. chia/ssl/chia_ca.crt +19 -0
  676. chia/ssl/chia_ca.key +28 -0
  677. chia/ssl/create_ssl.py +249 -0
  678. chia/ssl/dst_root_ca.pem +20 -0
  679. chia/timelord/__init__.py +0 -0
  680. chia/timelord/iters_from_block.py +50 -0
  681. chia/timelord/timelord.py +1226 -0
  682. chia/timelord/timelord_api.py +138 -0
  683. chia/timelord/timelord_launcher.py +190 -0
  684. chia/timelord/timelord_state.py +244 -0
  685. chia/timelord/types.py +22 -0
  686. chia/types/__init__.py +0 -0
  687. chia/types/aliases.py +35 -0
  688. chia/types/block_protocol.py +20 -0
  689. chia/types/blockchain_format/__init__.py +0 -0
  690. chia/types/blockchain_format/classgroup.py +5 -0
  691. chia/types/blockchain_format/coin.py +28 -0
  692. chia/types/blockchain_format/foliage.py +8 -0
  693. chia/types/blockchain_format/pool_target.py +5 -0
  694. chia/types/blockchain_format/program.py +269 -0
  695. chia/types/blockchain_format/proof_of_space.py +135 -0
  696. chia/types/blockchain_format/reward_chain_block.py +6 -0
  697. chia/types/blockchain_format/serialized_program.py +5 -0
  698. chia/types/blockchain_format/sized_bytes.py +11 -0
  699. chia/types/blockchain_format/slots.py +9 -0
  700. chia/types/blockchain_format/sub_epoch_summary.py +5 -0
  701. chia/types/blockchain_format/tree_hash.py +72 -0
  702. chia/types/blockchain_format/vdf.py +86 -0
  703. chia/types/clvm_cost.py +13 -0
  704. chia/types/coin_record.py +43 -0
  705. chia/types/coin_spend.py +115 -0
  706. chia/types/condition_opcodes.py +73 -0
  707. chia/types/condition_with_args.py +16 -0
  708. chia/types/eligible_coin_spends.py +365 -0
  709. chia/types/end_of_slot_bundle.py +5 -0
  710. chia/types/fee_rate.py +38 -0
  711. chia/types/full_block.py +5 -0
  712. chia/types/generator_types.py +13 -0
  713. chia/types/header_block.py +5 -0
  714. chia/types/internal_mempool_item.py +18 -0
  715. chia/types/mempool_inclusion_status.py +9 -0
  716. chia/types/mempool_item.py +85 -0
  717. chia/types/mempool_submission_status.py +30 -0
  718. chia/types/mojos.py +7 -0
  719. chia/types/peer_info.py +64 -0
  720. chia/types/signing_mode.py +29 -0
  721. chia/types/spend_bundle.py +30 -0
  722. chia/types/spend_bundle_conditions.py +7 -0
  723. chia/types/transaction_queue_entry.py +55 -0
  724. chia/types/unfinished_block.py +5 -0
  725. chia/types/unfinished_header_block.py +37 -0
  726. chia/types/validation_state.py +14 -0
  727. chia/types/weight_proof.py +49 -0
  728. chia/util/__init__.py +0 -0
  729. chia/util/action_scope.py +168 -0
  730. chia/util/async_pool.py +226 -0
  731. chia/util/augmented_chain.py +134 -0
  732. chia/util/batches.py +42 -0
  733. chia/util/bech32m.py +126 -0
  734. chia/util/beta_metrics.py +119 -0
  735. chia/util/block_cache.py +56 -0
  736. chia/util/byte_types.py +12 -0
  737. chia/util/check_fork_next_block.py +33 -0
  738. chia/util/chia_logging.py +144 -0
  739. chia/util/chia_version.py +33 -0
  740. chia/util/collection.py +17 -0
  741. chia/util/condition_tools.py +201 -0
  742. chia/util/config.py +367 -0
  743. chia/util/cpu.py +22 -0
  744. chia/util/db_synchronous.py +23 -0
  745. chia/util/db_version.py +32 -0
  746. chia/util/db_wrapper.py +430 -0
  747. chia/util/default_root.py +27 -0
  748. chia/util/dump_keyring.py +93 -0
  749. chia/util/english.txt +2048 -0
  750. chia/util/errors.py +353 -0
  751. chia/util/file_keyring.py +469 -0
  752. chia/util/files.py +97 -0
  753. chia/util/full_block_utils.py +345 -0
  754. chia/util/generator_tools.py +72 -0
  755. chia/util/hash.py +31 -0
  756. chia/util/initial-config.yaml +694 -0
  757. chia/util/inline_executor.py +26 -0
  758. chia/util/ints.py +19 -0
  759. chia/util/ip_address.py +39 -0
  760. chia/util/json_util.py +37 -0
  761. chia/util/keychain.py +676 -0
  762. chia/util/keyring_wrapper.py +327 -0
  763. chia/util/limited_semaphore.py +41 -0
  764. chia/util/lock.py +49 -0
  765. chia/util/log_exceptions.py +32 -0
  766. chia/util/logging.py +36 -0
  767. chia/util/lru_cache.py +31 -0
  768. chia/util/math.py +20 -0
  769. chia/util/network.py +182 -0
  770. chia/util/paginator.py +48 -0
  771. chia/util/path.py +31 -0
  772. chia/util/permissions.py +20 -0
  773. chia/util/prev_transaction_block.py +21 -0
  774. chia/util/priority_mutex.py +95 -0
  775. chia/util/profiler.py +197 -0
  776. chia/util/recursive_replace.py +24 -0
  777. chia/util/safe_cancel_task.py +16 -0
  778. chia/util/service_groups.py +47 -0
  779. chia/util/setproctitle.py +22 -0
  780. chia/util/significant_bits.py +32 -0
  781. chia/util/ssl_check.py +213 -0
  782. chia/util/streamable.py +642 -0
  783. chia/util/task_referencer.py +59 -0
  784. chia/util/task_timing.py +382 -0
  785. chia/util/timing.py +67 -0
  786. chia/util/vdf_prover.py +30 -0
  787. chia/util/virtual_project_analysis.py +540 -0
  788. chia/util/ws_message.py +66 -0
  789. chia/wallet/__init__.py +0 -0
  790. chia/wallet/cat_wallet/__init__.py +0 -0
  791. chia/wallet/cat_wallet/cat_constants.py +75 -0
  792. chia/wallet/cat_wallet/cat_info.py +47 -0
  793. chia/wallet/cat_wallet/cat_outer_puzzle.py +120 -0
  794. chia/wallet/cat_wallet/cat_utils.py +164 -0
  795. chia/wallet/cat_wallet/cat_wallet.py +855 -0
  796. chia/wallet/cat_wallet/dao_cat_info.py +28 -0
  797. chia/wallet/cat_wallet/dao_cat_wallet.py +669 -0
  798. chia/wallet/cat_wallet/lineage_store.py +74 -0
  799. chia/wallet/cat_wallet/puzzles/__init__.py +0 -0
  800. chia/wallet/cat_wallet/puzzles/cat_truths.clib +31 -0
  801. chia/wallet/cat_wallet/puzzles/cat_v2.clsp +397 -0
  802. chia/wallet/cat_wallet/puzzles/cat_v2.clsp.hex +1 -0
  803. chia/wallet/cat_wallet/puzzles/delegated_tail.clsp +25 -0
  804. chia/wallet/cat_wallet/puzzles/delegated_tail.clsp.hex +1 -0
  805. chia/wallet/cat_wallet/puzzles/everything_with_signature.clsp +15 -0
  806. chia/wallet/cat_wallet/puzzles/everything_with_signature.clsp.hex +1 -0
  807. chia/wallet/cat_wallet/puzzles/genesis_by_coin_id.clsp +26 -0
  808. chia/wallet/cat_wallet/puzzles/genesis_by_coin_id.clsp.hex +1 -0
  809. chia/wallet/cat_wallet/puzzles/genesis_by_coin_id_or_singleton.clsp +42 -0
  810. chia/wallet/cat_wallet/puzzles/genesis_by_coin_id_or_singleton.clsp.hex +1 -0
  811. chia/wallet/cat_wallet/puzzles/genesis_by_puzzle_hash.clsp +24 -0
  812. chia/wallet/cat_wallet/puzzles/genesis_by_puzzle_hash.clsp.hex +1 -0
  813. chia/wallet/coin_selection.py +188 -0
  814. chia/wallet/conditions.py +1512 -0
  815. chia/wallet/dao_wallet/__init__.py +0 -0
  816. chia/wallet/dao_wallet/dao_info.py +61 -0
  817. chia/wallet/dao_wallet/dao_utils.py +811 -0
  818. chia/wallet/dao_wallet/dao_wallet.py +2119 -0
  819. chia/wallet/db_wallet/__init__.py +0 -0
  820. chia/wallet/db_wallet/db_wallet_puzzles.py +111 -0
  821. chia/wallet/derivation_record.py +30 -0
  822. chia/wallet/derive_keys.py +146 -0
  823. chia/wallet/did_wallet/__init__.py +0 -0
  824. chia/wallet/did_wallet/did_info.py +39 -0
  825. chia/wallet/did_wallet/did_wallet.py +1494 -0
  826. chia/wallet/did_wallet/did_wallet_puzzles.py +221 -0
  827. chia/wallet/did_wallet/puzzles/__init__.py +0 -0
  828. chia/wallet/did_wallet/puzzles/did_innerpuz.clsp +135 -0
  829. chia/wallet/did_wallet/puzzles/did_innerpuz.clsp.hex +1 -0
  830. chia/wallet/driver_protocol.py +26 -0
  831. chia/wallet/key_val_store.py +55 -0
  832. chia/wallet/lineage_proof.py +58 -0
  833. chia/wallet/nft_wallet/__init__.py +0 -0
  834. chia/wallet/nft_wallet/metadata_outer_puzzle.py +92 -0
  835. chia/wallet/nft_wallet/nft_info.py +120 -0
  836. chia/wallet/nft_wallet/nft_puzzles.py +305 -0
  837. chia/wallet/nft_wallet/nft_wallet.py +1687 -0
  838. chia/wallet/nft_wallet/ownership_outer_puzzle.py +101 -0
  839. chia/wallet/nft_wallet/puzzles/__init__.py +0 -0
  840. chia/wallet/nft_wallet/puzzles/create_nft_launcher_from_did.clsp +6 -0
  841. chia/wallet/nft_wallet/puzzles/create_nft_launcher_from_did.clsp.hex +1 -0
  842. chia/wallet/nft_wallet/puzzles/nft_intermediate_launcher.clsp +6 -0
  843. chia/wallet/nft_wallet/puzzles/nft_intermediate_launcher.clsp.hex +1 -0
  844. chia/wallet/nft_wallet/puzzles/nft_metadata_updater_default.clsp +30 -0
  845. chia/wallet/nft_wallet/puzzles/nft_metadata_updater_default.clsp.hex +1 -0
  846. chia/wallet/nft_wallet/puzzles/nft_metadata_updater_updateable.clsp +28 -0
  847. chia/wallet/nft_wallet/puzzles/nft_metadata_updater_updateable.clsp.hex +1 -0
  848. chia/wallet/nft_wallet/puzzles/nft_ownership_layer.clsp +100 -0
  849. chia/wallet/nft_wallet/puzzles/nft_ownership_layer.clsp.hex +1 -0
  850. chia/wallet/nft_wallet/puzzles/nft_ownership_transfer_program_one_way_claim_with_royalties.clsp +78 -0
  851. chia/wallet/nft_wallet/puzzles/nft_ownership_transfer_program_one_way_claim_with_royalties.clsp.hex +1 -0
  852. chia/wallet/nft_wallet/puzzles/nft_state_layer.clsp +74 -0
  853. chia/wallet/nft_wallet/puzzles/nft_state_layer.clsp.hex +1 -0
  854. chia/wallet/nft_wallet/singleton_outer_puzzle.py +101 -0
  855. chia/wallet/nft_wallet/transfer_program_puzzle.py +82 -0
  856. chia/wallet/nft_wallet/uncurry_nft.py +217 -0
  857. chia/wallet/notification_manager.py +117 -0
  858. chia/wallet/notification_store.py +178 -0
  859. chia/wallet/outer_puzzles.py +84 -0
  860. chia/wallet/payment.py +33 -0
  861. chia/wallet/puzzle_drivers.py +118 -0
  862. chia/wallet/puzzles/__init__.py +0 -0
  863. chia/wallet/puzzles/augmented_condition.clsp +13 -0
  864. chia/wallet/puzzles/augmented_condition.clsp.hex +1 -0
  865. chia/wallet/puzzles/clawback/__init__.py +0 -0
  866. chia/wallet/puzzles/clawback/drivers.py +188 -0
  867. chia/wallet/puzzles/clawback/metadata.py +38 -0
  868. chia/wallet/puzzles/clawback/puzzle_decorator.py +67 -0
  869. chia/wallet/puzzles/condition_codes.clib +77 -0
  870. chia/wallet/puzzles/curry-and-treehash.clib +102 -0
  871. chia/wallet/puzzles/curry.clib +135 -0
  872. chia/wallet/puzzles/curry_by_index.clib +16 -0
  873. chia/wallet/puzzles/dao_cat_eve.clsp +17 -0
  874. chia/wallet/puzzles/dao_cat_eve.clsp.hex +1 -0
  875. chia/wallet/puzzles/dao_cat_launcher.clsp +36 -0
  876. chia/wallet/puzzles/dao_cat_launcher.clsp.hex +1 -0
  877. chia/wallet/puzzles/dao_finished_state.clsp +35 -0
  878. chia/wallet/puzzles/dao_finished_state.clsp.hex +1 -0
  879. chia/wallet/puzzles/dao_finished_state.clsp.hex.sha256tree +1 -0
  880. chia/wallet/puzzles/dao_lockup.clsp +288 -0
  881. chia/wallet/puzzles/dao_lockup.clsp.hex +1 -0
  882. chia/wallet/puzzles/dao_lockup.clsp.hex.sha256tree +1 -0
  883. chia/wallet/puzzles/dao_proposal.clsp +377 -0
  884. chia/wallet/puzzles/dao_proposal.clsp.hex +1 -0
  885. chia/wallet/puzzles/dao_proposal.clsp.hex.sha256tree +1 -0
  886. chia/wallet/puzzles/dao_proposal_timer.clsp +78 -0
  887. chia/wallet/puzzles/dao_proposal_timer.clsp.hex +1 -0
  888. chia/wallet/puzzles/dao_proposal_timer.clsp.hex.sha256tree +1 -0
  889. chia/wallet/puzzles/dao_proposal_validator.clsp +87 -0
  890. chia/wallet/puzzles/dao_proposal_validator.clsp.hex +1 -0
  891. chia/wallet/puzzles/dao_proposal_validator.clsp.hex.sha256tree +1 -0
  892. chia/wallet/puzzles/dao_spend_p2_singleton_v2.clsp +240 -0
  893. chia/wallet/puzzles/dao_spend_p2_singleton_v2.clsp.hex +1 -0
  894. chia/wallet/puzzles/dao_spend_p2_singleton_v2.clsp.hex.sha256tree +1 -0
  895. chia/wallet/puzzles/dao_treasury.clsp +115 -0
  896. chia/wallet/puzzles/dao_treasury.clsp.hex +1 -0
  897. chia/wallet/puzzles/dao_update_proposal.clsp +44 -0
  898. chia/wallet/puzzles/dao_update_proposal.clsp.hex +1 -0
  899. chia/wallet/puzzles/deployed_puzzle_hashes.json +67 -0
  900. chia/wallet/puzzles/json.clib +25 -0
  901. chia/wallet/puzzles/load_clvm.py +161 -0
  902. chia/wallet/puzzles/merkle_utils.clib +18 -0
  903. chia/wallet/puzzles/notification.clsp +7 -0
  904. chia/wallet/puzzles/notification.clsp.hex +1 -0
  905. chia/wallet/puzzles/p2_1_of_n.clsp +22 -0
  906. chia/wallet/puzzles/p2_1_of_n.clsp.hex +1 -0
  907. chia/wallet/puzzles/p2_conditions.clsp +3 -0
  908. chia/wallet/puzzles/p2_conditions.clsp.hex +1 -0
  909. chia/wallet/puzzles/p2_conditions.py +26 -0
  910. chia/wallet/puzzles/p2_delegated_conditions.clsp +18 -0
  911. chia/wallet/puzzles/p2_delegated_conditions.clsp.hex +1 -0
  912. chia/wallet/puzzles/p2_delegated_conditions.py +21 -0
  913. chia/wallet/puzzles/p2_delegated_puzzle.clsp +19 -0
  914. chia/wallet/puzzles/p2_delegated_puzzle.clsp.hex +1 -0
  915. chia/wallet/puzzles/p2_delegated_puzzle.py +34 -0
  916. chia/wallet/puzzles/p2_delegated_puzzle_or_hidden_puzzle.clsp +91 -0
  917. chia/wallet/puzzles/p2_delegated_puzzle_or_hidden_puzzle.clsp.hex +1 -0
  918. chia/wallet/puzzles/p2_delegated_puzzle_or_hidden_puzzle.py +160 -0
  919. chia/wallet/puzzles/p2_m_of_n_delegate_direct.clsp +108 -0
  920. chia/wallet/puzzles/p2_m_of_n_delegate_direct.clsp.hex +1 -0
  921. chia/wallet/puzzles/p2_m_of_n_delegate_direct.py +21 -0
  922. chia/wallet/puzzles/p2_parent.clsp +19 -0
  923. chia/wallet/puzzles/p2_parent.clsp.hex +1 -0
  924. chia/wallet/puzzles/p2_puzzle_hash.clsp +18 -0
  925. chia/wallet/puzzles/p2_puzzle_hash.clsp.hex +1 -0
  926. chia/wallet/puzzles/p2_puzzle_hash.py +27 -0
  927. chia/wallet/puzzles/p2_singleton.clsp +30 -0
  928. chia/wallet/puzzles/p2_singleton.clsp.hex +1 -0
  929. chia/wallet/puzzles/p2_singleton_aggregator.clsp +81 -0
  930. chia/wallet/puzzles/p2_singleton_aggregator.clsp.hex +1 -0
  931. chia/wallet/puzzles/p2_singleton_or_delayed_puzhash.clsp +50 -0
  932. chia/wallet/puzzles/p2_singleton_or_delayed_puzhash.clsp.hex +1 -0
  933. chia/wallet/puzzles/p2_singleton_via_delegated_puzzle.clsp +47 -0
  934. chia/wallet/puzzles/p2_singleton_via_delegated_puzzle.clsp.hex +1 -0
  935. chia/wallet/puzzles/puzzle_utils.py +34 -0
  936. chia/wallet/puzzles/settlement_payments.clsp +49 -0
  937. chia/wallet/puzzles/settlement_payments.clsp.hex +1 -0
  938. chia/wallet/puzzles/sha256tree.clib +11 -0
  939. chia/wallet/puzzles/singleton_launcher.clsp +16 -0
  940. chia/wallet/puzzles/singleton_launcher.clsp.hex +1 -0
  941. chia/wallet/puzzles/singleton_top_layer.clsp +177 -0
  942. chia/wallet/puzzles/singleton_top_layer.clsp.hex +1 -0
  943. chia/wallet/puzzles/singleton_top_layer.py +296 -0
  944. chia/wallet/puzzles/singleton_top_layer_v1_1.clsp +107 -0
  945. chia/wallet/puzzles/singleton_top_layer_v1_1.clsp.hex +1 -0
  946. chia/wallet/puzzles/singleton_top_layer_v1_1.py +345 -0
  947. chia/wallet/puzzles/singleton_truths.clib +21 -0
  948. chia/wallet/puzzles/tails.py +348 -0
  949. chia/wallet/puzzles/utility_macros.clib +48 -0
  950. chia/wallet/signer_protocol.py +125 -0
  951. chia/wallet/singleton.py +106 -0
  952. chia/wallet/singleton_record.py +30 -0
  953. chia/wallet/trade_manager.py +1102 -0
  954. chia/wallet/trade_record.py +67 -0
  955. chia/wallet/trading/__init__.py +0 -0
  956. chia/wallet/trading/offer.py +702 -0
  957. chia/wallet/trading/trade_status.py +13 -0
  958. chia/wallet/trading/trade_store.py +526 -0
  959. chia/wallet/transaction_record.py +158 -0
  960. chia/wallet/transaction_sorting.py +14 -0
  961. chia/wallet/uncurried_puzzle.py +17 -0
  962. chia/wallet/util/__init__.py +0 -0
  963. chia/wallet/util/address_type.py +55 -0
  964. chia/wallet/util/blind_signer_tl.py +164 -0
  965. chia/wallet/util/clvm_streamable.py +203 -0
  966. chia/wallet/util/compute_hints.py +66 -0
  967. chia/wallet/util/compute_memos.py +43 -0
  968. chia/wallet/util/curry_and_treehash.py +91 -0
  969. chia/wallet/util/debug_spend_bundle.py +232 -0
  970. chia/wallet/util/merkle_tree.py +100 -0
  971. chia/wallet/util/merkle_utils.py +102 -0
  972. chia/wallet/util/new_peak_queue.py +82 -0
  973. chia/wallet/util/notifications.py +12 -0
  974. chia/wallet/util/peer_request_cache.py +174 -0
  975. chia/wallet/util/pprint.py +39 -0
  976. chia/wallet/util/puzzle_compression.py +95 -0
  977. chia/wallet/util/puzzle_decorator.py +100 -0
  978. chia/wallet/util/puzzle_decorator_type.py +7 -0
  979. chia/wallet/util/query_filter.py +59 -0
  980. chia/wallet/util/transaction_type.py +23 -0
  981. chia/wallet/util/tx_config.py +158 -0
  982. chia/wallet/util/wallet_sync_utils.py +351 -0
  983. chia/wallet/util/wallet_types.py +72 -0
  984. chia/wallet/vc_wallet/__init__.py +0 -0
  985. chia/wallet/vc_wallet/cr_cat_drivers.py +664 -0
  986. chia/wallet/vc_wallet/cr_cat_wallet.py +877 -0
  987. chia/wallet/vc_wallet/cr_outer_puzzle.py +102 -0
  988. chia/wallet/vc_wallet/cr_puzzles/__init__.py +0 -0
  989. chia/wallet/vc_wallet/cr_puzzles/conditions_w_fee_announce.clsp +3 -0
  990. chia/wallet/vc_wallet/cr_puzzles/conditions_w_fee_announce.clsp.hex +1 -0
  991. chia/wallet/vc_wallet/cr_puzzles/credential_restriction.clsp +304 -0
  992. chia/wallet/vc_wallet/cr_puzzles/credential_restriction.clsp.hex +1 -0
  993. chia/wallet/vc_wallet/cr_puzzles/flag_proofs_checker.clsp +45 -0
  994. chia/wallet/vc_wallet/cr_puzzles/flag_proofs_checker.clsp.hex +1 -0
  995. chia/wallet/vc_wallet/vc_drivers.py +838 -0
  996. chia/wallet/vc_wallet/vc_puzzles/__init__.py +0 -0
  997. chia/wallet/vc_wallet/vc_puzzles/covenant_layer.clsp +30 -0
  998. chia/wallet/vc_wallet/vc_puzzles/covenant_layer.clsp.hex +1 -0
  999. chia/wallet/vc_wallet/vc_puzzles/eml_covenant_morpher.clsp +75 -0
  1000. chia/wallet/vc_wallet/vc_puzzles/eml_covenant_morpher.clsp.hex +1 -0
  1001. chia/wallet/vc_wallet/vc_puzzles/eml_transfer_program_covenant_adapter.clsp +32 -0
  1002. chia/wallet/vc_wallet/vc_puzzles/eml_transfer_program_covenant_adapter.clsp.hex +1 -0
  1003. chia/wallet/vc_wallet/vc_puzzles/eml_update_metadata_with_DID.clsp +80 -0
  1004. chia/wallet/vc_wallet/vc_puzzles/eml_update_metadata_with_DID.clsp.hex +1 -0
  1005. chia/wallet/vc_wallet/vc_puzzles/exigent_metadata_layer.clsp +163 -0
  1006. chia/wallet/vc_wallet/vc_puzzles/exigent_metadata_layer.clsp.hex +1 -0
  1007. chia/wallet/vc_wallet/vc_puzzles/p2_announced_delegated_puzzle.clsp +16 -0
  1008. chia/wallet/vc_wallet/vc_puzzles/p2_announced_delegated_puzzle.clsp.hex +1 -0
  1009. chia/wallet/vc_wallet/vc_puzzles/standard_vc_backdoor_puzzle.clsp +74 -0
  1010. chia/wallet/vc_wallet/vc_puzzles/standard_vc_backdoor_puzzle.clsp.hex +1 -0
  1011. chia/wallet/vc_wallet/vc_puzzles/std_parent_morpher.clsp +23 -0
  1012. chia/wallet/vc_wallet/vc_puzzles/std_parent_morpher.clsp.hex +1 -0
  1013. chia/wallet/vc_wallet/vc_puzzles/viral_backdoor.clsp +64 -0
  1014. chia/wallet/vc_wallet/vc_puzzles/viral_backdoor.clsp.hex +1 -0
  1015. chia/wallet/vc_wallet/vc_store.py +263 -0
  1016. chia/wallet/vc_wallet/vc_wallet.py +638 -0
  1017. chia/wallet/wallet.py +698 -0
  1018. chia/wallet/wallet_action_scope.py +96 -0
  1019. chia/wallet/wallet_blockchain.py +244 -0
  1020. chia/wallet/wallet_coin_record.py +72 -0
  1021. chia/wallet/wallet_coin_store.py +351 -0
  1022. chia/wallet/wallet_info.py +35 -0
  1023. chia/wallet/wallet_interested_store.py +188 -0
  1024. chia/wallet/wallet_nft_store.py +279 -0
  1025. chia/wallet/wallet_node.py +1765 -0
  1026. chia/wallet/wallet_node_api.py +207 -0
  1027. chia/wallet/wallet_pool_store.py +119 -0
  1028. chia/wallet/wallet_protocol.py +90 -0
  1029. chia/wallet/wallet_puzzle_store.py +396 -0
  1030. chia/wallet/wallet_retry_store.py +70 -0
  1031. chia/wallet/wallet_singleton_store.py +259 -0
  1032. chia/wallet/wallet_spend_bundle.py +25 -0
  1033. chia/wallet/wallet_state_manager.py +2819 -0
  1034. chia/wallet/wallet_transaction_store.py +496 -0
  1035. chia/wallet/wallet_user_store.py +110 -0
  1036. chia/wallet/wallet_weight_proof_handler.py +126 -0
  1037. chia_blockchain-2.5.1rc1.dist-info/LICENSE +201 -0
  1038. chia_blockchain-2.5.1rc1.dist-info/METADATA +156 -0
  1039. chia_blockchain-2.5.1rc1.dist-info/RECORD +1042 -0
  1040. chia_blockchain-2.5.1rc1.dist-info/WHEEL +4 -0
  1041. chia_blockchain-2.5.1rc1.dist-info/entry_points.txt +17 -0
  1042. mozilla-ca/cacert.pem +3611 -0
@@ -0,0 +1,1264 @@
1
+ from __future__ import annotations
2
+
3
+ import json
4
+ import logging
5
+ from dataclasses import dataclass
6
+ from time import time
7
+ from types import TracebackType
8
+ from typing import Any, Optional, Union, cast
9
+ from unittest.mock import ANY
10
+
11
+ import pytest
12
+ from chia_rs import AugSchemeMPL, G1Element, G2Element, PrivateKey
13
+ from pytest_mock import MockerFixture
14
+ from yarl import URL
15
+
16
+ from chia import __version__
17
+ from chia._tests.conftest import HarvesterFarmerEnvironment
18
+ from chia._tests.util.misc import DataCase, Marks, datacases
19
+ from chia.consensus.default_constants import DEFAULT_CONSTANTS
20
+ from chia.farmer.farmer import UPDATE_POOL_FARMER_INFO_INTERVAL, Farmer, increment_pool_stats, strip_old_entries
21
+ from chia.pools.pool_config import PoolWalletConfig
22
+ from chia.protocols import farmer_protocol, harvester_protocol
23
+ from chia.protocols.harvester_protocol import NewProofOfSpace, RespondSignatures
24
+ from chia.protocols.pool_protocol import PoolErrorCode
25
+ from chia.server.ws_connection import WSChiaConnection
26
+ from chia.simulator.block_tools import BlockTools
27
+ from chia.types.aliases import FarmerService, HarvesterService
28
+ from chia.types.blockchain_format.proof_of_space import (
29
+ ProofOfSpace,
30
+ generate_plot_public_key,
31
+ verify_and_get_quality_string,
32
+ )
33
+ from chia.types.blockchain_format.sized_bytes import bytes32
34
+ from chia.util.config import load_config, save_config
35
+ from chia.util.hash import std_hash
36
+ from chia.util.ints import uint8, uint16, uint32, uint64
37
+
38
+ log = logging.getLogger(__name__)
39
+
40
+
41
+ class StripOldEntriesCase:
42
+ pairs: list[tuple[float, int]]
43
+ before: float
44
+ expected_result: list[tuple[float, int]]
45
+
46
+ def __init__(self, pairs: list[tuple[float, int]], before: float, expected_result: list[tuple[float, int]]):
47
+ self.pairs = pairs
48
+ self.before = before
49
+ self.expected_result = expected_result
50
+
51
+
52
+ class IncrementPoolStatsCase:
53
+ pool_states: dict[bytes32, Any]
54
+ p2_singleton_puzzle_hash: bytes32
55
+ name: str
56
+ current_time: float
57
+ count: int
58
+ value: Optional[Union[int, dict[str, Any]]]
59
+ expected_result: Any
60
+
61
+ def __init__(
62
+ self,
63
+ p2_singleton_puzzle_hash: bytes32,
64
+ name: str,
65
+ current_time: float,
66
+ count: int,
67
+ value: Optional[Union[int, dict[str, Any]]],
68
+ expected_result: Any,
69
+ ):
70
+ prepared_p2_singleton_puzzle_hash = std_hash(b"11223344")
71
+ self.pool_states = {
72
+ prepared_p2_singleton_puzzle_hash: {
73
+ "p2_singleton_puzzle_hash": prepared_p2_singleton_puzzle_hash.hex(),
74
+ "xxx_since_start": 1,
75
+ "xxx_24h": [(1689491043, 1)],
76
+ "current_difficulty": 1,
77
+ }
78
+ }
79
+ self.p2_singleton_puzzle_hash = p2_singleton_puzzle_hash
80
+ self.name = name
81
+ self.current_time = current_time
82
+ self.count = count
83
+ self.value = value
84
+ self.expected_result = expected_result
85
+
86
+
87
+ class DummyHarvesterPeer:
88
+ return_invalid_response: bool
89
+ peer_node_id: bytes32 = std_hash(b"1")
90
+ version: str
91
+
92
+ def __init__(self, return_invalid_response: bool = False, version: str = "1.0.0"):
93
+ self.return_invalid_response = return_invalid_response
94
+ self.version = version
95
+
96
+ async def send_message(self, arg1: Any) -> None:
97
+ pass
98
+
99
+ async def call_api(self, arg1: Any, request: harvester_protocol.RequestSignatures) -> Any:
100
+ if self.return_invalid_response:
101
+ return 0
102
+
103
+ local_sk: PrivateKey = PrivateKey.from_bytes(
104
+ bytes.fromhex("185c61579e152bbd3b2face3245951e7c67f67f8d2fb2aa4a56e178186c1e5fa")
105
+ )
106
+ farmer_public_key = G1Element.from_bytes(
107
+ bytes.fromhex(
108
+ "af59ca047f2f34a4db6e4fa313764f6e71d4c53bc316517e0fa486b8e3d3f36f112ee078e3226854b97791452e3a9fd4"
109
+ )
110
+ )
111
+ agg_pk = generate_plot_public_key(local_sk.get_g1(), farmer_public_key, True)
112
+ message = request.messages[0]
113
+ signature: G2Element = AugSchemeMPL.sign(local_sk, message, agg_pk)
114
+ return RespondSignatures(
115
+ plot_identifier=request.plot_identifier,
116
+ challenge_hash=request.challenge_hash,
117
+ sp_hash=request.sp_hash,
118
+ local_pk=local_sk.get_g1(),
119
+ farmer_pk=farmer_public_key,
120
+ message_signatures=[(message, signature)],
121
+ include_source_signature_data=False,
122
+ farmer_reward_address_override=None,
123
+ )
124
+
125
+
126
+ @dataclass
127
+ class NewProofOfSpaceCase:
128
+ difficulty: uint64
129
+ sub_slot_iters: uint64
130
+ challenge_hash: bytes32
131
+ sp_hash: bytes32
132
+ plot_identifier: str
133
+ signage_point_index: uint8
134
+ plot_id: bytes32
135
+ plot_size: uint8
136
+ plot_challenge: bytes32
137
+ plot_public_key: G1Element
138
+ pool_public_key: Optional[G1Element]
139
+ pool_contract_puzzle_hash: bytes32
140
+ height: uint32
141
+ proof: bytes
142
+ pool_config: PoolWalletConfig
143
+ pool_difficulty: Optional[uint64]
144
+ authentication_token_timeout: Optional[uint8]
145
+ farmer_private_keys: list[PrivateKey]
146
+ authentication_keys: dict[bytes32, PrivateKey]
147
+ use_invalid_peer_response: bool
148
+ expected_pool_state: dict[str, Any]
149
+ marks: Marks = ()
150
+
151
+ # This creates a test case whose proof of space passes plot filter and quality check
152
+ @classmethod
153
+ def create_verified_quality_case(
154
+ cls,
155
+ difficulty: uint64,
156
+ sub_slot_iters: uint64,
157
+ pool_url: str,
158
+ pool_difficulty: Optional[uint64],
159
+ authentication_token_timeout: Optional[uint8],
160
+ use_invalid_peer_response: bool,
161
+ has_valid_authentication_keys: bool,
162
+ expected_pool_stats: dict[str, Any],
163
+ ) -> NewProofOfSpaceCase:
164
+ p2_singleton_puzzle_hash = bytes32.fromhex("302e05a1e6af431c22043ae2a9a8f71148c955c372697cb8ab348160976283df")
165
+ pool_config = PoolWalletConfig(
166
+ launcher_id=bytes32.fromhex("ae4ef3b9bfe68949691281a015a9c16630fc8f66d48c19ca548fb80768791afa"),
167
+ pool_url=pool_url,
168
+ payout_instructions="c2b08e41d766da4116e388357ed957d04ad754623a915f3fd65188a8746cf3e8",
169
+ target_puzzle_hash=bytes32.fromhex("344587cf06a39db471d2cc027504e8688a0a67cce961253500c956c73603fd58"),
170
+ p2_singleton_puzzle_hash=p2_singleton_puzzle_hash,
171
+ owner_public_key=G1Element.from_bytes(
172
+ bytes.fromhex(
173
+ "8348455278ecec68325b6754b1f3218cde1511ca9393197e7876d7ae04af1c4dd86b0c50601cf5daeb034e8f7c226537"
174
+ )
175
+ ),
176
+ )
177
+
178
+ expected_pool_state = expected_pool_stats.copy()
179
+ expected_pool_state["p2_singleton_puzzle_hash"] = p2_singleton_puzzle_hash.hex()
180
+ expected_pool_state["pool_config"] = pool_config
181
+
182
+ return NewProofOfSpaceCase(
183
+ difficulty=difficulty,
184
+ sub_slot_iters=sub_slot_iters,
185
+ challenge_hash=bytes32.fromhex("9dcc155cafa8723627d01d00118767269a15f5f0d496dc9c2940a45b63c0c00c"),
186
+ sp_hash=bytes32.fromhex("bca3ac17a0a8fc072188b3738455a3f006528d181c39e31c88e8755492575272"),
187
+ plot_identifier="test",
188
+ signage_point_index=uint8(1),
189
+ plot_id=bytes32.fromhex("baaa6780c53d4b3739b8807b4ae79a76644ddf0d9e03dc7d0a6a0e613e764d9f"),
190
+ plot_size=uint8(32),
191
+ plot_challenge=bytes32.fromhex("7580e4c366dc2c94c37ce44943f9629a3cd6e027d7b24cd014adeaa578d4b0a2"),
192
+ plot_public_key=G1Element.from_bytes(
193
+ bytes.fromhex(
194
+ "a6126295fbf0f50dbed8dc41e236241413fdc8a97e650e3e"
195
+ "d69d66d0921d3236f8961cc1cf8c1b195521c2d9143048e2"
196
+ )
197
+ ),
198
+ pool_public_key=None,
199
+ pool_contract_puzzle_hash=p2_singleton_puzzle_hash,
200
+ height=uint32(1),
201
+ proof=bytes.fromhex(
202
+ "2aa93f8c112c274d3c707d94d0a0e07b6710d96fca284baa198ca4632ee1f591"
203
+ "edad47d97e854aec76dc08519614a0d17253fc24fbaab66f558b2f6afd2210ac"
204
+ "ac09b48e01fd0b8334eae94d55db5df28c80c03a586ce8afe3b986e6aceef4d4"
205
+ "93a7c0b1bf30baaafedc1ef9af6a1eee911e3873e9229a0bf18cfc95bcfc3dac"
206
+ "36522c84a85fd16a3ff501b44024e371bbf7a3ee9d4ded6e05c82e95083ea5ea"
207
+ "d2d64e4a46e0550e98fc56b30760b4bcf439ac0ee675e157fcfecd4b294c6d14"
208
+ "987883e3659777c68d6a6962c770a66817f4dc641b5823b077093df10da031c1"
209
+ "4b32005bf38eb53944bf3bc5a1d19c21e3b96759b7f557b4668687b7a8a36344"
210
+ ),
211
+ pool_config=pool_config,
212
+ pool_difficulty=pool_difficulty,
213
+ authentication_token_timeout=authentication_token_timeout,
214
+ farmer_private_keys=[
215
+ PrivateKey.from_bytes(bytes.fromhex("3cb188d8b2469bb8414a3ec68857959e231a8bec836df199a896ff523b9d2f7d"))
216
+ ],
217
+ authentication_keys=(
218
+ {
219
+ p2_singleton_puzzle_hash: PrivateKey.from_bytes(
220
+ bytes.fromhex("11ed596eb95b31364a9185e948f6b66be30415f816819449d5d40751dc70e786")
221
+ ),
222
+ }
223
+ if has_valid_authentication_keys
224
+ else {}
225
+ ),
226
+ use_invalid_peer_response=use_invalid_peer_response,
227
+ expected_pool_state=expected_pool_state,
228
+ )
229
+
230
+
231
+ @pytest.mark.parametrize(
232
+ argnames="case",
233
+ argvalues=[
234
+ pytest.param(StripOldEntriesCase([], 0, []), id="no_params"),
235
+ pytest.param(StripOldEntriesCase([(1689491043.3493967, 1)], 1689491044, []), id="stripped"),
236
+ pytest.param(
237
+ StripOldEntriesCase([(1689491043.3493967, 1)], 1689491043, [(1689491043.3493967, 1)]),
238
+ id="not_stripped",
239
+ ),
240
+ ],
241
+ )
242
+ def test_strip_old_entries(case: StripOldEntriesCase) -> None:
243
+ assert strip_old_entries(case.pairs, case.before) == case.expected_result
244
+
245
+
246
+ @pytest.mark.parametrize(
247
+ argnames="case",
248
+ argvalues=[
249
+ pytest.param(
250
+ IncrementPoolStatsCase(
251
+ std_hash(b"0"),
252
+ "xxx",
253
+ 1689491043.3493967,
254
+ 1,
255
+ 1,
256
+ None,
257
+ ),
258
+ id="p2_singleton_puzzle_hash_not_exist",
259
+ ),
260
+ pytest.param(
261
+ IncrementPoolStatsCase(
262
+ std_hash(b"11223344"),
263
+ "yyy",
264
+ 1689491043.3493967,
265
+ 1,
266
+ 1,
267
+ {
268
+ "p2_singleton_puzzle_hash": std_hash(b"11223344").hex(),
269
+ "xxx_since_start": 1,
270
+ "xxx_24h": [(1689491043, 1)],
271
+ "current_difficulty": 1,
272
+ },
273
+ ),
274
+ id="p2_singleton_puzzle_hash_exist_but_no_change",
275
+ ),
276
+ pytest.param(
277
+ IncrementPoolStatsCase(
278
+ std_hash(b"11223344"),
279
+ "xxx",
280
+ 1689491044.1,
281
+ 1,
282
+ 2,
283
+ {
284
+ "p2_singleton_puzzle_hash": std_hash(b"11223344").hex(),
285
+ "xxx_since_start": 2,
286
+ "xxx_24h": [(1689491043, 1), (1689491044, 2)],
287
+ "current_difficulty": 1,
288
+ },
289
+ ),
290
+ id="value_is_set",
291
+ ),
292
+ pytest.param(
293
+ IncrementPoolStatsCase(
294
+ std_hash(b"11223344"),
295
+ "xxx",
296
+ 1689491044.1,
297
+ 1,
298
+ None,
299
+ {
300
+ "p2_singleton_puzzle_hash": std_hash(b"11223344").hex(),
301
+ "xxx_since_start": 2,
302
+ "xxx_24h": [(1689491043, 1), (1689491044, 1)],
303
+ "current_difficulty": 1,
304
+ },
305
+ ),
306
+ id="value_is_not_set",
307
+ ),
308
+ pytest.param(
309
+ IncrementPoolStatsCase(
310
+ std_hash(b"11223344"),
311
+ "xxx",
312
+ 1689577444,
313
+ 1,
314
+ None,
315
+ {
316
+ "p2_singleton_puzzle_hash": std_hash(b"11223344").hex(),
317
+ "xxx_since_start": 2,
318
+ "xxx_24h": [(1689577444, 1)],
319
+ "current_difficulty": 1,
320
+ },
321
+ ),
322
+ id="stripped",
323
+ ),
324
+ ],
325
+ )
326
+ def test_increment_pool_stats(case: IncrementPoolStatsCase) -> None:
327
+ increment_pool_stats(
328
+ case.pool_states, case.p2_singleton_puzzle_hash, case.name, case.current_time, case.count, case.value
329
+ )
330
+ if case.expected_result is None:
331
+ assert case.p2_singleton_puzzle_hash not in case.pool_states
332
+ else:
333
+ assert case.pool_states[case.p2_singleton_puzzle_hash] == case.expected_result
334
+
335
+
336
+ @pytest.mark.parametrize(
337
+ argnames="case",
338
+ argvalues=[
339
+ pytest.param(
340
+ NewProofOfSpaceCase.create_verified_quality_case(
341
+ difficulty=uint64(1),
342
+ sub_slot_iters=uint64(1000000000000),
343
+ pool_url="",
344
+ pool_difficulty=uint64(1),
345
+ authentication_token_timeout=uint8(10),
346
+ use_invalid_peer_response=False,
347
+ has_valid_authentication_keys=True,
348
+ expected_pool_stats={
349
+ "points_found_since_start": 0,
350
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
351
+ # so every `xxx_24h` item in this dict will be list[Any].
352
+ "points_found_24h": [],
353
+ "points_acknowledged_since_start": 0,
354
+ "points_acknowledged_24h": [],
355
+ "pool_errors_24h": [],
356
+ "valid_partials_since_start": 1,
357
+ "valid_partials_24h": [1],
358
+ "invalid_partials_since_start": 0,
359
+ "invalid_partials_24h": [],
360
+ "insufficient_partials_since_start": 0,
361
+ "insufficient_partials_24h": [],
362
+ "stale_partials_since_start": 0,
363
+ "stale_partials_24h": [],
364
+ "missing_partials_since_start": 0,
365
+ "missing_partials_24h": [],
366
+ },
367
+ ),
368
+ # Empty pool_url means solo plotNFT farming
369
+ id="empty_pool_url",
370
+ ),
371
+ pytest.param(
372
+ NewProofOfSpaceCase.create_verified_quality_case(
373
+ difficulty=uint64(1),
374
+ sub_slot_iters=uint64(1000000000000),
375
+ pool_url="http://localhost",
376
+ pool_difficulty=None,
377
+ authentication_token_timeout=uint8(10),
378
+ use_invalid_peer_response=False,
379
+ has_valid_authentication_keys=True,
380
+ expected_pool_stats={
381
+ "points_found_since_start": 0,
382
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
383
+ # so every `xxx_24h` item in this dict will be list[Any].
384
+ "points_found_24h": [],
385
+ "points_acknowledged_since_start": 0,
386
+ "points_acknowledged_24h": [],
387
+ "pool_errors_24h": [],
388
+ "valid_partials_since_start": 0,
389
+ "valid_partials_24h": [],
390
+ "invalid_partials_since_start": 0,
391
+ "invalid_partials_24h": [],
392
+ "insufficient_partials_since_start": 0,
393
+ "insufficient_partials_24h": [],
394
+ "stale_partials_since_start": 0,
395
+ "stale_partials_24h": [],
396
+ "missing_partials_since_start": 1,
397
+ "missing_partials_24h": [None],
398
+ },
399
+ ),
400
+ id="empty_current_difficulty",
401
+ ),
402
+ pytest.param(
403
+ NewProofOfSpaceCase.create_verified_quality_case(
404
+ difficulty=uint64(1),
405
+ sub_slot_iters=uint64(64),
406
+ pool_url="http://localhost",
407
+ pool_difficulty=uint64(1),
408
+ authentication_token_timeout=uint8(10),
409
+ use_invalid_peer_response=False,
410
+ has_valid_authentication_keys=True,
411
+ expected_pool_stats={
412
+ "points_found_since_start": 0,
413
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
414
+ # so every `xxx_24h` item in this dict will be list[Any].
415
+ "points_found_24h": [],
416
+ "points_acknowledged_since_start": 0,
417
+ "points_acknowledged_24h": [],
418
+ "pool_errors_24h": [],
419
+ "valid_partials_since_start": 0,
420
+ "valid_partials_24h": [],
421
+ "invalid_partials_since_start": 0,
422
+ "invalid_partials_24h": [],
423
+ "insufficient_partials_since_start": 1,
424
+ "insufficient_partials_24h": [1],
425
+ "stale_partials_since_start": 0,
426
+ "stale_partials_24h": [],
427
+ "missing_partials_since_start": 0,
428
+ "missing_partials_24h": [],
429
+ },
430
+ ),
431
+ id="quality_not_good",
432
+ ),
433
+ pytest.param(
434
+ NewProofOfSpaceCase.create_verified_quality_case(
435
+ difficulty=uint64(1),
436
+ sub_slot_iters=uint64(1000000000000),
437
+ pool_url="http://localhost",
438
+ pool_difficulty=uint64(1),
439
+ authentication_token_timeout=None,
440
+ use_invalid_peer_response=False,
441
+ has_valid_authentication_keys=True,
442
+ expected_pool_stats={
443
+ "points_found_since_start": 0,
444
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
445
+ # so every `xxx_24h` item in this dict will be list[Any].
446
+ "points_found_24h": [],
447
+ "points_acknowledged_since_start": 0,
448
+ "points_acknowledged_24h": [],
449
+ "pool_errors_24h": [],
450
+ "valid_partials_since_start": 0,
451
+ "valid_partials_24h": [],
452
+ "invalid_partials_since_start": 0,
453
+ "invalid_partials_24h": [],
454
+ "insufficient_partials_since_start": 0,
455
+ "insufficient_partials_24h": [],
456
+ "stale_partials_since_start": 0,
457
+ "stale_partials_24h": [],
458
+ "missing_partials_since_start": 1,
459
+ "missing_partials_24h": [1],
460
+ },
461
+ ),
462
+ id="empty_auth_timeout",
463
+ ),
464
+ pytest.param(
465
+ NewProofOfSpaceCase.create_verified_quality_case(
466
+ difficulty=uint64(1),
467
+ sub_slot_iters=uint64(1000000000000),
468
+ pool_url="http://localhost",
469
+ pool_difficulty=uint64(1),
470
+ authentication_token_timeout=uint8(10),
471
+ use_invalid_peer_response=True,
472
+ has_valid_authentication_keys=True,
473
+ expected_pool_stats={
474
+ "points_found_since_start": 0,
475
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
476
+ # so every `xxx_24h` item in this dict will be list[Any].
477
+ "points_found_24h": [],
478
+ "points_acknowledged_since_start": 0,
479
+ "points_acknowledged_24h": [],
480
+ "pool_errors_24h": [],
481
+ "valid_partials_since_start": 0,
482
+ "valid_partials_24h": [],
483
+ "invalid_partials_since_start": 1,
484
+ "invalid_partials_24h": [1],
485
+ "insufficient_partials_since_start": 0,
486
+ "insufficient_partials_24h": [],
487
+ "stale_partials_since_start": 0,
488
+ "stale_partials_24h": [],
489
+ "missing_partials_since_start": 0,
490
+ "missing_partials_24h": [],
491
+ },
492
+ ),
493
+ id="invalid_peer_response",
494
+ ),
495
+ pytest.param(
496
+ NewProofOfSpaceCase.create_verified_quality_case(
497
+ difficulty=uint64(1),
498
+ sub_slot_iters=uint64(1000000000000),
499
+ pool_url="http://localhost",
500
+ pool_difficulty=uint64(1),
501
+ authentication_token_timeout=uint8(10),
502
+ use_invalid_peer_response=False,
503
+ has_valid_authentication_keys=False,
504
+ expected_pool_stats={
505
+ "points_found_since_start": 0,
506
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
507
+ # so every `xxx_24h` item in this dict will be list[Any].
508
+ "points_found_24h": [],
509
+ "points_acknowledged_since_start": 0,
510
+ "points_acknowledged_24h": [],
511
+ "pool_errors_24h": [],
512
+ "valid_partials_since_start": 0,
513
+ "valid_partials_24h": [],
514
+ "invalid_partials_since_start": 0,
515
+ "invalid_partials_24h": [],
516
+ "insufficient_partials_since_start": 0,
517
+ "insufficient_partials_24h": [],
518
+ "stale_partials_since_start": 0,
519
+ "stale_partials_24h": [],
520
+ "missing_partials_since_start": 1,
521
+ "missing_partials_24h": [1],
522
+ },
523
+ ),
524
+ id="no_valid_auth_keys",
525
+ ),
526
+ pytest.param(
527
+ NewProofOfSpaceCase.create_verified_quality_case(
528
+ difficulty=uint64(1),
529
+ sub_slot_iters=uint64(1000000000000),
530
+ pool_url="http://192.168.0.256",
531
+ pool_difficulty=uint64(1),
532
+ authentication_token_timeout=uint8(10),
533
+ use_invalid_peer_response=False,
534
+ has_valid_authentication_keys=True,
535
+ expected_pool_stats={
536
+ "points_found_since_start": 1,
537
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
538
+ # so every `xxx_24h` item in this dict will be list[Any].
539
+ "points_found_24h": [1],
540
+ "points_acknowledged_since_start": 0,
541
+ "points_acknowledged_24h": [],
542
+ "pool_errors_24h": [{"error_code": uint16(PoolErrorCode.REQUEST_FAILED.value)}],
543
+ "valid_partials_since_start": 0,
544
+ "valid_partials_24h": [],
545
+ "invalid_partials_since_start": 1,
546
+ "invalid_partials_24h": [1],
547
+ "insufficient_partials_since_start": 0,
548
+ "insufficient_partials_24h": [],
549
+ "stale_partials_since_start": 0,
550
+ "stale_partials_24h": [],
551
+ "missing_partials_since_start": 0,
552
+ "missing_partials_24h": [],
553
+ },
554
+ ),
555
+ id="error_connecting_pool_server",
556
+ ),
557
+ ],
558
+ )
559
+ @pytest.mark.anyio
560
+ async def test_farmer_new_proof_of_space_for_pool_stats(
561
+ harvester_farmer_environment: HarvesterFarmerEnvironment,
562
+ case: NewProofOfSpaceCase,
563
+ ) -> None:
564
+ farmer_service, _farmer_rpc_client, _, _, _ = harvester_farmer_environment
565
+ farmer_api = farmer_service._api
566
+
567
+ sp = farmer_protocol.NewSignagePoint(
568
+ challenge_hash=case.challenge_hash,
569
+ challenge_chain_sp=case.sp_hash,
570
+ reward_chain_sp=std_hash(b"1"),
571
+ difficulty=case.difficulty,
572
+ sub_slot_iters=case.sub_slot_iters,
573
+ signage_point_index=case.signage_point_index,
574
+ peak_height=uint32(1),
575
+ )
576
+ pos = ProofOfSpace(
577
+ challenge=case.plot_challenge,
578
+ pool_public_key=case.pool_public_key,
579
+ pool_contract_puzzle_hash=case.pool_contract_puzzle_hash,
580
+ plot_public_key=case.plot_public_key,
581
+ size=case.plot_size,
582
+ proof=case.proof,
583
+ )
584
+ new_pos = NewProofOfSpace(
585
+ challenge_hash=case.challenge_hash,
586
+ sp_hash=case.sp_hash,
587
+ plot_identifier=case.plot_identifier,
588
+ proof=pos,
589
+ signage_point_index=case.signage_point_index,
590
+ include_source_signature_data=False,
591
+ farmer_reward_address_override=None,
592
+ fee_info=None,
593
+ )
594
+
595
+ p2_singleton_puzzle_hash = case.pool_contract_puzzle_hash
596
+ farmer_api.farmer.constants = DEFAULT_CONSTANTS.replace(POOL_SUB_SLOT_ITERS=case.sub_slot_iters)
597
+ farmer_api.farmer._private_keys = case.farmer_private_keys
598
+ farmer_api.farmer.authentication_keys = case.authentication_keys
599
+ farmer_api.farmer.sps[case.sp_hash] = [sp]
600
+ farmer_api.farmer.pool_state[p2_singleton_puzzle_hash] = {
601
+ "p2_singleton_puzzle_hash": p2_singleton_puzzle_hash.hex(),
602
+ "points_found_since_start": 0,
603
+ "points_found_24h": [],
604
+ "points_acknowledged_since_start": 0,
605
+ "points_acknowledged_24h": [],
606
+ "next_farmer_update": 0,
607
+ "next_pool_info_update": 0,
608
+ "current_points": 0,
609
+ "current_difficulty": case.pool_difficulty,
610
+ "pool_errors_24h": [],
611
+ "valid_partials_since_start": 0,
612
+ "valid_partials_24h": [],
613
+ "invalid_partials_since_start": 0,
614
+ "invalid_partials_24h": [],
615
+ "insufficient_partials_since_start": 0,
616
+ "insufficient_partials_24h": [],
617
+ "stale_partials_since_start": 0,
618
+ "stale_partials_24h": [],
619
+ "missing_partials_since_start": 0,
620
+ "missing_partials_24h": [],
621
+ "authentication_token_timeout": case.authentication_token_timeout,
622
+ "plot_count": 0,
623
+ "pool_config": case.pool_config,
624
+ }
625
+
626
+ assert (
627
+ verify_and_get_quality_string(pos, DEFAULT_CONSTANTS, case.challenge_hash, case.sp_hash, height=uint32(1))
628
+ is not None
629
+ )
630
+
631
+ peer: Any = DummyHarvesterPeer(case.use_invalid_peer_response)
632
+ await farmer_api.new_proof_of_space(new_pos, peer)
633
+
634
+ def assert_stats_since_start(name: str) -> None:
635
+ assert farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name] == case.expected_pool_state[name]
636
+
637
+ def assert_stats_24h(name: str) -> None:
638
+ assert len(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name]) == len(case.expected_pool_state[name])
639
+ for i, stat in enumerate(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name]):
640
+ assert stat[1] == case.expected_pool_state[name][i]
641
+
642
+ def assert_pool_errors_24h() -> None:
643
+ assert len(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash]["pool_errors_24h"]) == len(
644
+ case.expected_pool_state["pool_errors_24h"]
645
+ )
646
+ for i, stat in enumerate(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash]["pool_errors_24h"]):
647
+ assert stat[1]["error_code"] == case.expected_pool_state["pool_errors_24h"][i]["error_code"]
648
+
649
+ assert_stats_since_start("points_found_since_start")
650
+ assert_stats_24h("points_found_24h")
651
+ assert_stats_since_start("points_acknowledged_since_start")
652
+ assert_stats_24h("points_acknowledged_24h")
653
+ assert_pool_errors_24h()
654
+ assert_stats_since_start("valid_partials_since_start")
655
+ assert_stats_24h("valid_partials_24h")
656
+ assert_stats_since_start("invalid_partials_since_start")
657
+ assert_stats_24h("invalid_partials_24h")
658
+ assert_stats_since_start("insufficient_partials_since_start")
659
+ assert_stats_24h("insufficient_partials_24h")
660
+ assert_stats_since_start("stale_partials_since_start")
661
+ assert_stats_24h("stale_partials_24h")
662
+ assert_stats_since_start("missing_partials_since_start")
663
+ assert_stats_24h("missing_partials_24h")
664
+
665
+
666
+ @dataclass
667
+ class DummyPoolResponse:
668
+ ok: bool
669
+ status: int
670
+ error_code: Optional[int] = None
671
+ error_message: Optional[str] = None
672
+ new_difficulty: Optional[int] = None
673
+
674
+ async def text(self) -> str:
675
+ json_dict: dict[str, Any] = dict()
676
+ if self.error_code:
677
+ json_dict["error_code"] = self.error_code
678
+ json_dict["error_message"] = self.error_message if self.error_message else "error-msg"
679
+ elif self.new_difficulty:
680
+ json_dict["new_difficulty"] = self.new_difficulty
681
+
682
+ return json.dumps(json_dict)
683
+
684
+ async def __aenter__(self) -> DummyPoolResponse:
685
+ return self
686
+
687
+ async def __aexit__(
688
+ self,
689
+ exc_type: Optional[type[BaseException]],
690
+ exc_val: Optional[BaseException],
691
+ exc_tb: Optional[TracebackType],
692
+ ) -> None:
693
+ pass
694
+
695
+
696
+ def create_valid_pos(farmer: Farmer) -> tuple[farmer_protocol.NewSignagePoint, ProofOfSpace, NewProofOfSpace]:
697
+ case = NewProofOfSpaceCase.create_verified_quality_case(
698
+ difficulty=uint64(1),
699
+ sub_slot_iters=uint64(1000000000000),
700
+ pool_url="https://192.168.0.256",
701
+ pool_difficulty=uint64(1),
702
+ authentication_token_timeout=uint8(10),
703
+ use_invalid_peer_response=False,
704
+ has_valid_authentication_keys=True,
705
+ expected_pool_stats={},
706
+ )
707
+ sp = farmer_protocol.NewSignagePoint(
708
+ challenge_hash=case.challenge_hash,
709
+ challenge_chain_sp=case.sp_hash,
710
+ reward_chain_sp=std_hash(b"1"),
711
+ difficulty=case.difficulty,
712
+ sub_slot_iters=case.sub_slot_iters,
713
+ signage_point_index=case.signage_point_index,
714
+ peak_height=uint32(1),
715
+ )
716
+ pos = ProofOfSpace(
717
+ challenge=case.plot_challenge,
718
+ pool_public_key=case.pool_public_key,
719
+ pool_contract_puzzle_hash=case.pool_contract_puzzle_hash,
720
+ plot_public_key=case.plot_public_key,
721
+ size=case.plot_size,
722
+ proof=case.proof,
723
+ )
724
+ new_pos = NewProofOfSpace(
725
+ challenge_hash=case.challenge_hash,
726
+ sp_hash=case.sp_hash,
727
+ plot_identifier=case.plot_identifier,
728
+ proof=pos,
729
+ signage_point_index=case.signage_point_index,
730
+ include_source_signature_data=False,
731
+ farmer_reward_address_override=None,
732
+ fee_info=None,
733
+ )
734
+ p2_singleton_puzzle_hash = case.pool_contract_puzzle_hash
735
+ farmer.constants = DEFAULT_CONSTANTS.replace(POOL_SUB_SLOT_ITERS=case.sub_slot_iters)
736
+ farmer._private_keys = case.farmer_private_keys
737
+ farmer.authentication_keys = case.authentication_keys
738
+ farmer.sps[case.sp_hash] = [sp]
739
+ farmer.pool_state[p2_singleton_puzzle_hash] = {
740
+ "p2_singleton_puzzle_hash": p2_singleton_puzzle_hash.hex(),
741
+ "points_found_since_start": 0,
742
+ "points_found_24h": [],
743
+ "points_acknowledged_since_start": 0,
744
+ "points_acknowledged_24h": [],
745
+ "next_farmer_update": 0,
746
+ "next_pool_info_update": 0,
747
+ "current_points": 0,
748
+ "current_difficulty": case.pool_difficulty,
749
+ "pool_errors_24h": [],
750
+ "valid_partials_since_start": 0,
751
+ "valid_partials_24h": [],
752
+ "invalid_partials_since_start": 0,
753
+ "invalid_partials_24h": [],
754
+ "insufficient_partials_since_start": 0,
755
+ "insufficient_partials_24h": [],
756
+ "stale_partials_since_start": 0,
757
+ "stale_partials_24h": [],
758
+ "missing_partials_since_start": 0,
759
+ "missing_partials_24h": [],
760
+ "authentication_token_timeout": case.authentication_token_timeout,
761
+ "plot_count": 0,
762
+ "pool_config": case.pool_config,
763
+ }
764
+ return sp, pos, new_pos
765
+
766
+
767
+ def override_pool_state(overrides: dict[str, Any]) -> dict[str, Any]:
768
+ pool_state = {
769
+ "points_found_since_start": 0,
770
+ # Original item format here is (timestamp, value) but we'll ignore timestamp part
771
+ # so every `xxx_24h` item in this dict will be list[Any].
772
+ "points_found_24h": [],
773
+ "points_acknowledged_since_start": 0,
774
+ "points_acknowledged_24h": [],
775
+ "pool_errors_24h": [],
776
+ "valid_partials_since_start": 0,
777
+ "valid_partials_24h": [],
778
+ "invalid_partials_since_start": 0,
779
+ "invalid_partials_24h": [],
780
+ "insufficient_partials_since_start": 0,
781
+ "insufficient_partials_24h": [],
782
+ "stale_partials_since_start": 0,
783
+ "stale_partials_24h": [],
784
+ "missing_partials_since_start": 0,
785
+ "missing_partials_24h": [],
786
+ }
787
+ for key, value in overrides.items():
788
+ pool_state[key] = value
789
+ return pool_state
790
+
791
+
792
+ @dataclass
793
+ class PoolStateCase:
794
+ id: str
795
+ pool_response: DummyPoolResponse
796
+ expected_pool_state: dict[str, Any]
797
+ marks: Marks = ()
798
+
799
+
800
+ @datacases(
801
+ PoolStateCase(
802
+ "valid_response",
803
+ DummyPoolResponse(True, 200, new_difficulty=123),
804
+ override_pool_state(
805
+ {
806
+ "points_found_since_start": 1,
807
+ "points_found_24h": [1],
808
+ "points_acknowledged_since_start": 123,
809
+ "points_acknowledged_24h": [123],
810
+ "valid_partials_since_start": 1,
811
+ "valid_partials_24h": [1],
812
+ }
813
+ ),
814
+ ),
815
+ PoolStateCase(
816
+ "response_not_ok",
817
+ DummyPoolResponse(False, 500),
818
+ override_pool_state(
819
+ {
820
+ "points_found_since_start": 1,
821
+ "points_found_24h": [1],
822
+ "invalid_partials_since_start": 1,
823
+ "invalid_partials_24h": [1],
824
+ }
825
+ ),
826
+ ),
827
+ PoolStateCase(
828
+ "stale_partial",
829
+ DummyPoolResponse(True, 200, error_code=uint16(PoolErrorCode.TOO_LATE.value)),
830
+ override_pool_state(
831
+ {
832
+ "points_found_since_start": 1,
833
+ "points_found_24h": [1],
834
+ "pool_errors_24h": [{"error_code": uint16(PoolErrorCode.TOO_LATE.value)}],
835
+ "stale_partials_since_start": 1,
836
+ "stale_partials_24h": [1],
837
+ }
838
+ ),
839
+ ),
840
+ PoolStateCase(
841
+ "insufficient_partial",
842
+ DummyPoolResponse(True, 200, error_code=uint16(PoolErrorCode.PROOF_NOT_GOOD_ENOUGH.value)),
843
+ override_pool_state(
844
+ {
845
+ "points_found_since_start": 1,
846
+ "points_found_24h": [1],
847
+ "pool_errors_24h": [{"error_code": uint16(PoolErrorCode.PROOF_NOT_GOOD_ENOUGH.value)}],
848
+ "insufficient_partials_since_start": 1,
849
+ "insufficient_partials_24h": [1],
850
+ }
851
+ ),
852
+ ),
853
+ PoolStateCase(
854
+ "other_failed_partial",
855
+ DummyPoolResponse(True, 200, error_code=uint16(PoolErrorCode.SERVER_EXCEPTION.value)),
856
+ override_pool_state(
857
+ {
858
+ "points_found_since_start": 1,
859
+ "points_found_24h": [1],
860
+ "pool_errors_24h": [{"error_code": uint16(PoolErrorCode.SERVER_EXCEPTION.value)}],
861
+ "invalid_partials_since_start": 1,
862
+ "invalid_partials_24h": [1],
863
+ }
864
+ ),
865
+ ),
866
+ )
867
+ @pytest.mark.anyio
868
+ async def test_farmer_pool_response(
869
+ mocker: MockerFixture,
870
+ farmer_one_harvester: tuple[list[HarvesterService], FarmerService, BlockTools],
871
+ case: PoolStateCase,
872
+ ) -> None:
873
+ _, farmer_service, _ = farmer_one_harvester
874
+ assert farmer_service.rpc_server is not None
875
+ farmer_api = farmer_service._api
876
+
877
+ sp, pos, new_pos = create_valid_pos(farmer_api.farmer)
878
+ assert pos.pool_contract_puzzle_hash is not None
879
+ p2_singleton_puzzle_hash: bytes32 = pos.pool_contract_puzzle_hash
880
+
881
+ assert (
882
+ verify_and_get_quality_string(
883
+ pos, DEFAULT_CONSTANTS, sp.challenge_hash, sp.challenge_chain_sp, height=uint32(1)
884
+ )
885
+ is not None
886
+ )
887
+
888
+ pool_response = case.pool_response
889
+ expected_pool_state = case.expected_pool_state
890
+
891
+ mock_http_post = mocker.patch("aiohttp.ClientSession.post", return_value=pool_response)
892
+
893
+ peer = cast(WSChiaConnection, DummyHarvesterPeer(False))
894
+ await farmer_api.new_proof_of_space(new_pos, peer)
895
+
896
+ mock_http_post.assert_called_once()
897
+
898
+ def assert_stats_since_start(name: str) -> None:
899
+ assert farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name] == expected_pool_state[name]
900
+
901
+ def assert_stats_24h(name: str) -> None:
902
+ assert len(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name]) == len(expected_pool_state[name])
903
+ for i, stat in enumerate(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash][name]):
904
+ assert stat[1] == expected_pool_state[name][i]
905
+
906
+ def assert_pool_errors_24h() -> None:
907
+ assert len(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash]["pool_errors_24h"]) == len(
908
+ expected_pool_state["pool_errors_24h"]
909
+ )
910
+ for i, stat in enumerate(farmer_api.farmer.pool_state[p2_singleton_puzzle_hash]["pool_errors_24h"]):
911
+ assert stat[1]["error_code"] == expected_pool_state["pool_errors_24h"][i]["error_code"]
912
+
913
+ assert_stats_since_start("points_found_since_start")
914
+ assert_stats_24h("points_found_24h")
915
+ assert_stats_since_start("points_acknowledged_since_start")
916
+ assert_stats_24h("points_acknowledged_24h")
917
+ assert_pool_errors_24h()
918
+ assert_stats_since_start("valid_partials_since_start")
919
+ assert_stats_24h("valid_partials_24h")
920
+ assert_stats_since_start("invalid_partials_since_start")
921
+ assert_stats_24h("invalid_partials_24h")
922
+ assert_stats_since_start("insufficient_partials_since_start")
923
+ assert_stats_24h("insufficient_partials_24h")
924
+ assert_stats_since_start("stale_partials_since_start")
925
+ assert_stats_24h("stale_partials_24h")
926
+ assert_stats_since_start("missing_partials_since_start")
927
+ assert_stats_24h("missing_partials_24h")
928
+
929
+
930
+ def make_pool_list_entry(overrides: dict[str, Any]) -> dict[str, Any]:
931
+ pool_list_entry = {
932
+ "owner_public_key": "84c3fcf9d5581c1ddc702cb0f3b4a06043303b334dd993ab42b2c320ebfa98e5ce558448615b3f69638ba92cf7f43da5", # noqa: E501
933
+ "p2_singleton_puzzle_hash": "302e05a1e6af431c22043ae2a9a8f71148c955c372697cb8ab348160976283df",
934
+ "payout_instructions": "c2b08e41d766da4116e388357ed957d04ad754623a915f3fd65188a8746cf3e8",
935
+ "pool_url": "localhost",
936
+ "launcher_id": "ae4ef3b9bfe68949691281a015a9c16630fc8f66d48c19ca548fb80768791afa",
937
+ "target_puzzle_hash": "344587cf06a39db471d2cc027504e8688a0a67cce961253500c956c73603fd58",
938
+ }
939
+ for key, value in overrides.items():
940
+ pool_list_entry[key] = value
941
+ return pool_list_entry
942
+
943
+
944
+ def make_pool_info() -> dict[str, Any]:
945
+ return {
946
+ "name": "Pool Name",
947
+ "description": "Pool Description",
948
+ "logo_url": "https://subdomain.pool-domain.tld/path/to/logo.svg",
949
+ "target_puzzle_hash": "344587cf06a39db471d2cc027504e8688a0a67cce961253500c956c73603fd58",
950
+ "fee": "0.01",
951
+ "protocol_version": 1,
952
+ "relative_lock_height": 100,
953
+ "minimum_difficulty": 1,
954
+ "authentication_token_timeout": 5,
955
+ }
956
+
957
+
958
+ def make_pool_state(p2_singleton_puzzle_hash: bytes32, overrides: dict[str, Any]) -> dict[str, Any]:
959
+ pool_info = {
960
+ "p2_singleton_puzzle_hash": p2_singleton_puzzle_hash.hex(),
961
+ "points_found_since_start": 0,
962
+ "points_found_24h": [],
963
+ "points_acknowledged_since_start": 0,
964
+ "points_acknowledged_24h": [],
965
+ "next_farmer_update": 0,
966
+ "next_pool_info_update": 0,
967
+ "current_points": 0,
968
+ "current_difficulty": None,
969
+ "pool_errors_24h": [],
970
+ "valid_partials_since_start": 0,
971
+ "valid_partials_24h": [],
972
+ "invalid_partials_since_start": 0,
973
+ "invalid_partials_24h": [],
974
+ "insufficient_partials_since_start": 0,
975
+ "insufficient_partials_24h": [],
976
+ "stale_partials_since_start": 0,
977
+ "stale_partials_24h": [],
978
+ "missing_partials_since_start": 0,
979
+ "missing_partials_24h": [],
980
+ "authentication_token_timeout": None,
981
+ "plot_count": 0,
982
+ }
983
+ for key, value in overrides.items():
984
+ pool_info[key] = value
985
+ return pool_info
986
+
987
+
988
+ @dataclass
989
+ class DummyClientResponse:
990
+ status: int
991
+
992
+
993
+ @dataclass
994
+ class DummyPoolInfoResponse:
995
+ ok: bool
996
+ status: int
997
+ url: URL
998
+ pool_info: Optional[dict[str, Any]] = None
999
+ history: tuple[DummyClientResponse, ...] = ()
1000
+
1001
+ async def text(self) -> str:
1002
+ if self.pool_info is None:
1003
+ return ""
1004
+
1005
+ return json.dumps(self.pool_info)
1006
+
1007
+ async def __aenter__(self) -> DummyPoolInfoResponse:
1008
+ return self
1009
+
1010
+ async def __aexit__(
1011
+ self,
1012
+ exc_type: Optional[type[BaseException]],
1013
+ exc_val: Optional[BaseException],
1014
+ exc_tb: Optional[TracebackType],
1015
+ ) -> None:
1016
+ pass
1017
+
1018
+
1019
+ @dataclass
1020
+ class PoolInfoCase(DataCase):
1021
+ _id: str
1022
+ initial_pool_url_in_config: str
1023
+ pool_response: DummyPoolInfoResponse
1024
+ expected_pool_url_in_config: str
1025
+ marks: Marks = ()
1026
+
1027
+ @property
1028
+ def id(self) -> str:
1029
+ return self._id
1030
+
1031
+
1032
+ @datacases(
1033
+ PoolInfoCase(
1034
+ "valid_response_without_redirect",
1035
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1036
+ pool_response=DummyPoolInfoResponse(
1037
+ ok=True,
1038
+ status=200,
1039
+ url=URL("https://endpoint-1.pool-domain.tld/some-path"),
1040
+ pool_info=make_pool_info(),
1041
+ ),
1042
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1043
+ ),
1044
+ PoolInfoCase(
1045
+ "valid_response_with_301_redirect",
1046
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1047
+ pool_response=DummyPoolInfoResponse(
1048
+ ok=True,
1049
+ status=200,
1050
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1051
+ pool_info=make_pool_info(),
1052
+ history=tuple([DummyClientResponse(status=301)]),
1053
+ ),
1054
+ expected_pool_url_in_config="https://endpoint-1337.pool-domain.tld/some-other-path",
1055
+ ),
1056
+ PoolInfoCase(
1057
+ "valid_response_with_302_redirect",
1058
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1059
+ pool_response=DummyPoolInfoResponse(
1060
+ ok=True,
1061
+ status=200,
1062
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1063
+ pool_info=make_pool_info(),
1064
+ history=tuple([DummyClientResponse(status=302)]),
1065
+ ),
1066
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1067
+ ),
1068
+ PoolInfoCase(
1069
+ "valid_response_with_307_redirect",
1070
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1071
+ pool_response=DummyPoolInfoResponse(
1072
+ ok=True,
1073
+ status=200,
1074
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1075
+ pool_info=make_pool_info(),
1076
+ history=tuple([DummyClientResponse(status=307)]),
1077
+ ),
1078
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1079
+ ),
1080
+ PoolInfoCase(
1081
+ "valid_response_with_308_redirect",
1082
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1083
+ pool_response=DummyPoolInfoResponse(
1084
+ ok=True,
1085
+ status=200,
1086
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1087
+ pool_info=make_pool_info(),
1088
+ history=tuple([DummyClientResponse(status=308)]),
1089
+ ),
1090
+ expected_pool_url_in_config="https://endpoint-1337.pool-domain.tld/some-other-path",
1091
+ ),
1092
+ PoolInfoCase(
1093
+ "valid_response_with_multiple_308_redirects",
1094
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1095
+ pool_response=DummyPoolInfoResponse(
1096
+ ok=True,
1097
+ status=200,
1098
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1099
+ pool_info=make_pool_info(),
1100
+ history=tuple([DummyClientResponse(status=308), DummyClientResponse(status=308)]),
1101
+ ),
1102
+ expected_pool_url_in_config="https://endpoint-1337.pool-domain.tld/some-other-path",
1103
+ ),
1104
+ PoolInfoCase(
1105
+ "valid_response_with_multiple_307_and_308_redirects",
1106
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1107
+ pool_response=DummyPoolInfoResponse(
1108
+ ok=True,
1109
+ status=200,
1110
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1111
+ pool_info=make_pool_info(),
1112
+ history=tuple([DummyClientResponse(status=307), DummyClientResponse(status=308)]),
1113
+ ),
1114
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1115
+ ),
1116
+ PoolInfoCase(
1117
+ "failed_request_without_redirect",
1118
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1119
+ pool_response=DummyPoolInfoResponse(
1120
+ ok=False,
1121
+ status=500,
1122
+ url=URL("https://endpoint-1.pool-domain.tld/some-path"),
1123
+ ),
1124
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1125
+ ),
1126
+ PoolInfoCase(
1127
+ "failed_request_with_301_redirect",
1128
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1129
+ pool_response=DummyPoolInfoResponse(
1130
+ ok=False,
1131
+ status=500,
1132
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1133
+ history=tuple([DummyClientResponse(status=301)]),
1134
+ ),
1135
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1136
+ ),
1137
+ PoolInfoCase(
1138
+ "failed_request_with_302_redirect",
1139
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1140
+ pool_response=DummyPoolInfoResponse(
1141
+ ok=False,
1142
+ status=500,
1143
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1144
+ history=tuple([DummyClientResponse(status=302)]),
1145
+ ),
1146
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1147
+ ),
1148
+ PoolInfoCase(
1149
+ "failed_request_with_307_redirect",
1150
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1151
+ pool_response=DummyPoolInfoResponse(
1152
+ ok=False,
1153
+ status=500,
1154
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1155
+ history=tuple([DummyClientResponse(status=307)]),
1156
+ ),
1157
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1158
+ ),
1159
+ PoolInfoCase(
1160
+ "failed_request_with_308_redirect",
1161
+ initial_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1162
+ pool_response=DummyPoolInfoResponse(
1163
+ ok=False,
1164
+ status=500,
1165
+ url=URL("https://endpoint-1337.pool-domain.tld/some-other-path"),
1166
+ history=tuple([DummyClientResponse(status=308)]),
1167
+ ),
1168
+ expected_pool_url_in_config="https://endpoint-1.pool-domain.tld/some-path",
1169
+ ),
1170
+ )
1171
+ @pytest.mark.anyio
1172
+ async def test_farmer_pool_info_config_update(
1173
+ mocker: MockerFixture,
1174
+ farmer_one_harvester: tuple[list[HarvesterService], FarmerService, BlockTools],
1175
+ case: PoolInfoCase,
1176
+ ) -> None:
1177
+ _, farmer_service, _ = farmer_one_harvester
1178
+ p2_singleton_puzzle_hash = bytes32.fromhex("302e05a1e6af431c22043ae2a9a8f71148c955c372697cb8ab348160976283df")
1179
+ farmer_service._node.authentication_keys = {
1180
+ p2_singleton_puzzle_hash: PrivateKey.from_bytes(
1181
+ bytes.fromhex("11ed596eb95b31364a9185e948f6b66be30415f816819449d5d40751dc70e786")
1182
+ ),
1183
+ }
1184
+ farmer_service._node.pool_state[p2_singleton_puzzle_hash] = make_pool_state(
1185
+ p2_singleton_puzzle_hash,
1186
+ overrides={
1187
+ "next_farmer_update": time() + UPDATE_POOL_FARMER_INFO_INTERVAL,
1188
+ },
1189
+ )
1190
+ config = load_config(farmer_service.root_path, "config.yaml")
1191
+ config["pool"]["pool_list"] = [
1192
+ make_pool_list_entry(
1193
+ overrides={
1194
+ "p2_singleton_puzzle_hash": p2_singleton_puzzle_hash.hex(),
1195
+ "pool_url": case.initial_pool_url_in_config,
1196
+ }
1197
+ )
1198
+ ]
1199
+ save_config(farmer_service.root_path, "config.yaml", config)
1200
+ mock_http_get = mocker.patch("aiohttp.ClientSession.get", return_value=case.pool_response)
1201
+
1202
+ await farmer_service._node.update_pool_state()
1203
+
1204
+ mock_http_get.assert_called_once()
1205
+ config = load_config(farmer_service.root_path, "config.yaml")
1206
+ assert len(config["pool"]["pool_list"]) == 1
1207
+ assert config["pool"]["pool_list"][0]["p2_singleton_puzzle_hash"] == p2_singleton_puzzle_hash.hex()
1208
+ assert config["pool"]["pool_list"][0]["pool_url"] == case.expected_pool_url_in_config
1209
+
1210
+
1211
+ @dataclass
1212
+ class PartialSubmitHeaderCase(DataCase):
1213
+ _id: str
1214
+ harvester_peer: DummyHarvesterPeer
1215
+ expected_headers: dict[str, str]
1216
+ marks: Marks = ()
1217
+
1218
+ @property
1219
+ def id(self) -> str:
1220
+ return self._id
1221
+
1222
+
1223
+ @datacases(
1224
+ PartialSubmitHeaderCase(
1225
+ "additional version headers",
1226
+ harvester_peer=DummyHarvesterPeer(
1227
+ version="1.2.3.asdf42",
1228
+ ),
1229
+ expected_headers={
1230
+ "User-Agent": f"Chia Blockchain v.{__version__}",
1231
+ "chia-farmer-version": __version__,
1232
+ "chia-harvester-version": "1.2.3.asdf42",
1233
+ },
1234
+ ),
1235
+ )
1236
+ @pytest.mark.anyio
1237
+ async def test_farmer_additional_headers_on_partial_submit(
1238
+ mocker: MockerFixture,
1239
+ farmer_one_harvester: tuple[list[HarvesterService], FarmerService, BlockTools],
1240
+ case: PartialSubmitHeaderCase,
1241
+ ) -> None:
1242
+ _, farmer_service, _ = farmer_one_harvester
1243
+ assert farmer_service.rpc_server is not None
1244
+ farmer_api = farmer_service._api
1245
+
1246
+ sp, pos, new_pos = create_valid_pos(farmer_api.farmer)
1247
+ assert pos.pool_contract_puzzle_hash is not None
1248
+
1249
+ assert (
1250
+ verify_and_get_quality_string(
1251
+ pos, DEFAULT_CONSTANTS, sp.challenge_hash, sp.challenge_chain_sp, height=uint32(1)
1252
+ )
1253
+ is not None
1254
+ )
1255
+
1256
+ mock_http_post = mocker.patch(
1257
+ "aiohttp.ClientSession.post",
1258
+ return_value=DummyPoolResponse(True, 200, new_difficulty=123),
1259
+ )
1260
+
1261
+ peer = cast(WSChiaConnection, case.harvester_peer)
1262
+ await farmer_api.new_proof_of_space(new_pos, peer)
1263
+
1264
+ mock_http_post.assert_called_once_with(ANY, json=ANY, ssl=ANY, headers=case.expected_headers)