# flake8: noqa F501 from flask import current_app as Response from flask.testing import FlaskClient, FlaskCliRunner from app import models as m, db from tests.utils import login def test_create_edit_book(client: FlaskClient): login(client) BOOK_NAME = "Test Book" # label len < 6 response: Response = client.post( "/book/create", data=dict( label="12345", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Label must be between 6 and 256 characters long." in response.data book = m.Book.query.filter_by(label=BOOK_NAME).first() assert not book assert not m.Book.query.count() # label len > 256 response: Response = client.post( "/book/create", data=dict( label="".join(["0" for _ in range(1025)]), ), follow_redirects=True, ) assert response.status_code == 200 assert b"Label must be between 6 and 256 characters long." in response.data book = m.Book.query.filter_by(label=BOOK_NAME).first() assert not book assert not m.Book.query.count() response: Response = client.post( "/book/create", data=dict( label=BOOK_NAME, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Book added!" in response.data book = m.Book.query.filter_by(label=BOOK_NAME).first() assert book assert book.versions assert len(book.versions) == 1 response: Response = client.post( "/book/999/edit", data=dict( book_id=999, label="Book 1", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Book not found" in response.data response: Response = client.post( f"/book/{book.id}/edit", data=dict( book_id=book.id, label=BOOK_NAME, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Book label must be unique!" in response.data response: Response = client.post( f"/book/{book.id}/edit", data=dict( book_id=book.id, label=BOOK_NAME + " EDITED", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data book = db.session.get(m.Book, book.id) assert book.label != BOOK_NAME def test_add_contributor(client: FlaskClient): _, user = login(client) user: m.User moderator = m.User(username="Moderator", password="test").save() moderators_book = m.Book(label="Test Book", user_id=moderator.id).save() response: Response = client.post( f"/book/{moderators_book.id}/add_contributor", data=dict(user_id=moderator.id, role=m.BookContributor.Roles.MODERATOR), follow_redirects=True, ) assert response.status_code == 200 assert b"You are not owner of this book!" in response.data book = m.Book(label="Test Book", user_id=user.id).save() response: Response = client.post( f"/book/{book.id}/add_contributor", data=dict(user_id=moderator.id, role=m.BookContributor.Roles.MODERATOR), follow_redirects=True, ) assert response.status_code == 200 assert b"Contributor was added!" in response.data response: Response = client.post( f"/book/{book.id}/add_contributor", data=dict(user_id=moderator.id, role=m.BookContributor.Roles.MODERATOR), follow_redirects=True, ) assert response.status_code == 200 assert b"Already exists!" in response.data contributor: m.BookContributor = m.BookContributor.query.filter_by( user=moderator, book=book ).first() assert contributor.role == m.BookContributor.Roles.MODERATOR assert len(book.contributors) == 1 editor = m.User(username="Editor", password="test").save() response: Response = client.post( f"/book/{book.id}/add_contributor", data=dict(user_id=editor.id, role=m.BookContributor.Roles.EDITOR), follow_redirects=True, ) assert response.status_code == 200 assert b"Contributor was added!" in response.data contributor: m.BookContributor = m.BookContributor.query.filter_by( user=editor, book=book ).first() assert contributor.role == m.BookContributor.Roles.EDITOR assert len(book.contributors) == 2 def test_delete_contributor(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book = db.session.get(m.Book, 1) book.user_id = user.id book.save() contributors_len = len(book.contributors) assert contributors_len contributor_to_delete = book.contributors[0] response: Response = client.post( f"/book/{book.id}/delete_contributor", data=dict(user_id=contributor_to_delete.user_id), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data response: Response = client.post( f"/book/{book.id}/delete_contributor", data=dict(user_id=contributor_to_delete.user_id), follow_redirects=True, ) assert response.status_code == 200 assert b"Does not exists!" in response.data def test_edit_contributor_role(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book = db.session.get(m.Book, 1) book.user_id = user.id book.save() contributors_len = len(book.contributors) assert contributors_len contributor_edit = book.contributors[0] assert contributor_edit.role == m.BookContributor.Roles.MODERATOR response: Response = client.post( f"/book/{book.id}/edit_contributor_role", data=dict( user_id=contributor_edit.user_id, role=m.BookContributor.Roles.MODERATOR.value, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data def test_crud_collection(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book = db.session.get(m.Book, 1) book.user_id = user.id book.save() response: Response = client.post( f"/book/{book.id}/create_collection", data=dict(label="Test Collection #1 Label", about="Test Collection #1 About"), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data response: Response = client.post( f"/book/{book.id}/create_collection", data=dict(label="Test Collection #1 Label", about="Test Collection #1 About"), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection label must be unique!" in response.data collection: m.Collection = m.Collection.query.filter_by( label="Test Collection #1 Label" ).first() m.Collection( label="Test Collection #2 Label", version_id=collection.version_id, parent_id=collection.parent_id, ).save() response: Response = client.post( f"/book/{book.id}/{collection.id}/edit", data=dict( label="Test Collection #2 Label", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection label must be unique!" in response.data new_label = "Test Collection #1 Label(edited)" new_about = "Test Collection #1 About(edited)" response: Response = client.post( f"/book/{book.id}/{collection.id}/edit", data=dict( label=new_label, about=new_about, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data edited_collection: m.Collection = m.Collection.query.filter_by( label=new_label, about=new_about ).first() assert edited_collection response: Response = client.post( f"/book/{book.id}/999/edit", data=dict( label=new_label, about=new_about, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection not found" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/delete", follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_collection: m.Collection = db.session.get(m.Collection, collection.id) assert deleted_collection.is_deleted response: Response = client.post( f"/book/{book.id}/{collection.id}/delete", follow_redirects=True, ) assert response.status_code == 200 assert b"Collection not found" in response.data def test_crud_subcollection(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book: m.Book = db.session.get(m.Book, 1) book.user_id = user.id book.save() leaf_collection: m.Collection = m.Collection( label="Test Leaf Collection #1 Label", version_id=book.last_version.id, is_leaf=True, parent_id=book.last_version.root_collection.id, ).save() collection: m.Collection = m.Collection( label="Test Collection #1 Label", version_id=book.last_version.id ).save() response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/create_sub_collection", data=dict( label="Test SubCollection #1 Label", about="Test SubCollection #1 About" ), follow_redirects=True, ) assert response.status_code == 200 assert b"You can't create subcollection for this collection" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/create_sub_collection", data=dict( label="Test SubCollection #1 Label", about="Test SubCollection #1 About" ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/create_sub_collection", data=dict( label="Test SubCollection #1 Label", about="Test SubCollection #1 About" ), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection label must be unique!" in response.data sub_collection: m.Collection = m.Collection.query.filter_by( label="Test SubCollection #1 Label" ).first() assert sub_collection assert sub_collection.is_leaf assert sub_collection.parent_id == collection.id m.Collection( label="Test SubCollection #2 Label", version_id=collection.version_id, parent_id=collection.id, ).save() response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/edit", data=dict( label="Test SubCollection #2 Label", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection label must be unique!" in response.data new_label = "Test SubCollection #1 Label(edited)" new_about = "Test SubCollection #1 About(edited)" response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/edit", data=dict( label=new_label, about=new_about, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data edited_collection: m.Collection = m.Collection.query.filter_by( label=new_label, about=new_about ).first() assert edited_collection response: Response = client.post( f"/book/{book.id}/{collection.id}/9999/edit", data=dict( label=new_label, about=new_about, ), follow_redirects=True, ) assert response.status_code == 200 assert b"SubCollection not found" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/delete", follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_collection: m.Collection = db.session.get(m.Collection, sub_collection.id) assert deleted_collection.is_deleted response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/delete", follow_redirects=True, ) assert response.status_code == 200 assert b"Collection not found" in response.data def test_crud_sections(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book: m.Book = db.session.get(m.Book, 1) book.user_id = user.id book.save() leaf_collection: m.Collection = m.Collection( label="Test Leaf Collection #1 Label", version_id=book.last_version.id, is_leaf=True, parent_id=book.last_version.root_collection.id, ).save() collection: m.Collection = m.Collection( label="Test Collection #1 Label", version_id=book.last_version.id ).save() sub_collection: m.Collection = m.Collection( label="Test SubCollection #1 Label", version_id=book.last_version.id, parent_id=collection.id, is_leaf=True, ).save() leaf_collection.is_leaf = False leaf_collection.save() response: Response = client.post( f"/book/{book.id}/{collection.id}/create_section", data=dict( collection_id=collection.id, label="Test Section", about="Test Section #1 About", ), follow_redirects=True, ) assert b"You can't create section for this collection" in response.data leaf_collection.is_leaf = True leaf_collection.save() label_1 = "Test Section #1 Label" response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/create_section", data=dict( collection_id=leaf_collection.id, label=label_1, about="Test Section #1 About", ), follow_redirects=True, ) assert response.status_code == 200 section: m.Section = m.Section.query.filter_by( label=label_1, collection_id=leaf_collection.id ).first() assert section assert section.collection_id == leaf_collection.id assert section.version_id == book.last_version.id assert not section.interpretations response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/create_section", data=dict( collection_id=leaf_collection.id, label=label_1, about="Test Section #1 About", ), follow_redirects=True, ) assert b"Section label must be unique!" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/create_section", data=dict( collection_id=sub_collection.id, label=label_1, about="Test Section #1 About", ), follow_redirects=True, ) assert response.status_code == 200 section: m.Section = m.Section.query.filter_by( label=label_1, collection_id=sub_collection.id ).first() assert section assert section.collection_id == sub_collection.id assert section.version_id == book.last_version.id assert not section.interpretations response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/create_section", data=dict( collection_id=sub_collection.id, label=label_1, about="Test Section #1 About", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Section label must be unique!" in response.data response: Response = client.post( f"/book/{book.id}/999/create_section", data=dict( collection_id=999, label=label_1, about="Test Section #1 About", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Collection not found" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/999/create_section", data=dict(collection_id=999, label=label_1, about="Test Section #1 About"), follow_redirects=True, ) assert response.status_code == 200 assert b"Subcollection not found" in response.data # edit m.Section( label="Test", about="Test", collection_id=leaf_collection.id, version_id=book.last_version.id, ).save() m.Section( label="Test", about="Test", collection_id=sub_collection.id, version_id=book.last_version.id, ).save() section: m.Section = m.Section.query.filter_by( label=label_1, collection_id=leaf_collection.id ).first() response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section.id}/edit_section", data=dict( section_id=section.id, label="Test", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Section label must be unique!" in response.data new_label = "Test Section #1 Label(edited)" new_about = "Test Section #1 About(edited)" response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section.id}/edit_section", data=dict(section_id=section.id, label=new_label, about=new_about), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data edited_section: m.Section = m.Section.query.filter_by( label=new_label, about=new_about, id=section.id ).first() assert edited_section # section_2: m.Section = m.Section.query.filter_by( label=label_1, collection_id=sub_collection.id ).first() response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_2.id}/edit_section", data=dict( section_id=section_2.id, label="Test", ), follow_redirects=True, ) assert response.status_code == 200 assert b"Section label must be unique!" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_2.id}/edit_section", data=dict(section_id=section_2.id, label=new_label, about=new_about), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data edited_section: m.Section = m.Section.query.filter_by( label=new_label, about=new_about, id=section_2.id ).first() assert edited_section response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/999/edit_section", data=dict(section_id=section_2.id, label=new_label, about=new_about), follow_redirects=True, ) assert response.status_code == 200 assert b"Section not found" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/{leaf_collection.id}/{section.id}/delete_section", follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_section: m.Section = db.session.get(m.Section, section.id) assert deleted_section.is_deleted response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_2.id}/delete_section", follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_section: m.Section = db.session.get(m.Section, section_2.id) assert deleted_section.is_deleted response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/999/delete_section", follow_redirects=True, ) assert response.status_code == 200 assert b"Section not found" in response.data def test_crud_interpretation(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book: m.Book = db.session.get(m.Book, 1) book.user_id = user.id book.save() leaf_collection: m.Collection = m.Collection( label="Test Leaf Collection #1 Label", version_id=book.last_version.id, is_leaf=True, parent_id=book.last_version.root_collection.id, ).save() section_in_collection: m.Section = m.Section( label="Test Section in Collection #1 Label", about="Test Section in Collection #1 About", collection_id=leaf_collection.id, version_id=book.last_version.id, ).save() collection: m.Collection = m.Collection( label="Test Collection #1 Label", version_id=book.last_version.id ).save() sub_collection: m.Collection = m.Collection( label="Test SubCollection #1 Label", version_id=book.last_version.id, parent_id=collection.id, is_leaf=True, ).save() section_in_subcollection: m.Section = m.Section( label="Test Section in Subcollection #1 Label", about="Test Section in Subcollection #1 About", collection_id=sub_collection.id, version_id=book.last_version.id, ).save() label_1 = "Test Interpretation #1 Label" text_1 = "Test Interpretation #1 Text" response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_in_subcollection.id}/create_interpretation", data=dict(section_id=section_in_subcollection.id, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 interpretation: m.Interpretation = m.Interpretation.query.filter_by( label=label_1, section_id=section_in_subcollection.id ).first() assert interpretation assert interpretation.section_id == section_in_subcollection.id assert not interpretation.comments response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section_in_collection.id}/create_interpretation", data=dict(section_id=section_in_collection.id, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 interpretation: m.Interpretation = m.Interpretation.query.filter_by( label=label_1, section_id=section_in_collection.id ).first() assert interpretation assert interpretation.section_id == section_in_collection.id assert not interpretation.comments response: Response = client.post( f"/book/{book.id}/{collection.id}/999/create_section", data=dict(collection_id=999, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 assert b"Subcollection not found" in response.data response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/999/create_interpretation", data=dict(collection_id=999, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 assert b"Section not found" in response.data response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/888/create_interpretation", data=dict(collection_id=999, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 assert b"Section not found" in response.data # edit m.Interpretation( label="Test", text="Test", section_id=section_in_collection.id, ).save() m.Interpretation( label="Test", text="Test", section_id=section_in_subcollection.id, ).save() interpretation: m.Interpretation = m.Interpretation.query.filter_by( label=label_1, section_id=section_in_collection.id ).first() new_label = "Test Interpretation #1 Label(edited)" new_text = "Test Interpretation #1 Text(edited)" response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section_in_collection.id}/{interpretation.id}/edit_interpretation", data=dict( label=new_label, interpretation_id=interpretation.id, text=new_text, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data edited_interpretation: m.Interpretation = m.Interpretation.query.filter_by( label=new_label, text=new_text, id=interpretation.id ).first() assert edited_interpretation response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section_in_collection.id}/999/edit_interpretation", data=dict( interpretation_id=interpretation.id, label=new_label, text=new_text, ), follow_redirects=True, ) assert response.status_code == 200 assert b"Interpretation not found" in response.data response: Response = client.post( f"/book/{book.id}/{leaf_collection.id}/{section_in_collection.id}/999/delete_interpretation", follow_redirects=True, ) assert response.status_code == 200 assert b"Interpretation not found" in response.data response: Response = client.post( ( f"/book/{book.id}/{collection.id}/{sub_collection.id}/" f"{section_in_subcollection.id}/{section_in_subcollection.interpretations[0].id}/delete_interpretation" ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_interpretation: m.Interpretation = db.session.get( m.Interpretation, section_in_subcollection.interpretations[0].id ) assert deleted_interpretation.is_deleted response: Response = client.post( ( f"/book/{book.id}/{leaf_collection.id}/{section_in_collection.id}/" f"{section_in_collection.interpretations[0].id}/delete_interpretation" ), follow_redirects=True, ) assert response.status_code == 200 assert b"Success!" in response.data deleted_interpretation: m.Interpretation = db.session.get( m.Interpretation, section_in_collection.interpretations[0].id ) assert deleted_interpretation.is_deleted def test_crud_comment(client: FlaskClient, runner: FlaskCliRunner): _, user = login(client) user: m.User # add dummmy data runner.invoke(args=["db-populate"]) book: m.Book = db.session.get(m.Book, 1) book.user_id = user.id book.save() leaf_collection: m.Collection = m.Collection( label="Test Leaf Collection #1 Label", version_id=book.last_version.id, is_leaf=True, parent_id=book.last_version.root_collection.id, ).save() section_in_collection: m.Section = m.Section( label="Test Section in Collection #1 Label", about="Test Section in Collection #1 About", collection_id=leaf_collection.id, version_id=book.last_version.id, ).save() collection: m.Collection = m.Collection( label="Test Collection #1 Label", version_id=book.last_version.id ).save() sub_collection: m.Collection = m.Collection( label="Test SubCollection #1 Label", version_id=book.last_version.id, parent_id=collection.id, is_leaf=True, ).save() section_in_subcollection: m.Section = m.Section( label="Test Section in Subcollection #1 Label", about="Test Section in Subcollection #1 About", collection_id=sub_collection.id, version_id=book.last_version.id, ).save() label_1 = "Test Interpretation #1 Label" text_1 = "Test Interpretation #1 Text" response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_in_subcollection.id}/create_interpretation", data=dict(section_id=section_in_subcollection.id, label=label_1, text=text_1), follow_redirects=True, ) assert response.status_code == 200 interpretation: m.Interpretation = m.Interpretation.query.filter_by( label=label_1, section_id=section_in_subcollection.id ).first() assert interpretation assert interpretation.section_id == section_in_subcollection.id assert not interpretation.comments comment_text = "Some comment text" response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_in_subcollection.id}/{interpretation.id}/preview/create_comment", data=dict( section_id=section_in_subcollection.id, text=comment_text, interpretation_id=interpretation.id, ), follow_redirects=True, ) assert response assert response.status_code == 200 assert b"Success" in response.data assert str.encode(comment_text) in response.data comment: m.Comment = m.Comment.query.filter_by(text=comment_text).first() assert comment # delete response: Response = client.post( f"/book/{book.id}/{collection.id}/{sub_collection.id}/{section_in_subcollection.id}/{interpretation.id}/comment_delete", data=dict( section_id=section_in_subcollection.id, text=comment_text, interpretation_id=interpretation.id, comment_id=comment.id, ), follow_redirects=True, ) assert response assert response.status_code == 200 assert b"Success" in response.data assert str.encode(comment_text) not in response.data