146 lines
4.2 KiB
Python
146 lines
4.2 KiB
Python
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from hypothesis import given, strategies as st
|
|
from bson import ObjectId
|
|
from fastapi_server import app, db # Import your FastAPI app and database connection
|
|
|
|
client = TestClient(app)
|
|
|
|
# Helper function to clear the database before each test
|
|
|
|
|
|
@pytest.fixture(autouse=True)
|
|
def clear_db():
|
|
db.items.delete_many({})
|
|
yield
|
|
db.items.delete_many({})
|
|
|
|
# Traditional unit tests
|
|
|
|
|
|
def test_create_item():
|
|
response = client.post(
|
|
"/items/",
|
|
json={"name": "Test Item", "price": 10.99, "quantity": 5, "unit": "piece"}
|
|
)
|
|
assert response.status_code == 200
|
|
assert "id" in response.json()
|
|
|
|
|
|
def test_read_items():
|
|
# Create a test item
|
|
client.post("/items/", json={"name": "Test Item",
|
|
"price": 10.99, "quantity": 5, "unit": "piece"})
|
|
|
|
response = client.get("/items/")
|
|
assert response.status_code == 200
|
|
items = response.json()
|
|
assert len(items) > 0
|
|
assert items[0]["name"] == "Test Item"
|
|
|
|
|
|
def test_read_item():
|
|
# Create a test item
|
|
create_response = client.post(
|
|
"/items/", json={"name": "Test Item", "price": 10.99, "quantity": 5, "unit": "piece"})
|
|
item_id = create_response.json()["id"]
|
|
|
|
response = client.get(f"/items/{item_id}")
|
|
assert response.status_code == 200
|
|
assert response.json()["name"] == "Test Item"
|
|
|
|
|
|
def test_update_item():
|
|
# Create a test item
|
|
create_response = client.post(
|
|
"/items/", json={"name": "Test Item", "price": 10.99, "quantity": 5, "unit": "piece"})
|
|
item_id = create_response.json()["id"]
|
|
|
|
update_data = {"name": "Updated Item",
|
|
"price": 15.99, "quantity": 10, "unit": "piece"}
|
|
response = client.put(f"/items/{item_id}", json=update_data)
|
|
assert response.status_code == 200
|
|
assert response.json()["message"] == "Item updated successfully"
|
|
|
|
# Verify the update
|
|
get_response = client.get(f"/items/{item_id}")
|
|
assert get_response.json()["name"] == "Updated Item"
|
|
|
|
|
|
def test_delete_item():
|
|
# Create a test item
|
|
create_response = client.post(
|
|
"/items/", json={"name": "Test Item", "price": 10.99, "quantity": 5, "unit": "piece"})
|
|
item_id = create_response.json()["id"]
|
|
|
|
response = client.delete(f"/items/{item_id}")
|
|
assert response.status_code == 200
|
|
assert response.json()["message"] == "Item deleted successfully"
|
|
|
|
# Verify the deletion
|
|
get_response = client.get(f"/items/{item_id}")
|
|
assert get_response.status_code == 404
|
|
|
|
# Property-based tests
|
|
|
|
|
|
@given(
|
|
name=st.text(min_size=1, max_size=50),
|
|
price=st.floats(min_value=0.01, max_value=1000000,
|
|
allow_nan=False, allow_infinity=False),
|
|
quantity=st.integers(min_value=0, max_value=1000000),
|
|
unit=st.text(min_size=1, max_size=10)
|
|
)
|
|
def test_create_item_property(name, price, quantity, unit):
|
|
response = client.post(
|
|
"/items/",
|
|
json={"name": name, "price": price, "quantity": quantity, "unit": unit}
|
|
)
|
|
assert response.status_code == 200
|
|
assert "id" in response.json()
|
|
|
|
# Verify the created item
|
|
item_id = response.json()["id"]
|
|
get_response = client.get(f"/items/{item_id}")
|
|
assert get_response.status_code == 200
|
|
item = get_response.json()
|
|
assert item["name"] == name
|
|
assert pytest.approx(item["price"], 0.01) == price
|
|
assert item["quantity"] == quantity
|
|
assert item["unit"] == unit
|
|
|
|
|
|
@given(
|
|
st.lists(
|
|
st.fixed_dictionaries({
|
|
"name": st.text(min_size=1, max_size=50),
|
|
"price": st.floats(min_value=0.01, max_value=1000000, allow_nan=False, allow_infinity=False),
|
|
"quantity": st.integers(min_value=0, max_value=1000000),
|
|
"unit": st.text(min_size=1, max_size=10)
|
|
}),
|
|
min_size=1,
|
|
max_size=10
|
|
)
|
|
)
|
|
def test_read_items_property(items):
|
|
# Create multiple items
|
|
for item in items:
|
|
client.post("/items/", json=item)
|
|
|
|
response = client.get("/items/")
|
|
assert response.status_code == 200
|
|
retrieved_items = response.json()
|
|
assert len(retrieved_items) == len(items)
|
|
|
|
for retrieved_item in retrieved_items:
|
|
assert "name" in retrieved_item
|
|
assert "price" in retrieved_item
|
|
assert "quantity" in retrieved_item
|
|
assert "unit" in retrieved_item
|
|
|
|
# Add more property-based tests for update and delete operations if needed
|
|
|
|
|
|
if __name__ == "__main__":
|
|
pytest.main([__file__])
|