test_model.py

"""Specific tests for the Model and it's methods."""

import pytest

from sqliter.model.model import BaseDBModel


class TestBaseDBModel:
    """Test the Model and it's methods."""

    def test_should_create_pk(self) -> None:
        """Test that 'should_create_pk' returns True."""
        assert BaseDBModel.should_create_pk() is True

    def test_get_primary_key(self) -> None:
        """Test that 'get_primary_key' returns 'pk'."""
        assert BaseDBModel.get_primary_key() == "pk"

    def test_get_table_name_default(self) -> None:
        """Test that 'get_table_name' returns the default table name."""

        class TestModel(BaseDBModel):
            pass

        assert TestModel.get_table_name() == "tests"

    def test_get_table_name_custom(self) -> None:
        """Test that 'get_table_name' returns the custom table name."""

        class TestModel(BaseDBModel):
            class Meta:
                table_name = "custom_table"

        assert TestModel.get_table_name() == "custom_table"

    def test_get_table_name_invalid_characters(self) -> None:
        """Test that 'get_table_name' rejects invalid characters."""

        class TestModel(BaseDBModel):
            class Meta:
                table_name = "invalid-table-name"

        with pytest.raises(ValueError, match="Invalid table name"):
            TestModel.get_table_name()

    def test_get_table_name_sql_injection_attempt(self) -> None:
        """Test that 'get_table_name' prevents SQL injection."""

        class TestModel(BaseDBModel):
            class Meta:
                table_name = 'users"; DROP TABLE users; --'

        with pytest.raises(ValueError, match="Invalid table name"):
            TestModel.get_table_name()

    def test_get_table_name_starts_with_number(self) -> None:
        """Test that 'get_table_name' rejects names starting with numbers."""

        class TestModel(BaseDBModel):
            class Meta:
                table_name = "123table"

        with pytest.raises(
            ValueError, match=r"Invalid table name.*must start with"
        ):
            TestModel.get_table_name()

    def test_get_table_name_with_underscore(self) -> None:
        """Test that 'get_table_name' allows underscores."""

        class TestModel(BaseDBModel):
            class Meta:
                table_name = "my_table_name"

        assert TestModel.get_table_name() == "my_table_name"

    def test_model_validate_partial(self) -> None:
        """Test 'model_validate_partial' with partial data."""

        class TestModel(BaseDBModel):
            name: str
            age: int | None

        data = {"name": "John"}
        model_instance = TestModel.model_validate_partial(data)
        assert model_instance.name == "John"
        with pytest.raises(AttributeError):
            _ = model_instance.age