skylos 1.1.12__py3-none-any.whl → 1.2.0__py3-none-any.whl

test/test_framework_aware.py

@@ -0,0 +1,372 @@
+#!/usr/bin/env python3
+import pytest
+import ast
+from pathlib import Path
+from unittest.mock import Mock, patch
+import fnmatch
+
+try:
+    from skylos.framework_aware import (
+        FrameworkAwareVisitor, 
+        detect_framework_usage,
+        FRAMEWORK_DECORATORS,
+        FRAMEWORK_FUNCTIONS,
+        FRAMEWORK_IMPORTS
+    )
+except ImportError:
+    import sys
+    from pathlib import Path
+    sys.path.insert(0, str(Path(__file__).parent.parent))
+    from skylos.framework_aware import (
+        FrameworkAwareVisitor, 
+        detect_framework_usage,
+        FRAMEWORK_DECORATORS,
+        FRAMEWORK_FUNCTIONS,
+        FRAMEWORK_IMPORTS
+    )
+
+class TestFrameworkAwareVisitor:
+    
+    def test_init_default(self):
+        """Test default initialization"""
+        visitor = FrameworkAwareVisitor()
+        assert visitor.is_framework_file == False
+        assert visitor.framework_decorated_lines == set()
+        assert visitor.detected_frameworks == set()
+    
+    def test_flask_import_detection(self):
+        """Test Flask import detection"""
+        code = """
+import flask
+from flask import Flask, request
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 'flask' in visitor.detected_frameworks
+    
+    def test_fastapi_import_detection(self):
+        """Test FastAPI import detection"""
+        code = """
+from fastapi import FastAPI
+import fastapi
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 'fastapi' in visitor.detected_frameworks
+    
+    def test_django_import_detection(self):
+        """Test Django import detection"""
+        code = """
+from django.http import HttpResponse
+from django.views import View
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 'django' in visitor.detected_frameworks
+    
+    def test_flask_route_decorator_detection(self):
+        """Test Flask route decorator detection"""
+        code = """
+@app.route('/api/users')
+def get_users():
+    return []
+
+@app.post('/api/users')
+def create_user():
+    return {}
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 3 in visitor.framework_decorated_lines  # get_users function
+        assert 7 in visitor.framework_decorated_lines  # create_user function
+    
+    def test_fastapi_router_decorator_detection(self):
+        """Test FastAPI router decorator detection"""
+        code = """
+@router.get('/items')
+async def read_items():
+    return []
+
+@router.post('/items')
+async def create_item():
+    return {}
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 3 in visitor.framework_decorated_lines
+        assert 7 in visitor.framework_decorated_lines
+    
+    def test_django_decorator_detection(self):
+        """Test Django decorator detection"""
+        code = """
+@login_required
+def protected_view(request):
+    return HttpResponse("Protected")
+
+@permission_required('auth.add_user')
+def admin_view(request):
+    return HttpResponse("Admin")
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 3 in visitor.framework_decorated_lines
+        assert 7 in visitor.framework_decorated_lines
+    
+    def test_celery_task_decorator_detection(self):
+        """Test Celery task decorator detection"""
+        code = """
+@task
+def background_task():
+    return "done"
+
+@shared_task
+def shared_background_task():
+    return "shared done"
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 3 in visitor.framework_decorated_lines
+        assert 7 in visitor.framework_decorated_lines
+    
+    def test_django_view_class_detection(self):
+        """Test Django view class detection"""
+        code = """
+from django import views
+
+class UserView(View):
+    def get(self, request):
+        return HttpResponse("GET")
+
+class UserViewSet(ViewSet):
+    def list(self, request):
+        return Response([])
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 4 in visitor.framework_decorated_lines
+        assert 8 in visitor.framework_decorated_lines
+    
+    def test_django_model_methods_in_framework_file(self):
+        code = """
+from django.db import models
+
+class User(models.Model):
+    name = models.CharField(max_length=100)
+    
+    def save(self, *args, **kwargs):
+        # custom save logic
+        super().save(*args, **kwargs)
+    
+    def delete(self, *args, **kwargs):
+        # custom delete logic
+        super().delete(*args, **kwargs)
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 7 in visitor.framework_decorated_lines  # save method
+        assert 11 in visitor.framework_decorated_lines  # delete method
+    
+    def test_framework_functions_not_detected_in_non_framework_file(self):
+        code = """
+def save(self):
+    # regular save method, not Django
+    pass
+
+def get(self):
+    # regular get method, not Django view
+    pass
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == False
+        assert visitor.framework_decorated_lines == set()
+    
+    def test_multiple_decorators(self):
+        """Test function with multiple decorators"""
+        code = """
+@app.route('/users')
+@login_required
+@cache.cached(timeout=60)
+def get_users():
+    return []
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 5 in visitor.framework_decorated_lines
+    
+    def test_complex_decorator_patterns(self):
+        """Test complex decorator patterns"""
+        code = """
+@app.route('/api/v1/users/<int:user_id>', methods=['GET', 'POST'])
+def user_endpoint(user_id):
+    return {}
+
+@router.get('/items/{item_id}')
+async def get_item(item_id: int):
+    return {}
+"""
+        tree = ast.parse(code)
+        visitor = FrameworkAwareVisitor()
+        visitor.visit(tree)
+        
+        assert visitor.is_framework_file == True
+        assert 3 in visitor.framework_decorated_lines
+        assert 7 in visitor.framework_decorated_lines
+    
+    @patch('skylos.framework_aware.Path')
+    def test_file_content_framework_detection(self, mock_path):
+        """Test framework detection from file content"""
+        mock_file = Mock()
+        mock_file.read_text.return_value = "from flask import Flask\napp = Flask(__name__)"
+        mock_path.return_value = mock_file
+        
+        visitor = FrameworkAwareVisitor(filename="test.py")
+        
+        assert visitor.is_framework_file == True
+        assert 'flask' in visitor.detected_frameworks
+    
+    def test_normalize_decorator_name(self):
+        visitor = FrameworkAwareVisitor()
+        
+        node = ast.parse("@decorator\ndef func(): pass").body[0].decorator_list[0]
+        result = visitor._normalize_decorator(node)
+        assert result == "@decorator"
+        
+        node = ast.parse("@app.route\ndef func(): pass").body[0].decorator_list[0]
+        result = visitor._normalize_decorator(node)
+        assert result == "@app.route"
+
+class TestDetectFrameworkUsage:
+    
+    def test_framework_decorated_function_low_confidence(self):
+        mock_def = Mock()
+        mock_def.line = 10
+        mock_def.simple_name = "get_users"
+        mock_def.type = "function"
+        
+        mock_visitor = Mock()
+        mock_visitor.framework_decorated_lines = {10}
+        mock_visitor.is_framework_file = True
+        
+        confidence = detect_framework_usage(mock_def, visitor=mock_visitor)
+        assert confidence == 20
+    
+    def test_framework_file_function_medium_confidence(self):
+        mock_def = Mock()
+        mock_def.line = 15
+        mock_def.simple_name = "helper_function"
+        mock_def.type = "function"
+        
+        mock_visitor = Mock()
+        mock_visitor.framework_decorated_lines = set()
+        mock_visitor.is_framework_file = True
+        
+        confidence = detect_framework_usage(mock_def, visitor=mock_visitor)
+        assert confidence == 40
+    
+    def test_private_function_in_framework_file_no_confidence(self):
+        mock_def = Mock()
+        mock_def.line = 20
+        mock_def.simple_name = "_private_function"
+        mock_def.type = "function"
+        
+        mock_visitor = Mock()
+        mock_visitor.framework_decorated_lines = set()
+        mock_visitor.is_framework_file = True
+        
+        confidence = detect_framework_usage(mock_def, visitor=mock_visitor)
+        assert confidence is None
+    
+    def test_non_framework_file_no_confidence(self):
+        mock_def = Mock()
+        mock_def.line = 25
+        mock_def.simple_name = "regular_function"
+        mock_def.type = "function"
+        
+        mock_visitor = Mock()
+        mock_visitor.framework_decorated_lines = set()
+        mock_visitor.is_framework_file = False
+        
+        confidence = detect_framework_usage(mock_def, visitor=mock_visitor)
+        assert confidence is None
+    
+    def test_no_visitor_returns_none(self):
+        mock_def = Mock()
+        confidence = detect_framework_usage(mock_def, visitor=None)
+        assert confidence is None
+    
+    def test_non_function_in_framework_file_no_confidence(self):
+        mock_def = Mock()
+        mock_def.line = 30
+        mock_def.simple_name = "my_variable"
+        mock_def.type = "variable"
+        
+        mock_visitor = Mock()
+        mock_visitor.framework_decorated_lines = set()
+        mock_visitor.is_framework_file = True
+        
+        confidence = detect_framework_usage(mock_def, visitor=mock_visitor)
+        assert confidence is None
+
+
+class TestFrameworkPatterns:
+    
+    def test_framework_decorators_list(self):
+        """Test that FRAMEWORK_DECORATORS contains expected patterns"""
+        assert "@app.route" in FRAMEWORK_DECORATORS
+        assert "@router.get" in FRAMEWORK_DECORATORS
+        assert "@login_required" in FRAMEWORK_DECORATORS
+        assert "@task" in FRAMEWORK_DECORATORS
+        assert "@validator" in FRAMEWORK_DECORATORS
+    
+    def test_framework_functions_list(self):
+        """Test that FRAMEWORK_FUNCTIONS contains expected patterns"""
+        assert "get" in FRAMEWORK_FUNCTIONS
+        assert "post" in FRAMEWORK_FUNCTIONS
+        assert "save" in FRAMEWORK_FUNCTIONS
+        assert "*_queryset" in FRAMEWORK_FUNCTIONS
+        assert "get_context_data" in FRAMEWORK_FUNCTIONS
+    
+    def test_framework_imports_set(self):
+        """Test that FRAMEWORK_IMPORTS contains expected frameworks"""
+        assert 'flask' in FRAMEWORK_IMPORTS
+        assert 'django' in FRAMEWORK_IMPORTS
+        assert 'fastapi' in FRAMEWORK_IMPORTS
+        assert 'celery' in FRAMEWORK_IMPORTS
+        assert 'pydantic' in FRAMEWORK_IMPORTS
+
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])

test/test_test_aware.py

@@ -0,0 +1,328 @@
+import pytest
+import ast
+from unittest.mock import patch
+from skylos.test_aware import TestAwareVisitor
+
+class TestTestAwareVisitor:
+    
+    def test_init_without_filename(self):
+        visitor = TestAwareVisitor()
+        assert visitor.is_test_file == False
+        assert visitor.test_decorated_lines == set()
+    
+    def test_init_with_non_test_filename(self):
+        visitor = TestAwareVisitor(filename="mymodule.py")
+        assert visitor.is_test_file == False
+        assert visitor.test_decorated_lines == set()
+    
+    @patch('skylos.test_aware.TEST_FILE_RE')
+    def test_init_with_test_filename(self, mock_test_file_re):
+        mock_test_file_re.search.return_value = True
+        
+        visitor = TestAwareVisitor(filename="test_module.py")
+        assert visitor.is_test_file == True
+        assert visitor.test_decorated_lines == set()
+        mock_test_file_re.search.assert_called_once_with("test_module.py")
+    
+    def test_test_function_name_patterns(self):
+        code = """
+def test_something():
+    pass
+
+def another_test():
+    pass
+
+def setup_method():
+    pass
+
+def teardown_function():
+    pass
+
+def setUp(self):
+    pass
+
+def tearDown(self):
+    pass
+
+def setUpClass(cls):
+    pass
+
+def tearDownClass(cls):
+    pass
+
+def setUpModule():
+    pass
+
+def tearDownModule():
+    pass
+
+def regular_function():
+    pass
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 2 in visitor.test_decorated_lines
+        assert 5 in visitor.test_decorated_lines
+        assert 8 in visitor.test_decorated_lines
+        assert 11 in visitor.test_decorated_lines
+        assert 14 in visitor.test_decorated_lines
+        assert 17 in visitor.test_decorated_lines  # tearDown
+        assert 20 in visitor.test_decorated_lines  # setUpClass
+        assert 23 in visitor.test_decorated_lines  # tearDownClass
+        assert 26 in visitor.test_decorated_lines  # setUpModule
+        assert 29 in visitor.test_decorated_lines  # tearDownModule
+        assert 32 not in visitor.test_decorated_lines  # gd ol regular_function
+    
+    @patch('skylos.test_aware.TEST_DECOR_RE')
+    def test_test_decorator_detection(self, mock_test_decor_re):
+        mock_test_decor_re.match.return_value = True
+        
+        code = """
+@pytest.mark.parametrize('a,b', [(1,2), (3,4)])
+def test_with_parametrize():
+    pass
+
+@mock.patch('module.function')
+def test_with_mock():
+    pass
+
+@unittest.skip("reason")
+def test_with_skip():
+    pass
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 3 in visitor.test_decorated_lines
+        assert 7 in visitor.test_decorated_lines
+        assert 11 in visitor.test_decorated_lines
+    
+    def test_pytest_fixture_detection(self):
+        code = """
+@pytest.fixture
+def sample_data():
+    return {"key": "value"}
+
+@pytest.fixture(scope="session")
+def database():
+    return create_test_db()
+
+@fixture
+def simple_fixture():
+    return "test"
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 3 in visitor.test_decorated_lines 
+        assert 11 in visitor.test_decorated_lines
+    
+    def test_async_test_function_detection(self):
+        code = """
+async def test_async_function():
+    await some_async_operation()
+
+@pytest.mark.asyncio
+async def test_with_asyncio_decorator():
+    await another_operation()
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 2 in visitor.test_decorated_lines
+        assert 6 in visitor.test_decorated_lines
+    
+    def test_decorator_name_extraction(self):
+        visitor = TestAwareVisitor()
+        
+        node = ast.parse("@decorator\ndef func(): pass").body[0].decorator_list[0]
+        result = visitor._decorator_name(node)
+        assert result == "decorator"
+        
+        node = ast.parse("@pytest.mark.skip\ndef func(): pass").body[0].decorator_list[0]
+        result = visitor._decorator_name(node)
+        assert result == "pytest.mark.skip"
+        
+        node = ast.parse("@fixture\ndef func(): pass").body[0].decorator_list[0]
+        result = visitor._decorator_name(node)
+        assert result == "fixture"
+    
+    def test_multiple_decorators_on_function(self):
+        """testing function with couple of  decorators"""
+        code = """
+@pytest.mark.parametrize('x', [1, 2, 3])
+@pytest.mark.slow
+@mock.patch('module.function')
+def test_multiple_decorators():
+    pass
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        # should be detected as test
+        assert 5 in visitor.test_decorated_lines
+    
+    @patch('skylos.test_aware.TEST_IMPORT_RE')
+    def test_test_import_detection_when_test_file(self, mock_test_import_re):
+        mock_test_import_re.match.return_value = True
+        
+        code = """
+import pytest
+import unittest
+from mock import Mock
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.is_test_file = True
+        visitor.visit(tree)
+        
+        assert mock_test_import_re.match.call_count >= 2
+    
+    @patch('skylos.test_aware.TEST_IMPORT_RE')
+    def test_test_import_detection_when_not_test_file(self, mock_test_import_re):
+        code = """
+import pytest
+import unittest
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.is_test_file = False
+        visitor.visit(tree)
+        
+        #  no call regex match for imports
+        assert mock_test_import_re.match.call_count == 0
+    
+    @patch('skylos.test_aware.TEST_IMPORT_RE')
+    def test_import_from_detection_when_test_file(self, mock_test_import_re):
+        mock_test_import_re.match.return_value = True
+        
+        code = """
+from pytest import fixture
+from unittest.mock import Mock
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.is_test_file = True
+        visitor.visit(tree)
+        
+        # should call regex match from-imports
+        assert mock_test_import_re.match.call_count >= 2
+    
+    def test_complex_test_class(self):
+        code = """
+class TestUserModel:
+    def setUp(self):
+        self.user = User()
+    
+    def test_user_creation(self):
+        assert self.user.name == "test"
+    
+    def test_user_validation(self):
+        assert self.user.is_valid()
+    
+    def tearDown(self):
+        self.user.delete()
+    
+    def helper_method(self):
+        return "not a test"
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 3 in visitor.test_decorated_lines   # setUp
+        assert 6 in visitor.test_decorated_lines   # test_user_creation
+        assert 9 in visitor.test_decorated_lines   # test_user_validation
+        assert 12 in visitor.test_decorated_lines  # tearDown
+        assert 15 not in visitor.test_decorated_lines  # helper_method
+    
+    def test_edge_case_function_names(self):
+        code = """
+def test():  # 'test', no underscore
+    pass
+
+def testing_function():  # 'test' but not test_
+    pass
+
+def function_test():  # ending with _test
+    pass
+
+def setuptools_install():  # 'setup' but starts with setup
+    pass
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 2 not in visitor.test_decorated_lines
+        assert 5 not in visitor.test_decorated_lines 
+        assert 8 in visitor.test_decorated_lines
+        assert 11 in visitor.test_decorated_lines
+    
+    def test_nested_functions(self):
+        code = """
+def outer_function():
+    def test_nested():
+        pass
+    
+    def setup_nested():
+        pass
+    
+    return test_nested
+"""
+        tree = ast.parse(code)
+        visitor = TestAwareVisitor()
+        visitor.visit(tree)
+        
+        assert 3 in visitor.test_decorated_lines
+        assert 6 in visitor.test_decorated_lines
+        assert 2 not in visitor.test_decorated_lines
+
+class TestTestAwareVisitorIntegration:
+    
+    @patch('skylos.test_aware.TEST_FILE_RE')
+    def test_full_test_file_analysis(self, mock_test_file_re):
+        mock_test_file_re.search.return_value = True
+        
+        code = """
+import pytest
+from unittest.mock import Mock
+
+class TestCalculator:
+    @pytest.fixture
+    def calculator(self):
+        return Calculator()
+    
+    def setUp(self):
+        self.mock_logger = Mock()
+    
+    def test_addition(self, calculator):
+        result = calculator.add(2, 3)
+        assert result == 5
+    
+    @pytest.mark.parametrize('a,b,expected', [(1,2,3), (4,5,9)])
+    def test_addition_parametrized(self, calculator, a, b, expected):
+        assert calculator.add(a, b) == expected
+    
+    def helper_method(self):
+        return "utility function"
+"""
+        visitor = TestAwareVisitor(filename="test_calculator.py")
+        tree = ast.parse(code)
+        visitor.visit(tree)
+        
+        assert visitor.is_test_file == True
+        assert 7 in visitor.test_decorated_lines
+        assert 10 in visitor.test_decorated_lines
+        assert 13 in visitor.test_decorated_lines
+        assert 18 in visitor.test_decorated_lines
+        assert 21 not in visitor.test_decorated_lines
+
+if __name__ == "__main__":
+    pytest.main([__file__, "-v"])