Run auth tests on sqlite:memory: db

hitier
1 parent 2bf8caae
Showing 1 changed file with 21 additions and 3 deletions Show diff stats
tests/backend_tests.py
@@ -5,13 +5,16 @@ from app import create_app, db_mgr, db
 from app.auth.models import User
  
  
+
+
 class BaseTestCase(unittest.TestCase):
     def setUp(self):
         # configure data base
         self.app = create_app(TestConfig)
         self.app_context = self.app.app_context()
         self.app_context.push()
-        # TODO: shall we always copy db from resources ?
+        # TODO: shall we always copy db from resources sqlite file ?
+        #       that would allow us to run all tests in sqlite:memory:
         # db.create_all()
         # admin = User(email='admin@nowhere.org', name='admin', login='admin', password='admin', role='admin')
         # db.session.add(admin)
@@ -47,12 +50,27 @@ class DbMgrTestCase(BaseTestCase):
  
  
 class AuthModelTestCase(BaseTestCase):
+    def skip_if_no_sqlitememory(self):
+        if 'memory' not in self.app.config['SQLALCHEMY_DATABASE_URI']:
+            self.skipTest("Needs in memory sqlite")
+
+    def setUp(self):
+        BaseTestCase.setUp(self)
+        self.skip_if_no_sqlitememory()
+        db.create_all()
+        admin = User(email='admin@nowhere.org', name='admin', login='admin', role='admin')
+        db.session.add(admin)
+        db.session.commit()
+
+    def test_in_memory(self):
+        self.app.logger.info("In memory Sqlite DB for tests")
+        self.assertTrue(True)
  
     def test_setrole(self):
-        admin = User.query.filter(User.name == 'admin').one_or_none()
+        admin = User.query.filter(User.name == 'admin').one()
         admin.set_role("ADMIN")
         db.session.commit()
-        admin = User.query.filter(User.name == 'admin').one_or_none()
+        admin = User.query.filter(User.name == 'admin').one()
         self.assertTrue(admin is not None)
         self.assertTrue(admin.has_role("ADMIN"))
         self.assertFalse(admin.has_role("SERVICE"))