def test_filter_by_tags(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
tag1 = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag1'}))
tag2 = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag2'}))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'tag_ids': [tag1.id],
}
crud.create_post(db, schemas.CreatePost(**posts))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'tag_ids': [tag1.id, tag2.id],
}
crud.create_post(db, schemas.CreatePost(**posts))
q = db.query(models.Post)
q = crud.filter_post_by_tags(q, [tag1.id, tag2.id])
assert len(q.all()) == 1
db.close()
def test_delete_post(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
category = crud.create_category(db, schemas.CreateCategory(**{"name": 'cat'}))
series = crud.create_series(db, schemas.CreateSeries(**{"name": 'ser'}))
tag = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag'}))
tag_name = tag.name
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'category_id': category.id,
'series_id': series.id,
'tag_ids': [tag.id],
}
post = crud.create_post(db, schemas.CreatePost(**posts))
db.close()
db = SessionLocal()
crud.delete_post(db, post.id)
db.close()
db = SessionLocal()
post = crud.get_post(db, post.id)
assert post is None
db.close()
db = SessionLocal()
_tag = crud.get_tag(db, tag_name)
assert _tag.name == tag_name, 'tag is deleted'
db.close()
def test_filter_by_is_public(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'is_public': False
}
crud.create_post(db, schemas.CreatePost(**posts))
q = db.query(models.Post)
q = crud.filter_post_by_is_public(q)
assert len(q.all()) == 0
db.close()
def test_filter_by_series(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
series = crud.create_series(db, schemas.CreateSeries(**{"name": 'ser'}))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'series_id': series.id,
}
crud.create_post(db, schemas.CreatePost(**posts))
q = db.query(models.Post)
q = crud.filter_post_by_series(q, series.id)
assert len(q.all()) == 1
q = db.query(models.Post)
q = crud.filter_post_by_series(q, series.id + 1)
assert len(q.all()) == 0, 'filtering by category does not work'
db.close()
def create_post(
post: schemas.CreatePostReq,
db: Session = Depends(get_db),
current_user=Depends(get_current_active_user),
):
create_id_post_dict = utils.CreateIDPost(db, post).to_items()
create_id_post_dict["author_id"] = current_user.id
_post = schemas.CreatePost(**create_id_post_dict)
obj = crud.create_post(db, _post)
if obj.notification and obj.is_public:
post_mail_notify(obj)
return obj
def test_update_post(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
category = crud.create_category(db, schemas.CreateCategory(**{"name": 'cat'}))
category2 = crud.create_category(db, schemas.CreateCategory(**{"name": 'cat2'}))
series = crud.create_series(db, schemas.CreateSeries(**{"name": 'ser'}))
tag = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag'}))
tag2 = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag2'}))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'category_id': category.id,
'series_id': series.id,
'tag_ids': [tag.id],
}
post = crud.create_post(db, schemas.CreatePost(**posts))
assert post.title == 'hoge'
assert post.body == 'fuga'
assert post.author_id == user.id
assert post.category_id == category.id
assert post.series_id == series.id
assert post.tag[0] == tag
new_posts = {
'title': 'hoge2',
'body': 'fuga2',
'public_at': '2020-05-17T00:00:00',
'category': category2.name,
'tags': [tag2.name],
}
new_post = schemas.UpdatePostReq(**new_posts)
new_post = utils.UpdateIDPost(db, new_post)
updated_post = crud.update_post(db, base_post=post, post=new_post)
assert updated_post.title == 'hoge2'
assert updated_post.body == 'fuga2'
assert updated_post.author_id == user.id
assert updated_post.category_id == category2.id
assert updated_post.series_id == series.id
assert updated_post.tag[0] == tag2
db.close()
def test_create_post(SessionLocal):
db = SessionLocal()
user = u_crud.create_user_query(db, u_schemas.UserCreate(**{"email": "foo", "name": "fooo", "password": "******"}))
category = crud.create_category(db, schemas.CreateCategory(**{"name": 'cat'}))
series = crud.create_series(db, schemas.CreateSeries(**{"name": 'ser'}))
tag = crud.create_tag(db, schemas.CreateTag(**{"name": 'tag'}))
posts = {
'title': 'hoge',
'body': 'fuga',
'public_at': '2020-05-17T00:00:00',
'author_id': user.id,
'category_id': category.id,
'series_id': series.id,
'tag_ids': [tag.id],
}
post = crud.create_post(db, schemas.CreatePost(**posts))
assert post.title == 'hoge'
assert post.body == 'fuga'
assert post.author_id == user.id
assert post.category_id == category.id
assert post.series_id == series.id
assert post.tag[0] == tag
db.close()