crawlo 1.1.9__py3-none-any.whl → 1.2.1__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 (221) hide show
  1. crawlo/__init__.py +61 -61
  2. crawlo/__version__.py +1 -1
  3. crawlo/cleaners/__init__.py +60 -60
  4. crawlo/cleaners/data_formatter.py +225 -225
  5. crawlo/cleaners/encoding_converter.py +125 -125
  6. crawlo/cleaners/text_cleaner.py +232 -232
  7. crawlo/cli.py +65 -65
  8. crawlo/commands/__init__.py +14 -14
  9. crawlo/commands/check.py +594 -594
  10. crawlo/commands/genspider.py +151 -151
  11. crawlo/commands/help.py +142 -132
  12. crawlo/commands/list.py +155 -155
  13. crawlo/commands/run.py +292 -292
  14. crawlo/commands/startproject.py +418 -418
  15. crawlo/commands/stats.py +188 -188
  16. crawlo/commands/utils.py +186 -186
  17. crawlo/config.py +312 -312
  18. crawlo/config_validator.py +252 -252
  19. crawlo/core/__init__.py +2 -2
  20. crawlo/core/engine.py +354 -345
  21. crawlo/core/processor.py +40 -40
  22. crawlo/core/scheduler.py +143 -136
  23. crawlo/crawler.py +1027 -1027
  24. crawlo/downloader/__init__.py +266 -266
  25. crawlo/downloader/aiohttp_downloader.py +220 -220
  26. crawlo/downloader/cffi_downloader.py +256 -256
  27. crawlo/downloader/httpx_downloader.py +259 -259
  28. crawlo/downloader/hybrid_downloader.py +213 -213
  29. crawlo/downloader/playwright_downloader.py +402 -402
  30. crawlo/downloader/selenium_downloader.py +472 -472
  31. crawlo/event.py +11 -11
  32. crawlo/exceptions.py +81 -81
  33. crawlo/extension/__init__.py +37 -37
  34. crawlo/extension/health_check.py +141 -141
  35. crawlo/extension/log_interval.py +57 -57
  36. crawlo/extension/log_stats.py +81 -81
  37. crawlo/extension/logging_extension.py +43 -43
  38. crawlo/extension/memory_monitor.py +104 -104
  39. crawlo/extension/performance_profiler.py +133 -133
  40. crawlo/extension/request_recorder.py +107 -107
  41. crawlo/filters/__init__.py +154 -154
  42. crawlo/filters/aioredis_filter.py +280 -280
  43. crawlo/filters/memory_filter.py +269 -269
  44. crawlo/items/__init__.py +23 -23
  45. crawlo/items/base.py +21 -21
  46. crawlo/items/fields.py +53 -53
  47. crawlo/items/items.py +104 -104
  48. crawlo/middleware/__init__.py +21 -21
  49. crawlo/middleware/default_header.py +132 -32
  50. crawlo/middleware/download_delay.py +105 -28
  51. crawlo/middleware/middleware_manager.py +135 -135
  52. crawlo/middleware/offsite.py +116 -0
  53. crawlo/middleware/proxy.py +366 -272
  54. crawlo/middleware/request_ignore.py +88 -30
  55. crawlo/middleware/response_code.py +164 -18
  56. crawlo/middleware/response_filter.py +138 -26
  57. crawlo/middleware/retry.py +124 -124
  58. crawlo/mode_manager.py +211 -211
  59. crawlo/network/__init__.py +21 -21
  60. crawlo/network/request.py +338 -338
  61. crawlo/network/response.py +359 -359
  62. crawlo/pipelines/__init__.py +21 -21
  63. crawlo/pipelines/bloom_dedup_pipeline.py +156 -156
  64. crawlo/pipelines/console_pipeline.py +39 -39
  65. crawlo/pipelines/csv_pipeline.py +316 -316
  66. crawlo/pipelines/database_dedup_pipeline.py +224 -224
  67. crawlo/pipelines/json_pipeline.py +218 -218
  68. crawlo/pipelines/memory_dedup_pipeline.py +115 -115
  69. crawlo/pipelines/mongo_pipeline.py +131 -131
  70. crawlo/pipelines/mysql_pipeline.py +316 -316
  71. crawlo/pipelines/pipeline_manager.py +61 -61
  72. crawlo/pipelines/redis_dedup_pipeline.py +167 -167
  73. crawlo/project.py +187 -187
  74. crawlo/queue/pqueue.py +37 -37
  75. crawlo/queue/queue_manager.py +337 -334
  76. crawlo/queue/redis_priority_queue.py +298 -298
  77. crawlo/settings/__init__.py +7 -7
  78. crawlo/settings/default_settings.py +226 -219
  79. crawlo/settings/setting_manager.py +122 -122
  80. crawlo/spider/__init__.py +639 -639
  81. crawlo/stats_collector.py +59 -59
  82. crawlo/subscriber.py +130 -130
  83. crawlo/task_manager.py +30 -30
  84. crawlo/templates/crawlo.cfg.tmpl +10 -10
  85. crawlo/templates/project/__init__.py.tmpl +3 -3
  86. crawlo/templates/project/items.py.tmpl +17 -17
  87. crawlo/templates/project/middlewares.py.tmpl +118 -109
  88. crawlo/templates/project/pipelines.py.tmpl +96 -96
  89. crawlo/templates/project/run.py.tmpl +45 -45
  90. crawlo/templates/project/settings.py.tmpl +327 -326
  91. crawlo/templates/project/settings_distributed.py.tmpl +119 -119
  92. crawlo/templates/project/settings_gentle.py.tmpl +94 -94
  93. crawlo/templates/project/settings_high_performance.py.tmpl +151 -151
  94. crawlo/templates/project/settings_simple.py.tmpl +68 -68
  95. crawlo/templates/project/spiders/__init__.py.tmpl +5 -5
  96. crawlo/templates/spider/spider.py.tmpl +143 -141
  97. crawlo/tools/__init__.py +182 -182
  98. crawlo/tools/anti_crawler.py +268 -268
  99. crawlo/tools/authenticated_proxy.py +240 -240
  100. crawlo/tools/data_validator.py +180 -180
  101. crawlo/tools/date_tools.py +35 -35
  102. crawlo/tools/distributed_coordinator.py +386 -386
  103. crawlo/tools/retry_mechanism.py +220 -220
  104. crawlo/tools/scenario_adapter.py +262 -262
  105. crawlo/utils/__init__.py +35 -35
  106. crawlo/utils/batch_processor.py +260 -260
  107. crawlo/utils/controlled_spider_mixin.py +439 -439
  108. crawlo/utils/date_tools.py +290 -290
  109. crawlo/utils/db_helper.py +343 -343
  110. crawlo/utils/enhanced_error_handler.py +359 -359
  111. crawlo/utils/env_config.py +105 -105
  112. crawlo/utils/error_handler.py +125 -125
  113. crawlo/utils/func_tools.py +82 -82
  114. crawlo/utils/large_scale_config.py +286 -286
  115. crawlo/utils/large_scale_helper.py +343 -343
  116. crawlo/utils/log.py +128 -128
  117. crawlo/utils/performance_monitor.py +284 -284
  118. crawlo/utils/queue_helper.py +175 -175
  119. crawlo/utils/redis_connection_pool.py +334 -334
  120. crawlo/utils/redis_key_validator.py +199 -199
  121. crawlo/utils/request.py +267 -267
  122. crawlo/utils/request_serializer.py +219 -219
  123. crawlo/utils/spider_loader.py +62 -62
  124. crawlo/utils/system.py +11 -11
  125. crawlo/utils/tools.py +4 -4
  126. crawlo/utils/url.py +39 -39
  127. crawlo-1.2.1.dist-info/METADATA +692 -0
  128. crawlo-1.2.1.dist-info/RECORD +220 -0
  129. examples/__init__.py +7 -7
  130. examples/aiohttp_settings.py +42 -0
  131. examples/curl_cffi_settings.py +41 -0
  132. examples/default_header_middleware_example.py +107 -0
  133. examples/default_header_spider_example.py +129 -0
  134. examples/download_delay_middleware_example.py +160 -0
  135. examples/httpx_settings.py +42 -0
  136. examples/multi_downloader_proxy_example.py +81 -0
  137. examples/offsite_middleware_example.py +55 -0
  138. examples/offsite_spider_example.py +107 -0
  139. examples/proxy_spider_example.py +166 -0
  140. examples/request_ignore_middleware_example.py +51 -0
  141. examples/request_ignore_spider_example.py +99 -0
  142. examples/response_code_middleware_example.py +52 -0
  143. examples/response_filter_middleware_example.py +67 -0
  144. examples/tong_hua_shun_settings.py +62 -0
  145. examples/tong_hua_shun_spider.py +170 -0
  146. tests/DOUBLE_CRAWLO_PREFIX_FIX_REPORT.md +81 -81
  147. tests/__init__.py +7 -7
  148. tests/advanced_tools_example.py +275 -275
  149. tests/authenticated_proxy_example.py +236 -236
  150. tests/cleaners_example.py +160 -160
  151. tests/config_validation_demo.py +102 -102
  152. tests/controlled_spider_example.py +205 -205
  153. tests/date_tools_example.py +180 -180
  154. tests/dynamic_loading_example.py +523 -523
  155. tests/dynamic_loading_test.py +104 -104
  156. tests/env_config_example.py +133 -133
  157. tests/error_handling_example.py +171 -171
  158. tests/redis_key_validation_demo.py +130 -130
  159. tests/response_improvements_example.py +144 -144
  160. tests/test_advanced_tools.py +148 -148
  161. tests/test_all_redis_key_configs.py +145 -145
  162. tests/test_authenticated_proxy.py +141 -141
  163. tests/test_cleaners.py +54 -54
  164. tests/test_comprehensive.py +146 -146
  165. tests/test_config_validator.py +193 -193
  166. tests/test_crawlo_proxy_integration.py +173 -0
  167. tests/test_date_tools.py +123 -123
  168. tests/test_default_header_middleware.py +159 -0
  169. tests/test_double_crawlo_fix.py +207 -207
  170. tests/test_double_crawlo_fix_simple.py +124 -124
  171. tests/test_download_delay_middleware.py +222 -0
  172. tests/test_downloader_proxy_compatibility.py +269 -0
  173. tests/test_dynamic_downloaders_proxy.py +124 -124
  174. tests/test_dynamic_proxy.py +92 -92
  175. tests/test_dynamic_proxy_config.py +146 -146
  176. tests/test_dynamic_proxy_real.py +109 -109
  177. tests/test_edge_cases.py +303 -303
  178. tests/test_enhanced_error_handler.py +270 -270
  179. tests/test_env_config.py +121 -121
  180. tests/test_error_handler_compatibility.py +112 -112
  181. tests/test_final_validation.py +153 -153
  182. tests/test_framework_env_usage.py +103 -103
  183. tests/test_integration.py +356 -356
  184. tests/test_item_dedup_redis_key.py +122 -122
  185. tests/test_offsite_middleware.py +222 -0
  186. tests/test_parsel.py +29 -29
  187. tests/test_performance.py +327 -327
  188. tests/test_proxy_api.py +265 -0
  189. tests/test_proxy_health_check.py +32 -32
  190. tests/test_proxy_middleware.py +122 -0
  191. tests/test_proxy_middleware_enhanced.py +217 -0
  192. tests/test_proxy_middleware_integration.py +136 -136
  193. tests/test_proxy_providers.py +56 -56
  194. tests/test_proxy_stats.py +19 -19
  195. tests/test_proxy_strategies.py +59 -59
  196. tests/test_queue_manager_double_crawlo.py +174 -231
  197. tests/test_queue_manager_redis_key.py +176 -176
  198. tests/test_real_scenario_proxy.py +196 -0
  199. tests/test_redis_config.py +28 -28
  200. tests/test_redis_connection_pool.py +294 -294
  201. tests/test_redis_key_naming.py +181 -181
  202. tests/test_redis_key_validator.py +123 -123
  203. tests/test_redis_queue.py +224 -224
  204. tests/test_request_ignore_middleware.py +183 -0
  205. tests/test_request_serialization.py +70 -70
  206. tests/test_response_code_middleware.py +350 -0
  207. tests/test_response_filter_middleware.py +428 -0
  208. tests/test_response_improvements.py +152 -152
  209. tests/test_retry_middleware.py +242 -0
  210. tests/test_scheduler.py +241 -241
  211. tests/test_simple_response.py +61 -61
  212. tests/test_telecom_spider_redis_key.py +205 -205
  213. tests/test_template_content.py +87 -87
  214. tests/test_template_redis_key.py +134 -134
  215. tests/test_tools.py +153 -153
  216. tests/tools_example.py +257 -257
  217. crawlo-1.1.9.dist-info/METADATA +0 -626
  218. crawlo-1.1.9.dist-info/RECORD +0 -190
  219. {crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/WHEEL +0 -0
  220. {crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/entry_points.txt +0 -0
  221. {crawlo-1.1.9.dist-info → crawlo-1.2.1.dist-info}/top_level.txt +0 -0
tests/test_cleaners.py CHANGED
@@ -1,55 +1,55 @@
1
- #!/usr/bin/python
2
- # -*- coding: UTF-8 -*-
3
- """
4
- 数据清洗工具测试
5
- """
6
- import unittest
7
- from crawlo.cleaners import (
8
- TextCleaner,
9
- DataFormatter,
10
- remove_html_tags,
11
- decode_html_entities,
12
- clean_text,
13
- format_number,
14
- format_currency,
15
- format_phone_number
16
- )
17
-
18
-
19
- class TestCleaners(unittest.TestCase):
20
- """数据清洗工具测试类"""
21
-
22
- def test_text_cleaner(self):
23
- """测试文本清洗功能"""
24
- # 测试移除HTML标签
25
- html_text = "<p>这是一个<b>测试</b>文本</p>"
26
- clean_text_result = remove_html_tags(html_text)
27
- self.assertEqual(clean_text_result, "这是一个测试文本")
28
-
29
- # 测试解码HTML实体
30
- entity_text = "这是一个&nbsp;测试&amp;文本"
31
- decoded_text = decode_html_entities(entity_text)
32
- self.assertEqual(decoded_text, "这是一个 测试&文本")
33
-
34
- # 测试综合清洗
35
- complex_text = "<p>这是一个&nbsp;<b>测试</b>&amp;文本</p>"
36
- cleaned = clean_text(complex_text)
37
- self.assertEqual(cleaned, "这是一个 测试&文本")
38
-
39
- def test_data_formatter(self):
40
- """测试数据格式化功能"""
41
- # 测试数字格式化
42
- formatted_num = format_number(1234.567, precision=2, thousand_separator=True)
43
- self.assertEqual(formatted_num, "1,234.57")
44
-
45
- # 测试货币格式化
46
- formatted_currency = format_currency(1234.567, "¥", 2)
47
- self.assertEqual(formatted_currency, "¥1,234.57")
48
-
49
- # 测试电话号码格式化
50
- formatted_phone = format_phone_number("13812345678", "+86", "international")
51
- self.assertEqual(formatted_phone, "+86 138 1234 5678")
52
-
53
-
54
- if __name__ == '__main__':
1
+ #!/usr/bin/python
2
+ # -*- coding: UTF-8 -*-
3
+ """
4
+ 数据清洗工具测试
5
+ """
6
+ import unittest
7
+ from crawlo.cleaners import (
8
+ TextCleaner,
9
+ DataFormatter,
10
+ remove_html_tags,
11
+ decode_html_entities,
12
+ clean_text,
13
+ format_number,
14
+ format_currency,
15
+ format_phone_number
16
+ )
17
+
18
+
19
+ class TestCleaners(unittest.TestCase):
20
+ """数据清洗工具测试类"""
21
+
22
+ def test_text_cleaner(self):
23
+ """测试文本清洗功能"""
24
+ # 测试移除HTML标签
25
+ html_text = "<p>这是一个<b>测试</b>文本</p>"
26
+ clean_text_result = remove_html_tags(html_text)
27
+ self.assertEqual(clean_text_result, "这是一个测试文本")
28
+
29
+ # 测试解码HTML实体
30
+ entity_text = "这是一个&nbsp;测试&amp;文本"
31
+ decoded_text = decode_html_entities(entity_text)
32
+ self.assertEqual(decoded_text, "这是一个 测试&文本")
33
+
34
+ # 测试综合清洗
35
+ complex_text = "<p>这是一个&nbsp;<b>测试</b>&amp;文本</p>"
36
+ cleaned = clean_text(complex_text)
37
+ self.assertEqual(cleaned, "这是一个 测试&文本")
38
+
39
+ def test_data_formatter(self):
40
+ """测试数据格式化功能"""
41
+ # 测试数字格式化
42
+ formatted_num = format_number(1234.567, precision=2, thousand_separator=True)
43
+ self.assertEqual(formatted_num, "1,234.57")
44
+
45
+ # 测试货币格式化
46
+ formatted_currency = format_currency(1234.567, "¥", 2)
47
+ self.assertEqual(formatted_currency, "¥1,234.57")
48
+
49
+ # 测试电话号码格式化
50
+ formatted_phone = format_phone_number("13812345678", "+86", "international")
51
+ self.assertEqual(formatted_phone, "+86 138 1234 5678")
52
+
53
+
54
+ if __name__ == '__main__':
55
55
  unittest.main()
@@ -1,147 +1,147 @@
1
- #!/usr/bin/env python3
2
- # -*- coding: utf-8 -*-
3
- """
4
- 综合测试
5
- 验证所有改进的集成效果
6
- """
7
- import sys
8
- import os
9
- import asyncio
10
- import unittest
11
- from unittest.mock import patch, MagicMock
12
-
13
- # 添加项目根目录到Python路径
14
- sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
15
-
16
- from crawlo.utils.env_config import get_env_var, get_redis_config, get_runtime_config
17
- from crawlo.utils.error_handler import ErrorHandler, handle_exception
18
- from crawlo.core.engine import Engine
19
- from crawlo.settings.setting_manager import SettingManager
20
- from crawlo.settings import default_settings
21
- from crawlo.queue.queue_manager import QueueManager, QueueConfig, QueueType
22
-
23
-
24
- class TestComprehensiveIntegration(unittest.TestCase):
25
- """综合集成测试"""
26
-
27
- def setUp(self):
28
- """测试前准备"""
29
- # 设置测试环境变量
30
- self.test_env = {
31
- 'PROJECT_NAME': 'test_project',
32
- 'CONCURRENCY': '4',
33
- 'REDIS_HOST': 'localhost',
34
- 'REDIS_PORT': '6379'
35
- }
36
- self.original_env = {}
37
- for key, value in self.test_env.items():
38
- self.original_env[key] = os.environ.get(key)
39
- os.environ[key] = value
40
-
41
- def tearDown(self):
42
- """测试后清理"""
43
- # 恢复原始环境变量
44
- for key, value in self.original_env.items():
45
- if value is None:
46
- os.environ.pop(key, None)
47
- else:
48
- os.environ[key] = value
49
-
50
- def test_env_config_integration(self):
51
- """测试环境变量配置集成"""
52
- # 验证环境变量工具正常工作
53
- project_name = get_env_var('PROJECT_NAME', 'default', str)
54
- self.assertEqual(project_name, 'test_project')
55
-
56
- concurrency = get_env_var('CONCURRENCY', 1, int)
57
- self.assertEqual(concurrency, 4)
58
-
59
- # 验证Redis配置工具
60
- redis_config = get_redis_config()
61
- self.assertEqual(redis_config['REDIS_HOST'], 'localhost')
62
- self.assertEqual(redis_config['REDIS_PORT'], 6379)
63
-
64
- def test_error_handler_integration(self):
65
- """测试错误处理集成"""
66
- # 验证错误处理模块正常工作
67
- error_handler = ErrorHandler("test")
68
-
69
- # 测试错误处理
70
- try:
71
- error_handler.handle_error(ValueError("Test error"), raise_error=False)
72
- except Exception:
73
- self.fail("Error handler should not raise exception when raise_error=False")
74
-
75
- # 测试安全调用
76
- result = error_handler.safe_call(lambda x: x*2, 5, default_return=0)
77
- self.assertEqual(result, 10)
78
-
79
- # 测试装饰器
80
- @handle_exception(raise_error=False)
81
- def failing_function():
82
- raise RuntimeError("Test")
83
-
84
- try:
85
- failing_function()
86
- except Exception:
87
- self.fail("Decorated function should not raise exception")
88
-
89
- def test_settings_integration(self):
90
- """测试设置管理器集成"""
91
- # 重新加载默认设置以获取环境变量
92
- import importlib
93
- import crawlo.settings.default_settings
94
- importlib.reload(crawlo.settings.default_settings)
95
-
96
- # 创建设置管理器
97
- settings = SettingManager()
98
- settings.set_settings(crawlo.settings.default_settings)
99
-
100
- # 验证设置正确加载
101
- self.assertEqual(settings.get('PROJECT_NAME'), 'test_project')
102
- self.assertEqual(settings.get_int('CONCURRENCY'), 4)
103
- self.assertEqual(settings.get('REDIS_HOST'), 'localhost')
104
-
105
- def test_queue_manager_config(self):
106
- """测试队列管理器配置"""
107
- # 重新加载默认设置
108
- import importlib
109
- import crawlo.settings.default_settings
110
- importlib.reload(crawlo.settings.default_settings)
111
-
112
- # 创建设置管理器
113
- settings = SettingManager()
114
- settings.set_settings(crawlo.settings.default_settings)
115
-
116
- # 从设置创建队列配置
117
- queue_config = QueueConfig.from_settings(settings)
118
-
119
- # 验证配置正确
120
- self.assertEqual(queue_config.queue_type, QueueType.AUTO)
121
- self.assertIn('test_project', queue_config.queue_name)
122
-
123
- async def test_async_components(self):
124
- """测试异步组件"""
125
- # 测试异步错误处理装饰器
126
- @handle_exception(raise_error=False)
127
- async def async_failing_function():
128
- raise RuntimeError("Async test")
129
-
130
- try:
131
- await async_failing_function()
132
- except Exception:
133
- self.fail("Async decorated function should not raise exception")
134
-
135
-
136
- if __name__ == '__main__':
137
- # 运行同步测试
138
- unittest.main(exit=False)
139
-
140
- # 运行异步测试
141
- async def run_async_tests():
142
- test_instance = TestComprehensiveIntegration()
143
- test_instance.setUp()
144
- await test_instance.test_async_components()
145
- test_instance.tearDown()
146
-
1
+ #!/usr/bin/env python3
2
+ # -*- coding: utf-8 -*-
3
+ """
4
+ 综合测试
5
+ 验证所有改进的集成效果
6
+ """
7
+ import sys
8
+ import os
9
+ import asyncio
10
+ import unittest
11
+ from unittest.mock import patch, MagicMock
12
+
13
+ # 添加项目根目录到Python路径
14
+ sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
15
+
16
+ from crawlo.utils.env_config import get_env_var, get_redis_config, get_runtime_config
17
+ from crawlo.utils.error_handler import ErrorHandler, handle_exception
18
+ from crawlo.core.engine import Engine
19
+ from crawlo.settings.setting_manager import SettingManager
20
+ from crawlo.settings import default_settings
21
+ from crawlo.queue.queue_manager import QueueManager, QueueConfig, QueueType
22
+
23
+
24
+ class TestComprehensiveIntegration(unittest.TestCase):
25
+ """综合集成测试"""
26
+
27
+ def setUp(self):
28
+ """测试前准备"""
29
+ # 设置测试环境变量
30
+ self.test_env = {
31
+ 'PROJECT_NAME': 'test_project',
32
+ 'CONCURRENCY': '4',
33
+ 'REDIS_HOST': 'localhost',
34
+ 'REDIS_PORT': '6379'
35
+ }
36
+ self.original_env = {}
37
+ for key, value in self.test_env.items():
38
+ self.original_env[key] = os.environ.get(key)
39
+ os.environ[key] = value
40
+
41
+ def tearDown(self):
42
+ """测试后清理"""
43
+ # 恢复原始环境变量
44
+ for key, value in self.original_env.items():
45
+ if value is None:
46
+ os.environ.pop(key, None)
47
+ else:
48
+ os.environ[key] = value
49
+
50
+ def test_env_config_integration(self):
51
+ """测试环境变量配置集成"""
52
+ # 验证环境变量工具正常工作
53
+ project_name = get_env_var('PROJECT_NAME', 'default', str)
54
+ self.assertEqual(project_name, 'test_project')
55
+
56
+ concurrency = get_env_var('CONCURRENCY', 1, int)
57
+ self.assertEqual(concurrency, 4)
58
+
59
+ # 验证Redis配置工具
60
+ redis_config = get_redis_config()
61
+ self.assertEqual(redis_config['REDIS_HOST'], 'localhost')
62
+ self.assertEqual(redis_config['REDIS_PORT'], 6379)
63
+
64
+ def test_error_handler_integration(self):
65
+ """测试错误处理集成"""
66
+ # 验证错误处理模块正常工作
67
+ error_handler = ErrorHandler("test")
68
+
69
+ # 测试错误处理
70
+ try:
71
+ error_handler.handle_error(ValueError("Test error"), raise_error=False)
72
+ except Exception:
73
+ self.fail("Error handler should not raise exception when raise_error=False")
74
+
75
+ # 测试安全调用
76
+ result = error_handler.safe_call(lambda x: x*2, 5, default_return=0)
77
+ self.assertEqual(result, 10)
78
+
79
+ # 测试装饰器
80
+ @handle_exception(raise_error=False)
81
+ def failing_function():
82
+ raise RuntimeError("Test")
83
+
84
+ try:
85
+ failing_function()
86
+ except Exception:
87
+ self.fail("Decorated function should not raise exception")
88
+
89
+ def test_settings_integration(self):
90
+ """测试设置管理器集成"""
91
+ # 重新加载默认设置以获取环境变量
92
+ import importlib
93
+ import crawlo.settings.default_settings
94
+ importlib.reload(crawlo.settings.default_settings)
95
+
96
+ # 创建设置管理器
97
+ settings = SettingManager()
98
+ settings.set_settings(crawlo.settings.default_settings)
99
+
100
+ # 验证设置正确加载
101
+ self.assertEqual(settings.get('PROJECT_NAME'), 'test_project')
102
+ self.assertEqual(settings.get_int('CONCURRENCY'), 4)
103
+ self.assertEqual(settings.get('REDIS_HOST'), 'localhost')
104
+
105
+ def test_queue_manager_config(self):
106
+ """测试队列管理器配置"""
107
+ # 重新加载默认设置
108
+ import importlib
109
+ import crawlo.settings.default_settings
110
+ importlib.reload(crawlo.settings.default_settings)
111
+
112
+ # 创建设置管理器
113
+ settings = SettingManager()
114
+ settings.set_settings(crawlo.settings.default_settings)
115
+
116
+ # 从设置创建队列配置
117
+ queue_config = QueueConfig.from_settings(settings)
118
+
119
+ # 验证配置正确
120
+ self.assertEqual(queue_config.queue_type, QueueType.AUTO)
121
+ self.assertIn('test_project', queue_config.queue_name)
122
+
123
+ async def test_async_components(self):
124
+ """测试异步组件"""
125
+ # 测试异步错误处理装饰器
126
+ @handle_exception(raise_error=False)
127
+ async def async_failing_function():
128
+ raise RuntimeError("Async test")
129
+
130
+ try:
131
+ await async_failing_function()
132
+ except Exception:
133
+ self.fail("Async decorated function should not raise exception")
134
+
135
+
136
+ if __name__ == '__main__':
137
+ # 运行同步测试
138
+ unittest.main(exit=False)
139
+
140
+ # 运行异步测试
141
+ async def run_async_tests():
142
+ test_instance = TestComprehensiveIntegration()
143
+ test_instance.setUp()
144
+ await test_instance.test_async_components()
145
+ test_instance.tearDown()
146
+
147
147
  asyncio.run(run_async_tests())