diff --git a/app/views/book.py b/app/views/book.py index d2601c2..629c314 100644 --- a/app/views/book.py +++ b/app/views/book.py @@ -37,8 +37,10 @@ def create(): label=form.label.data, ) log(log.INFO, "Form submitted. Book: [%s]", book) - flash("Book added!", "success") book.save() + m.BookVersion(semver="1.0.0", book_id=book.id).save() + + flash("Book added!", "success") return redirect(url_for("book.get_all")) else: log(log.ERROR, "Book create errors: [%s]", form.errors) diff --git a/tests/test_book.py b/tests/test_book.py index f6bbdbf..94b8355 100644 --- a/tests/test_book.py +++ b/tests/test_book.py @@ -58,3 +58,25 @@ def test_create_book(client: FlaskClient): book = m.Book.query.filter_by(label=BOOK_NAME).first() assert book + assert book.versions + assert len(book.versions) == 1 + + +# def test_add_c(client: FlaskClient): +# login(client) + + +# 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