crawlo 1.1.9__py3-none-any.whl → 1.2.1__py3-none-any.whl

tests/test_item_dedup_redis_key.py

@@ -1,123 +1,123 @@
-#!/usr/bin/env python3
-# -*- coding: utf-8 -*-
-"""
-数据项去重Redis Key测试脚本
-用于验证RedisDedupPipeline和示例项目中的Redis去重管道是否使用统一的Redis key命名规范
-"""
-import sys
-import os
-import traceback
-
-# 添加项目根目录到路径
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
-
-from crawlo.pipelines.redis_dedup_pipeline import RedisDedupPipeline
-
-
-class MockSettings:
-    """模拟设置类"""
-    def __init__(self, project_name="test_project"):
-        self.project_name = project_name
-    
-    def get(self, key, default=None):
-        if key == 'PROJECT_NAME':
-            return self.project_name
-        elif key == 'REDIS_HOST':
-            return 'localhost'
-        elif key == 'REDIS_PORT':
-            return 6379
-        elif key == 'REDIS_DB':
-            return 2
-        elif key == 'REDIS_PASSWORD':
-            return None
-        elif key == 'LOG_LEVEL':
-            return 'INFO'
-        return default
-    
-    def getint(self, key, default=0):
-        if key == 'REDIS_PORT':
-            return 6379
-        elif key == 'REDIS_DB':
-            return 2
-        return default
-
-
-class MockCrawler:
-    """模拟爬虫类"""
-    def __init__(self, project_name="test_project"):
-        self.settings = MockSettings(project_name)
-
-
-async def test_item_dedup_redis_key():
-    """测试数据项去重Redis key命名规范"""
-    print("🔍 测试数据项去重Redis key命名规范...")
-    
-    try:
-        # 测试不同的项目名称
-        test_cases = [
-            {
-                "project_name": "books_distributed",
-                "expected_key": "crawlo:books_distributed:item:fingerprint",
-                "description": "书籍分布式项目"
-            },
-            {
-                "project_name": "api_data_collection",
-                "expected_key": "crawlo:api_data_collection:item:fingerprint",
-                "description": "API数据采集项目"
-            },
-            {
-                "project_name": "test_project",
-                "expected_key": "crawlo:test_project:item:fingerprint",
-                "description": "测试项目"
-            }
-        ]
-        
-        for i, test_case in enumerate(test_cases, 1):
-            print(f"   {i}. 测试 {test_case['description']}...")
-            
-            # 测试RedisDedupPipeline
-            mock_crawler = MockCrawler(test_case["project_name"])
-            pipeline = RedisDedupPipeline.from_crawler(mock_crawler)
-            
-            # 验证Redis key是否符合规范
-            assert pipeline.redis_key == test_case["expected_key"], \
-                f"Redis key不匹配: {pipeline.redis_key} != {test_case['expected_key']}"
-            
-            print(f"      ✅ Redis key: {pipeline.redis_key}")
-        
-        print("✅ 数据项去重Redis key命名规范测试通过！")
-        return True
-        
-    except Exception as e:
-        print(f"❌ 数据项去重Redis key命名规范测试失败: {e}")
-        traceback.print_exc()
-        return False
-
-
-def main():
-    """主测试函数"""
-    print("🚀 开始数据项去重Redis key命名规范测试...")
-    print("=" * 50)
-    
-    try:
-        success = test_item_dedup_redis_key()
-        
-        print("=" * 50)
-        if success:
-            print("🎉 所有测试通过！数据项去重使用统一的Redis key命名规范")
-        else:
-            print("❌ 测试失败，请检查实现")
-            return 1
-            
-    except Exception as e:
-        print("=" * 50)
-        print(f"❌ 测试过程中发生异常: {e}")
-        traceback.print_exc()
-        return 1
-    
-    return 0
-
-
-if __name__ == "__main__":
-    exit_code = main()
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+"""
+数据项去重Redis Key测试脚本
+用于验证RedisDedupPipeline和示例项目中的Redis去重管道是否使用统一的Redis key命名规范
+"""
+import sys
+import os
+import traceback
+
+# 添加项目根目录到路径
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
+
+from crawlo.pipelines.redis_dedup_pipeline import RedisDedupPipeline
+
+
+class MockSettings:
+    """模拟设置类"""
+    def __init__(self, project_name="test_project"):
+        self.project_name = project_name
+    
+    def get(self, key, default=None):
+        if key == 'PROJECT_NAME':
+            return self.project_name
+        elif key == 'REDIS_HOST':
+            return 'localhost'
+        elif key == 'REDIS_PORT':
+            return 6379
+        elif key == 'REDIS_DB':
+            return 2
+        elif key == 'REDIS_PASSWORD':
+            return None
+        elif key == 'LOG_LEVEL':
+            return 'INFO'
+        return default
+    
+    def getint(self, key, default=0):
+        if key == 'REDIS_PORT':
+            return 6379
+        elif key == 'REDIS_DB':
+            return 2
+        return default
+
+
+class MockCrawler:
+    """模拟爬虫类"""
+    def __init__(self, project_name="test_project"):
+        self.settings = MockSettings(project_name)
+
+
+async def test_item_dedup_redis_key():
+    """测试数据项去重Redis key命名规范"""
+    print("🔍 测试数据项去重Redis key命名规范...")
+    
+    try:
+        # 测试不同的项目名称
+        test_cases = [
+            {
+                "project_name": "books_distributed",
+                "expected_key": "crawlo:books_distributed:item:fingerprint",
+                "description": "书籍分布式项目"
+            },
+            {
+                "project_name": "api_data_collection",
+                "expected_key": "crawlo:api_data_collection:item:fingerprint",
+                "description": "API数据采集项目"
+            },
+            {
+                "project_name": "test_project",
+                "expected_key": "crawlo:test_project:item:fingerprint",
+                "description": "测试项目"
+            }
+        ]
+        
+        for i, test_case in enumerate(test_cases, 1):
+            print(f"   {i}. 测试 {test_case['description']}...")
+            
+            # 测试RedisDedupPipeline
+            mock_crawler = MockCrawler(test_case["project_name"])
+            pipeline = RedisDedupPipeline.from_crawler(mock_crawler)
+            
+            # 验证Redis key是否符合规范
+            assert pipeline.redis_key == test_case["expected_key"], \
+                f"Redis key不匹配: {pipeline.redis_key} != {test_case['expected_key']}"
+            
+            print(f"      ✅ Redis key: {pipeline.redis_key}")
+        
+        print("✅ 数据项去重Redis key命名规范测试通过！")
+        return True
+        
+    except Exception as e:
+        print(f"❌ 数据项去重Redis key命名规范测试失败: {e}")
+        traceback.print_exc()
+        return False
+
+
+def main():
+    """主测试函数"""
+    print("🚀 开始数据项去重Redis key命名规范测试...")
+    print("=" * 50)
+    
+    try:
+        success = test_item_dedup_redis_key()
+        
+        print("=" * 50)
+        if success:
+            print("🎉 所有测试通过！数据项去重使用统一的Redis key命名规范")
+        else:
+            print("❌ 测试失败，请检查实现")
+            return 1
+            
+    except Exception as e:
+        print("=" * 50)
+        print(f"❌ 测试过程中发生异常: {e}")
+        traceback.print_exc()
+        return 1
+    
+    return 0
+
+
+if __name__ == "__main__":
+    exit_code = main()
     sys.exit(exit_code)

tests/test_offsite_middleware.py

@@ -0,0 +1,222 @@
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+OffsiteMiddleware 测试文件
+用于测试站外请求过滤中间件的功能
+"""
+
+import asyncio
+import unittest
+from unittest.mock import Mock, patch
+
+from crawlo.middleware.offsite import OffsiteMiddleware
+from crawlo.exceptions import NotConfiguredError, IgnoreRequestError
+from crawlo.settings.setting_manager import SettingManager
+
+
+class MockLogger:
+    """Mock Logger 类，用于测试日志输出"""
+    def __init__(self, name, level=None):
+        self.name = name
+        self.level = level
+        self.logs = []
+
+    def debug(self, msg):
+        self.logs.append(('debug', msg))
+
+    def info(self, msg):
+        self.logs.append(('info', msg))
+
+    def warning(self, msg):
+        self.logs.append(('warning', msg))
+
+    def error(self, msg):
+        self.logs.append(('error', msg))
+
+
+class MockStats:
+    """Mock Stats 类，用于测试统计信息"""
+    def __init__(self):
+        self.stats = {}
+
+    def inc_value(self, key, value=1):
+        if key in self.stats:
+            self.stats[key] += value
+        else:
+            self.stats[key] = value
+
+
+class TestOffsiteMiddleware(unittest.TestCase):
+    """OffsiteMiddleware 测试类"""
+
+    def setUp(self):
+        """测试前准备"""
+        # 创建设置管理器
+        self.settings = SettingManager()
+        
+        # 创建爬虫模拟对象
+        self.crawler = Mock()
+        self.crawler.settings = self.settings
+        self.crawler.stats = MockStats()
+        self.crawler.logger = Mock()
+
+    @patch('crawlo.utils.log.get_logger')
+    def test_middleware_initialization_without_allowed_domains(self, mock_get_logger):
+        """测试没有配置允许域名时中间件初始化"""
+        mock_get_logger.return_value = MockLogger('OffsiteMiddleware')
+        
+        # 应该抛出NotConfiguredError异常
+        with self.assertRaises(NotConfiguredError):
+            OffsiteMiddleware.create_instance(self.crawler)
+
+    @patch('crawlo.utils.log.get_logger')
+    def test_middleware_initialization_with_allowed_domains(self, mock_get_logger):
+        """测试配置允许域名时中间件初始化"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com', 'test.com'])
+        self.settings.set('LOG_LEVEL', 'INFO')
+        
+        mock_get_logger.return_value = MockLogger('OffsiteMiddleware')
+        
+        # 应该正常创建实例
+        middleware = OffsiteMiddleware.create_instance(self.crawler)
+        self.assertIsInstance(middleware, OffsiteMiddleware)
+        self.assertEqual(len(middleware.allowed_domains), 2)
+        self.assertIn('example.com', middleware.allowed_domains)
+        self.assertIn('test.com', middleware.allowed_domains)
+
+    def test_is_offsite_request_with_valid_domain(self):
+        """测试有效域名的站外请求判断"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'INFO')
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware(
+            stats=MockStats(),
+            log_level='INFO',
+            allowed_domains=['example.com']
+        )
+        middleware._compile_domains()
+        
+        # 创建请求对象
+        request = Mock()
+        request.url = 'http://example.com/page'
+        
+        # 应该不是站外请求
+        self.assertFalse(middleware._is_offsite_request(request))
+
+    def test_is_offsite_request_with_subdomain(self):
+        """测试子域名的站外请求判断"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'INFO')
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware(
+            stats=MockStats(),
+            log_level='INFO',
+            allowed_domains=['example.com']
+        )
+        middleware._compile_domains()
+        
+        # 创建请求对象（子域名）
+        request = Mock()
+        request.url = 'http://sub.example.com/page'
+        
+        # 应该不是站外请求（子域名应该被允许）
+        self.assertFalse(middleware._is_offsite_request(request))
+
+    def test_is_offsite_request_with_invalid_domain(self):
+        """测试无效域名的站外请求判断"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'INFO')
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware(
+            stats=MockStats(),
+            log_level='INFO',
+            allowed_domains=['example.com']
+        )
+        middleware._compile_domains()
+        
+        # 创建请求对象
+        request = Mock()
+        request.url = 'http://other.com/page'
+        
+        # 应该是站外请求
+        self.assertTrue(middleware._is_offsite_request(request))
+
+    def test_is_offsite_request_with_invalid_url(self):
+        """测试无效URL的站外请求判断"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'INFO')
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware(
+            stats=MockStats(),
+            log_level='INFO',
+            allowed_domains=['example.com']
+        )
+        middleware._compile_domains()
+        
+        # 创建请求对象（无效URL）
+        request = Mock()
+        request.url = 'invalid-url'
+        
+        # 应该是站外请求
+        self.assertTrue(middleware._is_offsite_request(request))
+
+    @patch('crawlo.utils.log.get_logger')
+    def test_process_request_with_offsite_request(self, mock_get_logger):
+        """测试处理站外请求"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'DEBUG')
+        
+        mock_logger = MockLogger('OffsiteMiddleware')
+        mock_get_logger.return_value = mock_logger
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware.create_instance(self.crawler)
+        
+        # 创建请求对象（站外请求）
+        request = Mock()
+        request.url = 'http://other.com/page'
+        
+        # 应该抛出IgnoreRequestError异常
+        with self.assertRaises(IgnoreRequestError):
+            asyncio.run(middleware.process_request(request, Mock()))
+        
+        # 验证统计信息
+        self.assertIn('offsite_request_count', self.crawler.stats.stats)
+        self.assertEqual(self.crawler.stats.stats['offsite_request_count'], 1)
+
+    @patch('crawlo.utils.log.get_logger')
+    def test_process_request_with_valid_request(self, mock_get_logger):
+        """测试处理有效请求"""
+        # 设置允许的域名
+        self.settings.set('ALLOWED_DOMAINS', ['example.com'])
+        self.settings.set('LOG_LEVEL', 'DEBUG')
+        
+        mock_logger = MockLogger('OffsiteMiddleware')
+        mock_get_logger.return_value = mock_logger
+        
+        # 创建中间件实例
+        middleware = OffsiteMiddleware.create_instance(self.crawler)
+        
+        # 创建请求对象（有效请求）
+        request = Mock()
+        request.url = 'http://example.com/page'
+        
+        # 应该正常处理，不抛出异常
+        result = asyncio.run(middleware.process_request(request, Mock()))
+        
+        # 返回None表示继续处理
+        self.assertIsNone(result)
+
+
+if __name__ == '__main__':
+    unittest.main()

tests/test_parsel.py

@@ -1,30 +1,30 @@
-#!/usr/bin/python
-# -*- coding:UTF-8 -*-
-"""
-parsel 库测试
-"""
-import sys
-import os
-
-# 添加项目根目录到路径
-sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
-
-try:
-    from parsel import Selector, SelectorList
-    print("parsel 导入成功")
-    
-    # 测试基本功能
-    html = "<html><body><h1>测试标题</h1></body></html>"
-    selector = Selector(html)
-    print("Selector 创建成功")
-    
-    elements = selector.css('h1')
-    print("CSS 选择器执行成功")
-    
-    text = elements.get()
-    print(f"获取文本: {text}")
-    
-except Exception as e:
-    print(f"错误: {e}")
-    import traceback
+#!/usr/bin/python
+# -*- coding:UTF-8 -*-
+"""
+parsel 库测试
+"""
+import sys
+import os
+
+# 添加项目根目录到路径
+sys.path.insert(0, os.path.join(os.path.dirname(__file__), '..'))
+
+try:
+    from parsel import Selector, SelectorList
+    print("parsel 导入成功")
+    
+    # 测试基本功能
+    html = "<html><body><h1>测试标题</h1></body></html>"
+    selector = Selector(html)
+    print("Selector 创建成功")
+    
+    elements = selector.css('h1')
+    print("CSS 选择器执行成功")
+    
+    text = elements.get()
+    print(f"获取文本: {text}")
+    
+except Exception as e:
+    print(f"错误: {e}")
+    import traceback
     traceback.print_exc()