diff --git a/app/crud.py b/app/crud.py index 31d7b45..53f2f24 100644 --- a/app/crud.py +++ b/app/crud.py @@ -2,7 +2,9 @@ from app.schemas import ContactCreate async def create_contact(db, contact_data: ContactCreate) -> Contact: - contact = Contact(**contact_data.dict()) + # Exclude recaptcha_token as it's not part of Contact model + contact_dict = contact_data.model_dump(exclude={"recaptcha_token"}) + contact = Contact(**contact_dict) db.add(contact) await db.commit() await db.refresh(contact) diff --git a/tests/test_crud.py b/tests/test_crud.py index a32280d..43e0af9 100644 --- a/tests/test_crud.py +++ b/tests/test_crud.py @@ -20,7 +20,7 @@ mock_session.commit = AsyncMock() mock_session.refresh = AsyncMock() - contact_data = ContactCreate(name="Test", email="test@example.com", subject="Test", message="Hello") + contact_data = ContactCreate(name="Test", email="test@example.com", subject="Test", message="Hello", recaptcha_token="fake-token") result = await create_contact(mock_session, contact_data) mock_session.add.assert_called_once() mock_session.commit.assert_called_once() diff --git a/tests/test_schemas.py b/tests/test_schemas.py index 0e8a43a..55ad189 100644 --- a/tests/test_schemas.py +++ b/tests/test_schemas.py @@ -5,7 +5,8 @@ "name": "Jean Dupont", "email": "jean@example.com", "subject": "Test", - "message": "Hello" + "message": "Hello", + "recaptcha_token": "fake-token" } contact = ContactCreate(**data) assert contact.name == "Jean Dupont" @@ -16,7 +17,8 @@ "name": "Test", "email": "not-an-email", "subject": "Test", - "message": "Hello" + "message": "Hello", + "recaptcha_token": "fake-token" } try: ContactCreate(**data)