diff --git a/tests/test_forms.py b/tests/test_forms.py index 90ecab7a..64ff01c0 100644 --- a/tests/test_forms.py +++ b/tests/test_forms.py @@ -426,6 +426,7 @@ class TestModel(db.Document): def test_modelselectfield_multiple_selected_elements_must_be_retained(app, db): with app.test_request_context("/"): + class Dog(db.Document): name = db.StringField() @@ -448,7 +449,7 @@ class DogOwner(db.Document): # Should have one selected option assert m is not None - assert "fido" == m.group(1) + assert "fido" == m[1] def test_model_form_help_text(app, db): @@ -497,6 +498,7 @@ class BlogPost(db.Document): def test_embedded_model_form(app, db): with app.test_request_context("/"): + class Content(db.EmbeddedDocument): text = db.StringField() lang = db.StringField(max_length=3) @@ -508,21 +510,25 @@ class Post(db.Document): PostForm = model_form(Post) form = PostForm() - assert "content-text" in "%s" % form.content.text + assert "content-text" in f"{form.content.text}" def test_form_label_modifier(app, db): with app.test_request_context("/"): + class FoodItem(db.Document): title = db.StringField() + + class FoodStore(db.Document): title = db.StringField(max_length=120, required=True) food_items = db.ListField(field=db.ReferenceField(document_type=FoodItem)) - def food_items_label_modifier(obj): - return obj.title + def food_items_label_modifier(self): + return self.title + fruit_names = ["banana", "apple", "pear"]