crawlo 1.4.5__py3-none-any.whl → 1.4.7__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.

Potentially problematic release.


This version of crawlo might be problematic. Click here for more details.

Files changed (375) hide show
  1. crawlo/__init__.py +90 -89
  2. crawlo/__version__.py +1 -1
  3. crawlo/cli.py +75 -75
  4. crawlo/commands/__init__.py +14 -14
  5. crawlo/commands/check.py +594 -594
  6. crawlo/commands/genspider.py +186 -186
  7. crawlo/commands/help.py +140 -138
  8. crawlo/commands/list.py +155 -155
  9. crawlo/commands/run.py +379 -341
  10. crawlo/commands/startproject.py +460 -460
  11. crawlo/commands/stats.py +187 -187
  12. crawlo/commands/utils.py +196 -196
  13. crawlo/config.py +320 -312
  14. crawlo/config_validator.py +277 -277
  15. crawlo/core/__init__.py +52 -52
  16. crawlo/core/engine.py +451 -438
  17. crawlo/core/processor.py +47 -47
  18. crawlo/core/scheduler.py +290 -291
  19. crawlo/crawler.py +698 -657
  20. crawlo/data/__init__.py +5 -5
  21. crawlo/data/user_agents.py +194 -194
  22. crawlo/downloader/__init__.py +280 -276
  23. crawlo/downloader/aiohttp_downloader.py +233 -233
  24. crawlo/downloader/cffi_downloader.py +250 -245
  25. crawlo/downloader/httpx_downloader.py +265 -259
  26. crawlo/downloader/hybrid_downloader.py +212 -212
  27. crawlo/downloader/playwright_downloader.py +425 -402
  28. crawlo/downloader/selenium_downloader.py +486 -472
  29. crawlo/event.py +45 -11
  30. crawlo/exceptions.py +215 -82
  31. crawlo/extension/__init__.py +65 -64
  32. crawlo/extension/health_check.py +141 -141
  33. crawlo/extension/log_interval.py +94 -94
  34. crawlo/extension/log_stats.py +70 -70
  35. crawlo/extension/logging_extension.py +53 -61
  36. crawlo/extension/memory_monitor.py +104 -104
  37. crawlo/extension/performance_profiler.py +133 -133
  38. crawlo/extension/request_recorder.py +107 -107
  39. crawlo/factories/__init__.py +27 -27
  40. crawlo/factories/base.py +68 -68
  41. crawlo/factories/crawler.py +104 -103
  42. crawlo/factories/registry.py +84 -84
  43. crawlo/factories/utils.py +135 -0
  44. crawlo/filters/__init__.py +170 -153
  45. crawlo/filters/aioredis_filter.py +348 -264
  46. crawlo/filters/memory_filter.py +261 -276
  47. crawlo/framework.py +306 -292
  48. crawlo/initialization/__init__.py +44 -44
  49. crawlo/initialization/built_in.py +391 -434
  50. crawlo/initialization/context.py +141 -141
  51. crawlo/initialization/core.py +240 -194
  52. crawlo/initialization/phases.py +230 -149
  53. crawlo/initialization/registry.py +143 -145
  54. crawlo/initialization/utils.py +49 -0
  55. crawlo/interfaces.py +23 -23
  56. crawlo/items/__init__.py +23 -23
  57. crawlo/items/base.py +23 -23
  58. crawlo/items/fields.py +52 -52
  59. crawlo/items/items.py +104 -104
  60. crawlo/logging/__init__.py +42 -46
  61. crawlo/logging/config.py +277 -197
  62. crawlo/logging/factory.py +175 -171
  63. crawlo/logging/manager.py +104 -112
  64. crawlo/middleware/__init__.py +87 -24
  65. crawlo/middleware/default_header.py +132 -132
  66. crawlo/middleware/download_delay.py +104 -104
  67. crawlo/middleware/middleware_manager.py +142 -142
  68. crawlo/middleware/offsite.py +123 -123
  69. crawlo/middleware/proxy.py +209 -386
  70. crawlo/middleware/request_ignore.py +86 -86
  71. crawlo/middleware/response_code.py +150 -150
  72. crawlo/middleware/response_filter.py +136 -136
  73. crawlo/middleware/retry.py +124 -124
  74. crawlo/mode_manager.py +287 -253
  75. crawlo/network/__init__.py +21 -21
  76. crawlo/network/request.py +375 -379
  77. crawlo/network/response.py +569 -664
  78. crawlo/pipelines/__init__.py +53 -22
  79. crawlo/pipelines/base_pipeline.py +452 -0
  80. crawlo/pipelines/bloom_dedup_pipeline.py +146 -146
  81. crawlo/pipelines/console_pipeline.py +39 -39
  82. crawlo/pipelines/csv_pipeline.py +316 -316
  83. crawlo/pipelines/database_dedup_pipeline.py +197 -197
  84. crawlo/pipelines/json_pipeline.py +218 -218
  85. crawlo/pipelines/memory_dedup_pipeline.py +105 -105
  86. crawlo/pipelines/mongo_pipeline.py +140 -132
  87. crawlo/pipelines/mysql_pipeline.py +470 -326
  88. crawlo/pipelines/pipeline_manager.py +100 -100
  89. crawlo/pipelines/redis_dedup_pipeline.py +155 -156
  90. crawlo/project.py +347 -347
  91. crawlo/queue/__init__.py +10 -0
  92. crawlo/queue/pqueue.py +38 -38
  93. crawlo/queue/queue_manager.py +591 -525
  94. crawlo/queue/redis_priority_queue.py +519 -370
  95. crawlo/settings/__init__.py +7 -7
  96. crawlo/settings/default_settings.py +285 -270
  97. crawlo/settings/setting_manager.py +219 -219
  98. crawlo/spider/__init__.py +657 -657
  99. crawlo/stats_collector.py +82 -73
  100. crawlo/subscriber.py +129 -129
  101. crawlo/task_manager.py +138 -138
  102. crawlo/templates/crawlo.cfg.tmpl +10 -10
  103. crawlo/templates/project/__init__.py.tmpl +2 -4
  104. crawlo/templates/project/items.py.tmpl +13 -17
  105. crawlo/templates/project/middlewares.py.tmpl +38 -38
  106. crawlo/templates/project/pipelines.py.tmpl +35 -36
  107. crawlo/templates/project/settings.py.tmpl +110 -157
  108. crawlo/templates/project/settings_distributed.py.tmpl +156 -161
  109. crawlo/templates/project/settings_gentle.py.tmpl +170 -171
  110. crawlo/templates/project/settings_high_performance.py.tmpl +171 -172
  111. crawlo/templates/project/settings_minimal.py.tmpl +99 -77
  112. crawlo/templates/project/settings_simple.py.tmpl +168 -169
  113. crawlo/templates/project/spiders/__init__.py.tmpl +9 -9
  114. crawlo/templates/run.py.tmpl +23 -30
  115. crawlo/templates/spider/spider.py.tmpl +33 -144
  116. crawlo/templates/spiders_init.py.tmpl +5 -10
  117. crawlo/tools/__init__.py +86 -189
  118. crawlo/tools/date_tools.py +289 -289
  119. crawlo/tools/distributed_coordinator.py +384 -384
  120. crawlo/tools/scenario_adapter.py +262 -262
  121. crawlo/tools/text_cleaner.py +232 -232
  122. crawlo/utils/__init__.py +50 -50
  123. crawlo/utils/batch_processor.py +276 -259
  124. crawlo/utils/config_manager.py +442 -0
  125. crawlo/utils/controlled_spider_mixin.py +439 -439
  126. crawlo/utils/db_helper.py +250 -244
  127. crawlo/utils/error_handler.py +410 -410
  128. crawlo/utils/fingerprint.py +121 -121
  129. crawlo/utils/func_tools.py +82 -82
  130. crawlo/utils/large_scale_helper.py +344 -344
  131. crawlo/utils/leak_detector.py +335 -0
  132. crawlo/utils/log.py +79 -79
  133. crawlo/utils/misc.py +81 -81
  134. crawlo/utils/mongo_connection_pool.py +157 -0
  135. crawlo/utils/mysql_connection_pool.py +197 -0
  136. crawlo/utils/performance_monitor.py +285 -285
  137. crawlo/utils/queue_helper.py +175 -175
  138. crawlo/utils/redis_checker.py +91 -0
  139. crawlo/utils/redis_connection_pool.py +578 -388
  140. crawlo/utils/redis_key_validator.py +198 -198
  141. crawlo/utils/request.py +278 -256
  142. crawlo/utils/request_serializer.py +225 -225
  143. crawlo/utils/resource_manager.py +337 -0
  144. crawlo/utils/selector_helper.py +137 -137
  145. crawlo/utils/singleton.py +70 -0
  146. crawlo/utils/spider_loader.py +201 -201
  147. crawlo/utils/text_helper.py +94 -94
  148. crawlo/utils/{url.py → url_utils.py} +39 -39
  149. crawlo-1.4.7.dist-info/METADATA +689 -0
  150. crawlo-1.4.7.dist-info/RECORD +347 -0
  151. examples/__init__.py +7 -7
  152. tests/__init__.py +7 -7
  153. tests/advanced_tools_example.py +217 -275
  154. tests/authenticated_proxy_example.py +110 -106
  155. tests/baidu_performance_test.py +108 -108
  156. tests/baidu_test.py +59 -59
  157. tests/bug_check_test.py +250 -250
  158. tests/cleaners_example.py +160 -160
  159. tests/comprehensive_framework_test.py +212 -212
  160. tests/comprehensive_test.py +81 -81
  161. tests/comprehensive_testing_summary.md +186 -186
  162. tests/config_validation_demo.py +142 -142
  163. tests/controlled_spider_example.py +205 -205
  164. tests/date_tools_example.py +180 -180
  165. tests/debug_configure.py +69 -69
  166. tests/debug_framework_logger.py +84 -84
  167. tests/debug_log_config.py +126 -126
  168. tests/debug_log_levels.py +63 -63
  169. tests/debug_pipelines.py +66 -66
  170. tests/detailed_log_test.py +233 -233
  171. tests/direct_selector_helper_test.py +96 -96
  172. tests/distributed_dedup_test.py +467 -0
  173. tests/distributed_test.py +66 -66
  174. tests/distributed_test_debug.py +76 -76
  175. tests/dynamic_loading_example.py +523 -523
  176. tests/dynamic_loading_test.py +104 -104
  177. tests/error_handling_example.py +171 -171
  178. tests/explain_mysql_update_behavior.py +77 -0
  179. tests/final_comprehensive_test.py +151 -151
  180. tests/final_log_test.py +260 -260
  181. tests/final_validation_test.py +182 -182
  182. tests/fix_log_test.py +142 -142
  183. tests/framework_performance_test.py +202 -202
  184. tests/log_buffering_test.py +111 -111
  185. tests/log_generation_timing_test.py +153 -153
  186. tests/monitor_redis_dedup.sh +72 -0
  187. tests/ofweek_scrapy/ofweek_scrapy/items.py +12 -12
  188. tests/ofweek_scrapy/ofweek_scrapy/middlewares.py +100 -100
  189. tests/ofweek_scrapy/ofweek_scrapy/pipelines.py +13 -13
  190. tests/ofweek_scrapy/ofweek_scrapy/settings.py +84 -84
  191. tests/ofweek_scrapy/scrapy.cfg +11 -11
  192. tests/optimized_performance_test.py +211 -211
  193. tests/performance_comparison.py +244 -244
  194. tests/queue_blocking_test.py +113 -113
  195. tests/queue_test.py +89 -89
  196. tests/redis_key_validation_demo.py +130 -130
  197. tests/request_params_example.py +150 -150
  198. tests/response_improvements_example.py +144 -144
  199. tests/scrapy_comparison/ofweek_scrapy.py +138 -138
  200. tests/scrapy_comparison/scrapy_test.py +133 -133
  201. tests/simple_cli_test.py +55 -0
  202. tests/simple_command_test.py +119 -119
  203. tests/simple_crawlo_test.py +126 -126
  204. tests/simple_follow_test.py +38 -38
  205. tests/simple_log_test2.py +137 -137
  206. tests/simple_optimization_test.py +128 -128
  207. tests/simple_queue_type_test.py +41 -41
  208. tests/simple_response_selector_test.py +94 -94
  209. tests/simple_selector_helper_test.py +154 -154
  210. tests/simple_selector_test.py +207 -207
  211. tests/simple_spider_test.py +49 -49
  212. tests/simple_url_test.py +73 -73
  213. tests/simulate_mysql_update_test.py +140 -0
  214. tests/spider_log_timing_test.py +177 -177
  215. tests/test_advanced_tools.py +148 -148
  216. tests/test_all_commands.py +230 -230
  217. tests/test_all_pipeline_fingerprints.py +133 -133
  218. tests/test_all_redis_key_configs.py +145 -145
  219. tests/test_asyncmy_usage.py +57 -0
  220. tests/test_batch_processor.py +178 -178
  221. tests/test_cleaners.py +54 -54
  222. tests/test_cli_arguments.py +119 -0
  223. tests/test_component_factory.py +174 -174
  224. tests/test_config_consistency.py +80 -80
  225. tests/test_config_merge.py +152 -152
  226. tests/test_config_validator.py +182 -182
  227. tests/test_controlled_spider_mixin.py +79 -79
  228. tests/test_crawler_process_import.py +38 -38
  229. tests/test_crawler_process_spider_modules.py +47 -47
  230. tests/test_crawlo_proxy_integration.py +114 -108
  231. tests/test_date_tools.py +123 -123
  232. tests/test_dedup_fix.py +220 -220
  233. tests/test_dedup_pipeline_consistency.py +124 -124
  234. tests/test_default_header_middleware.py +313 -313
  235. tests/test_distributed.py +65 -65
  236. tests/test_double_crawlo_fix.py +204 -204
  237. tests/test_double_crawlo_fix_simple.py +124 -124
  238. tests/test_download_delay_middleware.py +221 -221
  239. tests/test_downloader_proxy_compatibility.py +272 -268
  240. tests/test_edge_cases.py +305 -305
  241. tests/test_encoding_core.py +56 -56
  242. tests/test_encoding_detection.py +126 -126
  243. tests/test_enhanced_error_handler.py +270 -270
  244. tests/test_enhanced_error_handler_comprehensive.py +245 -245
  245. tests/test_error_handler_compatibility.py +112 -112
  246. tests/test_factories.py +252 -252
  247. tests/test_factory_compatibility.py +196 -196
  248. tests/test_final_validation.py +153 -153
  249. tests/test_fingerprint_consistency.py +135 -135
  250. tests/test_fingerprint_simple.py +51 -51
  251. tests/test_get_component_logger.py +83 -83
  252. tests/test_hash_performance.py +99 -99
  253. tests/test_integration.py +169 -169
  254. tests/test_item_dedup_redis_key.py +122 -122
  255. tests/test_large_scale_helper.py +235 -235
  256. tests/test_logging_enhancements.py +374 -374
  257. tests/test_logging_final.py +184 -184
  258. tests/test_logging_integration.py +312 -312
  259. tests/test_logging_system.py +282 -282
  260. tests/test_middleware_debug.py +141 -141
  261. tests/test_mode_consistency.py +51 -51
  262. tests/test_multi_directory.py +67 -67
  263. tests/test_multiple_spider_modules.py +80 -80
  264. tests/test_mysql_pipeline_config.py +165 -0
  265. tests/test_mysql_pipeline_error.py +99 -0
  266. tests/test_mysql_pipeline_init_log.py +83 -0
  267. tests/test_mysql_pipeline_integration.py +133 -0
  268. tests/test_mysql_pipeline_refactor.py +144 -0
  269. tests/test_mysql_pipeline_refactor_simple.py +86 -0
  270. tests/test_mysql_pipeline_robustness.py +196 -0
  271. tests/test_mysql_pipeline_types.py +89 -0
  272. tests/test_mysql_update_columns.py +94 -0
  273. tests/test_offsite_middleware.py +244 -244
  274. tests/test_offsite_middleware_simple.py +203 -203
  275. tests/test_optimized_selector_naming.py +100 -100
  276. tests/test_parsel.py +29 -29
  277. tests/test_performance.py +327 -327
  278. tests/test_performance_monitor.py +115 -115
  279. tests/test_pipeline_fingerprint_consistency.py +86 -86
  280. tests/test_priority_behavior.py +211 -211
  281. tests/test_priority_consistency.py +151 -151
  282. tests/test_priority_consistency_fixed.py +249 -249
  283. tests/test_proxy_health_check.py +32 -32
  284. tests/test_proxy_middleware.py +217 -121
  285. tests/test_proxy_middleware_enhanced.py +212 -216
  286. tests/test_proxy_middleware_integration.py +142 -137
  287. tests/test_proxy_middleware_refactored.py +207 -184
  288. tests/test_proxy_only.py +84 -0
  289. tests/test_proxy_providers.py +56 -56
  290. tests/test_proxy_stats.py +19 -19
  291. tests/test_proxy_strategies.py +59 -59
  292. tests/test_proxy_with_downloader.py +153 -0
  293. tests/test_queue_empty_check.py +41 -41
  294. tests/test_queue_manager_double_crawlo.py +173 -173
  295. tests/test_queue_manager_redis_key.py +179 -179
  296. tests/test_queue_naming.py +154 -154
  297. tests/test_queue_type.py +106 -106
  298. tests/test_queue_type_redis_config_consistency.py +130 -130
  299. tests/test_random_headers_default.py +322 -322
  300. tests/test_random_headers_necessity.py +308 -308
  301. tests/test_random_user_agent.py +72 -72
  302. tests/test_redis_config.py +28 -28
  303. tests/test_redis_connection_pool.py +294 -294
  304. tests/test_redis_key_naming.py +181 -181
  305. tests/test_redis_key_validator.py +123 -123
  306. tests/test_redis_queue.py +224 -224
  307. tests/test_redis_queue_name_fix.py +175 -175
  308. tests/test_redis_queue_type_fallback.py +129 -129
  309. tests/test_request_ignore_middleware.py +182 -182
  310. tests/test_request_params.py +111 -111
  311. tests/test_request_serialization.py +70 -70
  312. tests/test_response_code_middleware.py +349 -349
  313. tests/test_response_filter_middleware.py +427 -427
  314. tests/test_response_follow.py +104 -104
  315. tests/test_response_improvements.py +152 -152
  316. tests/test_response_selector_methods.py +92 -92
  317. tests/test_response_url_methods.py +70 -70
  318. tests/test_response_urljoin.py +86 -86
  319. tests/test_retry_middleware.py +333 -333
  320. tests/test_retry_middleware_realistic.py +273 -273
  321. tests/test_scheduler.py +252 -252
  322. tests/test_scheduler_config_update.py +133 -133
  323. tests/test_scrapy_style_encoding.py +112 -112
  324. tests/test_selector_helper.py +100 -100
  325. tests/test_selector_optimizations.py +146 -146
  326. tests/test_simple_response.py +61 -61
  327. tests/test_spider_loader.py +49 -49
  328. tests/test_spider_loader_comprehensive.py +69 -69
  329. tests/test_spider_modules.py +84 -84
  330. tests/test_spiders/test_spider.py +9 -9
  331. tests/test_telecom_spider_redis_key.py +205 -205
  332. tests/test_template_content.py +87 -87
  333. tests/test_template_redis_key.py +134 -134
  334. tests/test_tools.py +159 -159
  335. tests/test_user_agent_randomness.py +176 -176
  336. tests/test_user_agents.py +96 -96
  337. tests/untested_features_report.md +138 -138
  338. tests/verify_debug.py +51 -51
  339. tests/verify_distributed.py +117 -117
  340. tests/verify_log_fix.py +111 -111
  341. tests/verify_mysql_warnings.py +110 -0
  342. crawlo/logging/async_handler.py +0 -181
  343. crawlo/logging/monitor.py +0 -153
  344. crawlo/logging/sampler.py +0 -167
  345. crawlo/middleware/simple_proxy.py +0 -65
  346. crawlo/tools/authenticated_proxy.py +0 -241
  347. crawlo/tools/data_formatter.py +0 -226
  348. crawlo/tools/data_validator.py +0 -181
  349. crawlo/tools/encoding_converter.py +0 -127
  350. crawlo/tools/network_diagnostic.py +0 -365
  351. crawlo/tools/request_tools.py +0 -83
  352. crawlo/tools/retry_mechanism.py +0 -224
  353. crawlo/utils/env_config.py +0 -143
  354. crawlo/utils/large_scale_config.py +0 -287
  355. crawlo/utils/system.py +0 -11
  356. crawlo/utils/tools.py +0 -5
  357. crawlo-1.4.5.dist-info/METADATA +0 -329
  358. crawlo-1.4.5.dist-info/RECORD +0 -347
  359. tests/env_config_example.py +0 -134
  360. tests/ofweek_scrapy/ofweek_scrapy/spiders/ofweek_spider.py +0 -162
  361. tests/test_authenticated_proxy.py +0 -142
  362. tests/test_comprehensive.py +0 -147
  363. tests/test_dynamic_downloaders_proxy.py +0 -125
  364. tests/test_dynamic_proxy.py +0 -93
  365. tests/test_dynamic_proxy_config.py +0 -147
  366. tests/test_dynamic_proxy_real.py +0 -110
  367. tests/test_env_config.py +0 -122
  368. tests/test_framework_env_usage.py +0 -104
  369. tests/test_large_scale_config.py +0 -113
  370. tests/test_proxy_api.py +0 -265
  371. tests/test_real_scenario_proxy.py +0 -196
  372. tests/tools_example.py +0 -261
  373. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/WHEEL +0 -0
  374. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/entry_points.txt +0 -0
  375. {crawlo-1.4.5.dist-info → crawlo-1.4.7.dist-info}/top_level.txt +0 -0
@@ -1,428 +1,428 @@
1
- #!/usr/bin/env python
2
- # -*- coding: utf-8 -*-
3
- """
4
- ResponseFilterMiddleware 测试脚本
5
- 验证ResponseFilterMiddleware的功能是否正常工作
6
- """
7
-
8
- import asyncio
9
- import sys
10
- import os
11
-
12
- # 添加项目根目录到Python路径
13
- sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
-
15
- from crawlo.middleware.response_filter import ResponseFilterMiddleware
16
- from crawlo.network.request import Request
17
- from crawlo.network.response import Response
18
- from crawlo.settings.setting_manager import SettingManager
19
- from crawlo.exceptions import IgnoreRequestError
20
-
21
-
22
- class MockStats:
23
- """模拟统计对象"""
24
- def __init__(self):
25
- self.counts = {}
26
-
27
- def inc_value(self, key, count=1):
28
- if key in self.counts:
29
- self.counts[key] += count
30
- else:
31
- self.counts[key] = count
32
-
33
-
34
- class MockLogger:
35
- """模拟日志对象"""
36
- def __init__(self):
37
- self.messages = []
38
-
39
- def debug(self, message):
40
- self.messages.append(f"DEBUG: {message}")
41
-
42
-
43
- class MockCrawler:
44
- """模拟爬虫实例"""
45
- def __init__(self, settings):
46
- self.settings = settings
47
- self.stats = MockStats()
48
- self.logger = MockLogger()
49
-
50
-
51
- def create_test_settings(allowed_codes=None, denied_codes=None):
52
- """创建测试设置"""
53
- settings = SettingManager()
54
- settings.set("LOG_LEVEL", "DEBUG")
55
-
56
- if allowed_codes is not None:
57
- # 将列表转换为字符串格式存储
58
- settings.set("ALLOWED_RESPONSE_CODES", ",".join(map(str, allowed_codes)))
59
- else:
60
- settings.set("ALLOWED_RESPONSE_CODES", "")
61
-
62
- if denied_codes is not None:
63
- # 将列表转换为字符串格式存储
64
- settings.set("DENIED_RESPONSE_CODES", ",".join(map(str, denied_codes)))
65
- else:
66
- settings.set("DENIED_RESPONSE_CODES", "")
67
-
68
- return settings
69
-
70
-
71
- async def test_response_filter_middleware():
72
- """测试ResponseFilterMiddleware功能"""
73
- print("开始测试ResponseFilterMiddleware...")
74
-
75
- # 测试用例
76
- test_cases = [
77
- # (状态码, 应该被允许, 描述)
78
- (200, True, "正常响应"),
79
- (201, True, "创建成功"),
80
- (299, True, "2xx范围内的其他状态码"),
81
- (404, False, "页面未找到"),
82
- (500, False, "服务器错误"),
83
- (403, False, "禁止访问"),
84
- (301, False, "重定向"),
85
- ]
86
-
87
- # 创建设置和中间件(不配置ALLOWED_RESPONSE_CODES,只允许2xx状态码)
88
- settings = create_test_settings(None, None)
89
- crawler = MockCrawler(settings)
90
- middleware = ResponseFilterMiddleware.create_instance(crawler)
91
-
92
- print(f"✓ ResponseFilterMiddleware创建成功")
93
- print(f" 默认允许的状态码: 200-299")
94
- print(f" 允许列表: {middleware.allowed_codes}")
95
- print(f" 拒绝列表: {middleware.denied_codes}")
96
-
97
- # 运行测试用例
98
- passed = 0
99
- total = len(test_cases)
100
-
101
- for status_code, should_allow, description in test_cases:
102
- try:
103
- request = Request(url="https://example.com")
104
- response = Response(
105
- url="https://example.com",
106
- status_code=status_code,
107
- body=b"Test response body"
108
- )
109
-
110
- result = middleware.process_response(request, response, None)
111
-
112
- # 如果到达这里,说明响应被允许
113
- if should_allow:
114
- if result == response:
115
- print(f"✓ {description}: 状态码 {status_code} (正确允许)")
116
- passed += 1
117
- else:
118
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
119
- else:
120
- print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
121
-
122
- except IgnoreRequestError as e:
123
- # 响应被过滤
124
- if not should_allow:
125
- print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
126
- passed += 1
127
- else:
128
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
129
- except Exception as e:
130
- print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
131
-
132
- # 输出测试结果
133
- print(f"\n测试结果: {passed}/{total} 通过")
134
- if passed == total:
135
- print("✓ 所有测试通过,ResponseFilterMiddleware工作正常")
136
- return True
137
- else:
138
- print("✗ 部分测试失败,ResponseFilterMiddleware存在问题")
139
- return False
140
-
141
-
142
- async def test_allowed_codes_configuration():
143
- """测试ALLOWED_RESPONSE_CODES配置功能"""
144
- print("\n测试ALLOWED_RESPONSE_CODES配置功能...")
145
-
146
- # 配置允许的状态码
147
- allowed_codes = [404, 500, 301]
148
- settings = create_test_settings(allowed_codes, None)
149
- crawler = MockCrawler(settings)
150
- middleware = ResponseFilterMiddleware.create_instance(crawler)
151
-
152
- print(f"✓ ResponseFilterMiddleware创建成功")
153
- print(f" 配置的允许状态码: {allowed_codes}")
154
- print(f" 实际允许列表: {middleware.allowed_codes}")
155
-
156
- # 测试用例
157
- test_cases = [
158
- # (状态码, 应该被允许, 描述)
159
- (200, True, "正常响应(2xx范围)"),
160
- (404, True, "页面未找到(配置允许)"),
161
- (500, True, "服务器错误(配置允许)"),
162
- (301, True, "重定向(配置允许)"),
163
- (403, False, "禁止访问(未配置允许)"),
164
- (502, False, "网关错误(未配置允许)"),
165
- ]
166
-
167
- passed = 0
168
- total = len(test_cases)
169
-
170
- for status_code, should_allow, description in test_cases:
171
- try:
172
- request = Request(url="https://example.com")
173
- response = Response(
174
- url="https://example.com",
175
- status_code=status_code,
176
- body=b"Test response body"
177
- )
178
-
179
- result = middleware.process_response(request, response, None)
180
-
181
- # 如果到达这里,说明响应被允许
182
- if should_allow:
183
- if result == response:
184
- print(f"✓ {description}: 状态码 {status_code} (正确允许)")
185
- passed += 1
186
- else:
187
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
188
- else:
189
- print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
190
-
191
- except IgnoreRequestError as e:
192
- # 响应被过滤
193
- if not should_allow:
194
- print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
195
- passed += 1
196
- else:
197
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
198
- except Exception as e:
199
- print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
200
-
201
- # 输出测试结果
202
- print(f"\n测试结果: {passed}/{total} 通过")
203
- if passed == total:
204
- print("✓ 所有测试通过,ALLOWED_RESPONSE_CODES配置功能正常")
205
- return True
206
- else:
207
- print("✗ 部分测试失败,ALLOWED_RESPONSE_CODES配置功能存在问题")
208
- return False
209
-
210
-
211
- async def test_denied_codes_configuration():
212
- """测试DENIED_RESPONSE_CODES配置功能"""
213
- print("\n测试DENIED_RESPONSE_CODES配置功能...")
214
-
215
- # 配置拒绝的状态码(注意:这里需要将整数转换为字符串,因为SettingManager.get_list会处理字符串)
216
- denied_codes = ["200", "201"]
217
- settings = create_test_settings(None, denied_codes)
218
- crawler = MockCrawler(settings)
219
- middleware = ResponseFilterMiddleware.create_instance(crawler)
220
-
221
- print(f"✓ ResponseFilterMiddleware创建成功")
222
- print(f" 配置的拒绝状态码: {denied_codes}")
223
- print(f" 实际拒绝列表: {middleware.denied_codes}")
224
-
225
- # 测试用例
226
- test_cases = [
227
- # (状态码, 应该被允许, 描述)
228
- (200, False, "正常响应(配置拒绝)"),
229
- (201, False, "创建成功(配置拒绝)"),
230
- (404, False, "页面未找到(默认拒绝)"),
231
- (202, True, "接受请求(2xx范围,未被拒绝)"),
232
- (301, False, "重定向(默认拒绝)"),
233
- (500, False, "服务器错误(默认拒绝)"),
234
- ]
235
-
236
- passed = 0
237
- total = len(test_cases)
238
-
239
- for status_code, should_allow, description in test_cases:
240
- try:
241
- request = Request(url="https://example.com")
242
- response = Response(
243
- url="https://example.com",
244
- status_code=status_code,
245
- body=b"Test response body"
246
- )
247
-
248
- result = middleware.process_response(request, response, None)
249
-
250
- # 如果到达这里,说明响应被允许
251
- if should_allow:
252
- if result == response:
253
- print(f"✓ {description}: 状态码 {status_code} (正确允许)")
254
- passed += 1
255
- else:
256
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
257
- else:
258
- print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
259
-
260
- except IgnoreRequestError as e:
261
- # 响应被过滤
262
- if not should_allow:
263
- print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
264
- passed += 1
265
- else:
266
- print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
267
- except Exception as e:
268
- print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
269
-
270
- # 输出测试结果
271
- print(f"\n测试结果: {passed}/{total} 通过")
272
- if passed == total:
273
- print("✓ 所有测试通过,DENIED_RESPONSE_CODES配置功能正常")
274
- return True
275
- else:
276
- print("✗ 部分测试失败,DENIED_RESPONSE_CODES配置功能存在问题")
277
- return False
278
-
279
-
280
- def test_middleware_creation():
281
- """测试中间件创建功能"""
282
- print("\n测试中间件创建功能...")
283
-
284
- try:
285
- settings = create_test_settings(None, None)
286
- crawler = MockCrawler(settings)
287
- middleware = ResponseFilterMiddleware.create_instance(crawler)
288
-
289
- if middleware and hasattr(middleware, 'allowed_codes') and hasattr(middleware, 'denied_codes') and hasattr(middleware, 'logger'):
290
- print("✓ 中间件创建成功,属性完整")
291
- print(f" allowed_codes: {middleware.allowed_codes}")
292
- print(f" denied_codes: {middleware.denied_codes}")
293
- return True
294
- else:
295
- print("✗ 中间件创建失败,缺少必要属性")
296
- return False
297
- except Exception as e:
298
- print(f"✗ 中间件创建时出现异常: {e}")
299
- return False
300
-
301
-
302
- async def test_filter_reason_function():
303
- """测试过滤原因功能"""
304
- print("\n测试过滤原因功能...")
305
-
306
- try:
307
- # 配置允许和拒绝的状态码(注意:这里需要将整数转换为字符串)
308
- allowed_codes = ["404"]
309
- denied_codes = ["200"]
310
- settings = create_test_settings(allowed_codes, denied_codes)
311
- crawler = MockCrawler(settings)
312
- middleware = ResponseFilterMiddleware.create_instance(crawler)
313
-
314
- # 测试拒绝状态码的原因
315
- reason1 = middleware._get_filter_reason(200)
316
- if "被明确拒绝" in reason1:
317
- print("✓ 拒绝状态码原因正确")
318
- else:
319
- print(f"✗ 拒绝状态码原因错误: {reason1}")
320
-
321
- # 测试未允许状态码的原因
322
- reason2 = middleware._get_filter_reason(500)
323
- if "不在允许列表中" in reason2:
324
- print("✓ 未允许状态码原因正确")
325
- else:
326
- print(f"✗ 未允许状态码原因错误: {reason2}")
327
-
328
- print("✓ 过滤原因功能正常")
329
- return True
330
-
331
- except Exception as e:
332
- print(f"✗ 过滤原因测试时出现异常: {e}")
333
- return False
334
-
335
-
336
- async def test_is_response_allowed_function():
337
- """测试响应允许检查功能"""
338
- print("\n测试响应允许检查功能...")
339
-
340
- try:
341
- # 配置允许和拒绝的状态码(注意:这里需要将整数转换为字符串)
342
- allowed_codes = ["404", "301"]
343
- denied_codes = ["200"]
344
- settings = create_test_settings(allowed_codes, denied_codes)
345
- crawler = MockCrawler(settings)
346
- middleware = ResponseFilterMiddleware.create_instance(crawler)
347
-
348
- # 创建模拟响应对象
349
- class MockResponse:
350
- def __init__(self, status_code):
351
- self.status_code = status_code
352
-
353
- # 测试用例: (状态码, 应该被允许, 描述)
354
- test_cases = [
355
- (200, False, "被明确拒绝的状态码"),
356
- (404, True, "被明确允许的状态码"),
357
- (301, True, "被明确允许的状态码"),
358
- (201, True, "2xx范围内的默认允许状态码"),
359
- (500, False, "未被允许的错误状态码"),
360
- ]
361
-
362
- passed = 0
363
- total = len(test_cases)
364
-
365
- for status_code, should_allow, description in test_cases:
366
- response = MockResponse(status_code)
367
- is_allowed = middleware._is_response_allowed(response)
368
-
369
- if is_allowed == should_allow:
370
- print(f"✓ {description}: 状态码 {status_code}")
371
- passed += 1
372
- else:
373
- print(f"✗ {description}: 状态码 {status_code} (期望: {should_allow}, 实际: {is_allowed})")
374
-
375
- print(f"\n允许检查测试结果: {passed}/{total} 通过")
376
- if passed == total:
377
- print("✓ 响应允许检查功能正常")
378
- return True
379
- else:
380
- print("✗ 响应允许检查功能存在问题")
381
- return False
382
-
383
- except Exception as e:
384
- print(f"✗ 响应允许检查测试时出现异常: {e}")
385
- return False
386
-
387
-
388
- async def main():
389
- """主测试函数"""
390
- print("ResponseFilterMiddleware 功能测试")
391
- print("=" * 50)
392
-
393
- # 测试正常功能
394
- test1_result = await test_response_filter_middleware()
395
-
396
- # 测试ALLOWED_RESPONSE_CODES配置
397
- test2_result = await test_allowed_codes_configuration()
398
-
399
- # 测试DENIED_RESPONSE_CODES配置
400
- test3_result = await test_denied_codes_configuration()
401
-
402
- # 测试创建功能
403
- test4_result = test_middleware_creation()
404
-
405
- # 测试过滤原因功能
406
- test5_result = await test_filter_reason_function()
407
-
408
- # 测试响应允许检查功能
409
- test6_result = await test_is_response_allowed_function()
410
-
411
- # 汇总结果
412
- print("\n" + "=" * 50)
413
- print("测试结果汇总:")
414
- print(f"基本功能测试: {'✓ 通过' if test1_result else '✗ 失败'}")
415
- print(f"允许配置测试: {'✓ 通过' if test2_result else '✗ 失败'}")
416
- print(f"拒绝配置测试: {'✓ 通过' if test3_result else '✗ 失败'}")
417
- print(f"创建测试: {'✓ 通过' if test4_result else '✗ 失败'}")
418
- print(f"原因功能测试: {'✓ 通过' if test5_result else '✗ 失败'}")
419
- print(f"允许检查测试: {'✓ 通过' if test6_result else '✗ 失败'}")
420
-
421
- overall_result = test1_result and test2_result and test3_result and test4_result and test5_result and test6_result
422
- print(f"\n总体结果: {'✓ 所有测试通过' if overall_result else '✗ 部分测试失败'}")
423
-
424
- return overall_result
425
-
426
-
427
- if __name__ == "__main__":
1
+ #!/usr/bin/env python
2
+ # -*- coding: utf-8 -*-
3
+ """
4
+ ResponseFilterMiddleware 测试脚本
5
+ 验证ResponseFilterMiddleware的功能是否正常工作
6
+ """
7
+
8
+ import asyncio
9
+ import sys
10
+ import os
11
+
12
+ # 添加项目根目录到Python路径
13
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
14
+
15
+ from crawlo.middleware.response_filter import ResponseFilterMiddleware
16
+ from crawlo.network.request import Request
17
+ from crawlo.network.response import Response
18
+ from crawlo.settings.setting_manager import SettingManager
19
+ from crawlo.exceptions import IgnoreRequestError
20
+
21
+
22
+ class MockStats:
23
+ """模拟统计对象"""
24
+ def __init__(self):
25
+ self.counts = {}
26
+
27
+ def inc_value(self, key, count=1):
28
+ if key in self.counts:
29
+ self.counts[key] += count
30
+ else:
31
+ self.counts[key] = count
32
+
33
+
34
+ class MockLogger:
35
+ """模拟日志对象"""
36
+ def __init__(self):
37
+ self.messages = []
38
+
39
+ def debug(self, message):
40
+ self.messages.append(f"DEBUG: {message}")
41
+
42
+
43
+ class MockCrawler:
44
+ """模拟爬虫实例"""
45
+ def __init__(self, settings):
46
+ self.settings = settings
47
+ self.stats = MockStats()
48
+ self.logger = MockLogger()
49
+
50
+
51
+ def create_test_settings(allowed_codes=None, denied_codes=None):
52
+ """创建测试设置"""
53
+ settings = SettingManager()
54
+ settings.set("LOG_LEVEL", "DEBUG")
55
+
56
+ if allowed_codes is not None:
57
+ # 将列表转换为字符串格式存储
58
+ settings.set("ALLOWED_RESPONSE_CODES", ",".join(map(str, allowed_codes)))
59
+ else:
60
+ settings.set("ALLOWED_RESPONSE_CODES", "")
61
+
62
+ if denied_codes is not None:
63
+ # 将列表转换为字符串格式存储
64
+ settings.set("DENIED_RESPONSE_CODES", ",".join(map(str, denied_codes)))
65
+ else:
66
+ settings.set("DENIED_RESPONSE_CODES", "")
67
+
68
+ return settings
69
+
70
+
71
+ async def test_response_filter_middleware():
72
+ """测试ResponseFilterMiddleware功能"""
73
+ print("开始测试ResponseFilterMiddleware...")
74
+
75
+ # 测试用例
76
+ test_cases = [
77
+ # (状态码, 应该被允许, 描述)
78
+ (200, True, "正常响应"),
79
+ (201, True, "创建成功"),
80
+ (299, True, "2xx范围内的其他状态码"),
81
+ (404, False, "页面未找到"),
82
+ (500, False, "服务器错误"),
83
+ (403, False, "禁止访问"),
84
+ (301, False, "重定向"),
85
+ ]
86
+
87
+ # 创建设置和中间件(不配置ALLOWED_RESPONSE_CODES,只允许2xx状态码)
88
+ settings = create_test_settings(None, None)
89
+ crawler = MockCrawler(settings)
90
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
91
+
92
+ print(f"✓ ResponseFilterMiddleware创建成功")
93
+ print(f" 默认允许的状态码: 200-299")
94
+ print(f" 允许列表: {middleware.allowed_codes}")
95
+ print(f" 拒绝列表: {middleware.denied_codes}")
96
+
97
+ # 运行测试用例
98
+ passed = 0
99
+ total = len(test_cases)
100
+
101
+ for status_code, should_allow, description in test_cases:
102
+ try:
103
+ request = Request(url="https://example.com")
104
+ response = Response(
105
+ url="https://example.com",
106
+ status_code=status_code,
107
+ body=b"Test response body"
108
+ )
109
+
110
+ result = middleware.process_response(request, response, None)
111
+
112
+ # 如果到达这里,说明响应被允许
113
+ if should_allow:
114
+ if result == response:
115
+ print(f"✓ {description}: 状态码 {status_code} (正确允许)")
116
+ passed += 1
117
+ else:
118
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
119
+ else:
120
+ print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
121
+
122
+ except IgnoreRequestError as e:
123
+ # 响应被过滤
124
+ if not should_allow:
125
+ print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
126
+ passed += 1
127
+ else:
128
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
129
+ except Exception as e:
130
+ print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
131
+
132
+ # 输出测试结果
133
+ print(f"\n测试结果: {passed}/{total} 通过")
134
+ if passed == total:
135
+ print("✓ 所有测试通过,ResponseFilterMiddleware工作正常")
136
+ return True
137
+ else:
138
+ print("✗ 部分测试失败,ResponseFilterMiddleware存在问题")
139
+ return False
140
+
141
+
142
+ async def test_allowed_codes_configuration():
143
+ """测试ALLOWED_RESPONSE_CODES配置功能"""
144
+ print("\n测试ALLOWED_RESPONSE_CODES配置功能...")
145
+
146
+ # 配置允许的状态码
147
+ allowed_codes = [404, 500, 301]
148
+ settings = create_test_settings(allowed_codes, None)
149
+ crawler = MockCrawler(settings)
150
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
151
+
152
+ print(f"✓ ResponseFilterMiddleware创建成功")
153
+ print(f" 配置的允许状态码: {allowed_codes}")
154
+ print(f" 实际允许列表: {middleware.allowed_codes}")
155
+
156
+ # 测试用例
157
+ test_cases = [
158
+ # (状态码, 应该被允许, 描述)
159
+ (200, True, "正常响应(2xx范围)"),
160
+ (404, True, "页面未找到(配置允许)"),
161
+ (500, True, "服务器错误(配置允许)"),
162
+ (301, True, "重定向(配置允许)"),
163
+ (403, False, "禁止访问(未配置允许)"),
164
+ (502, False, "网关错误(未配置允许)"),
165
+ ]
166
+
167
+ passed = 0
168
+ total = len(test_cases)
169
+
170
+ for status_code, should_allow, description in test_cases:
171
+ try:
172
+ request = Request(url="https://example.com")
173
+ response = Response(
174
+ url="https://example.com",
175
+ status_code=status_code,
176
+ body=b"Test response body"
177
+ )
178
+
179
+ result = middleware.process_response(request, response, None)
180
+
181
+ # 如果到达这里,说明响应被允许
182
+ if should_allow:
183
+ if result == response:
184
+ print(f"✓ {description}: 状态码 {status_code} (正确允许)")
185
+ passed += 1
186
+ else:
187
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
188
+ else:
189
+ print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
190
+
191
+ except IgnoreRequestError as e:
192
+ # 响应被过滤
193
+ if not should_allow:
194
+ print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
195
+ passed += 1
196
+ else:
197
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
198
+ except Exception as e:
199
+ print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
200
+
201
+ # 输出测试结果
202
+ print(f"\n测试结果: {passed}/{total} 通过")
203
+ if passed == total:
204
+ print("✓ 所有测试通过,ALLOWED_RESPONSE_CODES配置功能正常")
205
+ return True
206
+ else:
207
+ print("✗ 部分测试失败,ALLOWED_RESPONSE_CODES配置功能存在问题")
208
+ return False
209
+
210
+
211
+ async def test_denied_codes_configuration():
212
+ """测试DENIED_RESPONSE_CODES配置功能"""
213
+ print("\n测试DENIED_RESPONSE_CODES配置功能...")
214
+
215
+ # 配置拒绝的状态码(注意:这里需要将整数转换为字符串,因为SettingManager.get_list会处理字符串)
216
+ denied_codes = ["200", "201"]
217
+ settings = create_test_settings(None, denied_codes)
218
+ crawler = MockCrawler(settings)
219
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
220
+
221
+ print(f"✓ ResponseFilterMiddleware创建成功")
222
+ print(f" 配置的拒绝状态码: {denied_codes}")
223
+ print(f" 实际拒绝列表: {middleware.denied_codes}")
224
+
225
+ # 测试用例
226
+ test_cases = [
227
+ # (状态码, 应该被允许, 描述)
228
+ (200, False, "正常响应(配置拒绝)"),
229
+ (201, False, "创建成功(配置拒绝)"),
230
+ (404, False, "页面未找到(默认拒绝)"),
231
+ (202, True, "接受请求(2xx范围,未被拒绝)"),
232
+ (301, False, "重定向(默认拒绝)"),
233
+ (500, False, "服务器错误(默认拒绝)"),
234
+ ]
235
+
236
+ passed = 0
237
+ total = len(test_cases)
238
+
239
+ for status_code, should_allow, description in test_cases:
240
+ try:
241
+ request = Request(url="https://example.com")
242
+ response = Response(
243
+ url="https://example.com",
244
+ status_code=status_code,
245
+ body=b"Test response body"
246
+ )
247
+
248
+ result = middleware.process_response(request, response, None)
249
+
250
+ # 如果到达这里,说明响应被允许
251
+ if should_allow:
252
+ if result == response:
253
+ print(f"✓ {description}: 状态码 {status_code} (正确允许)")
254
+ passed += 1
255
+ else:
256
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但返回结果不正确)")
257
+ else:
258
+ print(f"✗ {description}: 状态码 {status_code} (应该被过滤但被允许)")
259
+
260
+ except IgnoreRequestError as e:
261
+ # 响应被过滤
262
+ if not should_allow:
263
+ print(f"✓ {description}: 状态码 {status_code} (正确过滤)")
264
+ passed += 1
265
+ else:
266
+ print(f"✗ {description}: 状态码 {status_code} (应该被允许但被过滤)")
267
+ except Exception as e:
268
+ print(f"✗ {description}: 状态码 {status_code} (出现异常: {e})")
269
+
270
+ # 输出测试结果
271
+ print(f"\n测试结果: {passed}/{total} 通过")
272
+ if passed == total:
273
+ print("✓ 所有测试通过,DENIED_RESPONSE_CODES配置功能正常")
274
+ return True
275
+ else:
276
+ print("✗ 部分测试失败,DENIED_RESPONSE_CODES配置功能存在问题")
277
+ return False
278
+
279
+
280
+ def test_middleware_creation():
281
+ """测试中间件创建功能"""
282
+ print("\n测试中间件创建功能...")
283
+
284
+ try:
285
+ settings = create_test_settings(None, None)
286
+ crawler = MockCrawler(settings)
287
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
288
+
289
+ if middleware and hasattr(middleware, 'allowed_codes') and hasattr(middleware, 'denied_codes') and hasattr(middleware, 'logger'):
290
+ print("✓ 中间件创建成功,属性完整")
291
+ print(f" allowed_codes: {middleware.allowed_codes}")
292
+ print(f" denied_codes: {middleware.denied_codes}")
293
+ return True
294
+ else:
295
+ print("✗ 中间件创建失败,缺少必要属性")
296
+ return False
297
+ except Exception as e:
298
+ print(f"✗ 中间件创建时出现异常: {e}")
299
+ return False
300
+
301
+
302
+ async def test_filter_reason_function():
303
+ """测试过滤原因功能"""
304
+ print("\n测试过滤原因功能...")
305
+
306
+ try:
307
+ # 配置允许和拒绝的状态码(注意:这里需要将整数转换为字符串)
308
+ allowed_codes = ["404"]
309
+ denied_codes = ["200"]
310
+ settings = create_test_settings(allowed_codes, denied_codes)
311
+ crawler = MockCrawler(settings)
312
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
313
+
314
+ # 测试拒绝状态码的原因
315
+ reason1 = middleware._get_filter_reason(200)
316
+ if "被明确拒绝" in reason1:
317
+ print("✓ 拒绝状态码原因正确")
318
+ else:
319
+ print(f"✗ 拒绝状态码原因错误: {reason1}")
320
+
321
+ # 测试未允许状态码的原因
322
+ reason2 = middleware._get_filter_reason(500)
323
+ if "不在允许列表中" in reason2:
324
+ print("✓ 未允许状态码原因正确")
325
+ else:
326
+ print(f"✗ 未允许状态码原因错误: {reason2}")
327
+
328
+ print("✓ 过滤原因功能正常")
329
+ return True
330
+
331
+ except Exception as e:
332
+ print(f"✗ 过滤原因测试时出现异常: {e}")
333
+ return False
334
+
335
+
336
+ async def test_is_response_allowed_function():
337
+ """测试响应允许检查功能"""
338
+ print("\n测试响应允许检查功能...")
339
+
340
+ try:
341
+ # 配置允许和拒绝的状态码(注意:这里需要将整数转换为字符串)
342
+ allowed_codes = ["404", "301"]
343
+ denied_codes = ["200"]
344
+ settings = create_test_settings(allowed_codes, denied_codes)
345
+ crawler = MockCrawler(settings)
346
+ middleware = ResponseFilterMiddleware.create_instance(crawler)
347
+
348
+ # 创建模拟响应对象
349
+ class MockResponse:
350
+ def __init__(self, status_code):
351
+ self.status_code = status_code
352
+
353
+ # 测试用例: (状态码, 应该被允许, 描述)
354
+ test_cases = [
355
+ (200, False, "被明确拒绝的状态码"),
356
+ (404, True, "被明确允许的状态码"),
357
+ (301, True, "被明确允许的状态码"),
358
+ (201, True, "2xx范围内的默认允许状态码"),
359
+ (500, False, "未被允许的错误状态码"),
360
+ ]
361
+
362
+ passed = 0
363
+ total = len(test_cases)
364
+
365
+ for status_code, should_allow, description in test_cases:
366
+ response = MockResponse(status_code)
367
+ is_allowed = middleware._is_response_allowed(response)
368
+
369
+ if is_allowed == should_allow:
370
+ print(f"✓ {description}: 状态码 {status_code}")
371
+ passed += 1
372
+ else:
373
+ print(f"✗ {description}: 状态码 {status_code} (期望: {should_allow}, 实际: {is_allowed})")
374
+
375
+ print(f"\n允许检查测试结果: {passed}/{total} 通过")
376
+ if passed == total:
377
+ print("✓ 响应允许检查功能正常")
378
+ return True
379
+ else:
380
+ print("✗ 响应允许检查功能存在问题")
381
+ return False
382
+
383
+ except Exception as e:
384
+ print(f"✗ 响应允许检查测试时出现异常: {e}")
385
+ return False
386
+
387
+
388
+ async def main():
389
+ """主测试函数"""
390
+ print("ResponseFilterMiddleware 功能测试")
391
+ print("=" * 50)
392
+
393
+ # 测试正常功能
394
+ test1_result = await test_response_filter_middleware()
395
+
396
+ # 测试ALLOWED_RESPONSE_CODES配置
397
+ test2_result = await test_allowed_codes_configuration()
398
+
399
+ # 测试DENIED_RESPONSE_CODES配置
400
+ test3_result = await test_denied_codes_configuration()
401
+
402
+ # 测试创建功能
403
+ test4_result = test_middleware_creation()
404
+
405
+ # 测试过滤原因功能
406
+ test5_result = await test_filter_reason_function()
407
+
408
+ # 测试响应允许检查功能
409
+ test6_result = await test_is_response_allowed_function()
410
+
411
+ # 汇总结果
412
+ print("\n" + "=" * 50)
413
+ print("测试结果汇总:")
414
+ print(f"基本功能测试: {'✓ 通过' if test1_result else '✗ 失败'}")
415
+ print(f"允许配置测试: {'✓ 通过' if test2_result else '✗ 失败'}")
416
+ print(f"拒绝配置测试: {'✓ 通过' if test3_result else '✗ 失败'}")
417
+ print(f"创建测试: {'✓ 通过' if test4_result else '✗ 失败'}")
418
+ print(f"原因功能测试: {'✓ 通过' if test5_result else '✗ 失败'}")
419
+ print(f"允许检查测试: {'✓ 通过' if test6_result else '✗ 失败'}")
420
+
421
+ overall_result = test1_result and test2_result and test3_result and test4_result and test5_result and test6_result
422
+ print(f"\n总体结果: {'✓ 所有测试通过' if overall_result else '✗ 部分测试失败'}")
423
+
424
+ return overall_result
425
+
426
+
427
+ if __name__ == "__main__":
428
428
  asyncio.run(main())