improvements on unit test

Author: JoeVictor22

Makefile

@@ -19,6 +19,13 @@ migrate_db:
 start:
 	python run.py runserver
 
+test: export STAGE=test
+test:
+	rm -Rf migrations/
+	python run.py db init
+	python run.py db migrate
+	python run.py db upgrade
+	python -m utils.scripts.insertData
+
 test:
-    ENV=test
 	PYTHONPATH=. pytest

config.py

@@ -15,7 +15,7 @@
 SQLALCHEMY_DATABASE_URI = os.getenv('DATABASE_URI', default)
 SQLALCHEMY_TEST_DATABASE_URI = os.getenv('DATABASE_TEST_URI', default)
 
-if os.environ.get("ENV", None) == "test":
+if os.environ.get("STAGE", None) == "test":
     SQLALCHEMY_DATABASE_URI = SQLALCHEMY_TEST_DATABASE_URI
 
 SQLALCHEMY_TRACK_MODIFICATIONS = True

tests/conftest.py

@@ -7,33 +7,36 @@
 
 from config import BASE_DIR
 
-@pytest.fixture
-def app():
-    return _app
 
-@pytest.fixture
-def db(app):
-    with app.app_context():
+def pytest_sessionstart(session):
+    with _app.app_context():
         _db.drop_all()
         _db.create_all()
 
-        session = db.session()
-        sql_file = open(BASE_DIR + "/utils/scripts/db/dml.sql", "r")
+        session = _db.session()
+
+        sql_file = open(BASE_DIR + "/utils/scripts/db/cidade_uf.sql", "r")
         escaped_sql = text(sql_file.read())
         session.execute(escaped_sql)
         session.flush()
 
-        sql_file = open(BASE_DIR + "/utils/scripts/db/tests.sql", "r")
+        sql_file = open(BASE_DIR + "/utils/scripts/db/rules.sql", "r")
         escaped_sql = text(sql_file.read())
         session.execute(escaped_sql)
         session.commit()
 
         session.close()
-        yield _db
-        _db.drop_all()
-        _db.session.commit()
 
 
+def pytest_sessionfinish(session, exitstatus):
+    _db.drop_all()
+    _db.session.commit()
+
+
+@pytest.fixture
+def app():
+    return _app
+
 
 @pytest.fixture
 def admin_login(app):

tests/test_cadastro_controller.py

@@ -4,7 +4,7 @@
 
 from pprint import pprint
 
-def test_get_perfis(app, db, admin_login):
+def test_get_perfis(app, admin_login):
     with app.app_context():
         client = app.test_client()
 
@@ -14,7 +14,7 @@ def test_get_perfis(app, db, admin_login):
         assert not output["error"]
         assert response.status_code == 200
 
-def test_post_perfil(app, db, admin_login):
+def test_post_perfil(app, admin_login):
     with app.app_context():
         client = app.test_client()
 
@@ -24,7 +24,7 @@ def test_post_perfil(app, db, admin_login):
         assert not output["error"]
         assert response.status_code == 200
 
-def test_view_perfil(app, db, admin_login):
+def test_view_perfil(app, admin_login):
     with app.app_context():
         client = app.test_client()
 

tests/test_user_controller.py

@@ -4,7 +4,7 @@
 
 from pprint import pprint
 
-def test_get_usuarios(app, db, admin_login):
+def test_get_usuarios(app, admin_login):
     with app.app_context():
         client = app.test_client()
 
@@ -14,7 +14,7 @@ def test_get_usuarios(app, db, admin_login):
         assert not output["error"]
         assert response.status_code == 200
 
-def test_post_usuario(app, db, admin_login):
+def test_post_usuario(app, admin_login):
     with app.app_context():
         client = app.test_client()
 
@@ -24,7 +24,7 @@ def test_post_usuario(app, db, admin_login):
         assert not output["error"]
         assert response.status_code == 200
 
-def test_view_usuario(app, db, admin_login, created_login):
+def test_view_usuario(app, admin_login, created_login):
     with app.app_context():
         client = app.test_client()
 

utils/scripts/db/dml_acl.sql renamed to utils/scripts/db/rules.sql

@@ -29,7 +29,7 @@ def insert(filePath, output):
 
     insert("db/cidade_uf.sql", output)
 
-    insert("db/dml_acl.sql", output)
+    insert("db/rules.sql", output)
 
     try:
         db.session.commit()