def create_random_category(db: Session) -> Category:
name = random_lower_string()
desc = random_lower_string()
return category_repo.create(db,
obj_in=CategoryCreate(name=name,
description=desc))
def test_create_category(db: Session):
name = random_lower_string()
desc = random_lower_string()
category = category_repo.create(db,
obj_in=CategoryCreate(name=name,
description=desc))
db_obj = category_repo.get(db, id=category.id)
assert db_obj
assert db_obj.name == name
assert db_obj.description == desc
def test_delete_category(db: Session):
name = random_lower_string()
desc = random_lower_string()
category = category_repo.create(db,
obj_in=CategoryCreate(name=name,
description=desc))
category_repo.remove(db, id=category.id)
db_obj = category_repo.get(db, id=category.id)
assert not db_obj
def test_get_category(client: TestClient, db: Session):
name = random_lower_string()
desc = random_lower_string()
category = category_repo.create(db,
obj_in=CategoryCreate(name=name,
description=desc))
data = {"name": name, "description": desc}
response = client.get(f"{settings.API_PREFIX}/v1/categories/{category.id}")
assert response.status_code == 200
content = response.json()
assert content["name"] == name
assert content["description"] == desc
def test_delete_category(client: TestClient, superuser_token_headers: dict,
db: Session):
name = random_lower_string()
desc = random_lower_string()
category = category_repo.create(db,
obj_in=CategoryCreate(name=name,
description=desc))
response = client.delete(
f"{settings.API_PREFIX}/v1/categories/{category.id}",
headers=superuser_token_headers)
assert response.status_code == 200
content = response.json()
assert content["name"] == name
assert content["description"] == desc
def create_category(*,
category_in: CategoryCreate,
db: Session = Depends(database.get_db),
current_user: User = Depends(
auth.get_current_active_superuser)):
return category_repo.create(db, obj_in=category_in)
