Migrate test_basic_app to pytest engine

Author: insspb

tests/conftest.py

@@ -1,7 +1,11 @@
+from datetime import datetime
+
 import mongoengine
 import pytest
 from flask import Flask
 
+from flask_mongoengine import MongoEngine
+
 
 @pytest.fixture()
 def app():
@@ -14,12 +18,32 @@ def app():
     mongoengine.connection.disconnect_all()
 
 
+@pytest.fixture()
+def db(app):
+    app.config["MONGODB_HOST"] = "mongodb://localhost:27017/flask_mongoengine_test_db"
+    test_db = MongoEngine(app)
+    db_name = test_db.connection.get_database("flask_mongoengine_test_db").name
+
+    if not db_name.endswith("_test_db"):
+        raise RuntimeError(
+            f"DATABASE_URL must point to testing db, not to master db ({db_name})"
+        )
+
+    # Clear database before tests, for cases when some test failed before.
+    test_db.connection.drop_database(db_name)
+
+    yield test_db
+
+    # Clear database after tests, for graceful exit.
+    test_db.connection.drop_database(db_name)
+
+
 @pytest.fixture()
 def todo(db):
     class Todo(db.Document):
-        # meta = {"db_alias": alias}
         title = mongoengine.StringField(max_length=60)
         text = mongoengine.StringField()
         done = mongoengine.BooleanField(default=False)
+        pub_date = mongoengine.DateTimeField(default=datetime.utcnow)
 
     return Todo

tests/test_basic_app.py

@@ -0,0 +1,45 @@
+import flask
+import pytest
+from bson import ObjectId
+
+
+@pytest.fixture(autouse=True)
+def setup_endpoints(app, todo):
+    Todo = todo
+
+    @app.route("/")
+    def index():
+        return "\n".join(x.title for x in Todo.objects)
+
+    @app.route("/add", methods=["POST"])
+    def add():
+        form = flask.request.form
+        todo = Todo(title=form["title"], text=form["text"])
+        todo.save()
+        return "added"
+
+    @app.route("/show/<id>/")
+    def show(id):
+        todo = Todo.objects.get_or_404(id=id)
+        return "\n".join([todo.title, todo.text])
+
+
+def test_with_id(app, todo):
+    Todo = todo
+    client = app.test_client()
+    response = client.get("/show/%s/" % ObjectId())
+    assert response.status_code == 404
+
+    client.post("/add", data={"title": "First Item", "text": "The text"})
+
+    response = client.get("/show/%s/" % Todo.objects.first_or_404().id)
+    assert response.status_code == 200
+    assert response.data.decode("utf-8") == "First Item\nThe text"
+
+
+def test_basic_insert(app):
+    client = app.test_client()
+    client.post("/add", data={"title": "First Item", "text": "The text"})
+    client.post("/add", data={"title": "2nd Item", "text": "The text"})
+    response = client.get("/")
+    assert response.data.decode("utf-8") == "First Item\n2nd Item"