Browse Source

fix tests sample

Guénaël Muller 7 years ago
parent
commit
45d9b3acde
2 changed files with 6 additions and 4 deletions
  1. 2 0
      tracim/scripts/initializedb.py
  2. 4 4
      tracim/tests.py

+ 2 - 0
tracim/scripts/initializedb.py View File

@@ -43,3 +43,5 @@ def main(argv=sys.argv):
43 43
 
44 44
         model = MyModel(name='one', value=1)
45 45
         dbsession.add(model)
46
+
47
+        ### Add global manager data, just for test

+ 4 - 4
tracim/tests.py View File

@@ -28,15 +28,15 @@ class BaseTest(unittest.TestCase):
28 28
         self.session = get_tm_session(session_factory, transaction.manager)
29 29
 
30 30
     def init_database(self):
31
-        from .models.meta import Base
32
-        Base.metadata.create_all(self.engine)
31
+        from .models.meta import DeclarativeBase
32
+        DeclarativeBase.metadata.create_all(self.engine)
33 33
 
34 34
     def tearDown(self):
35
-        from .models.meta import Base
35
+        from .models.meta import DeclarativeBase
36 36
 
37 37
         testing.tearDown()
38 38
         transaction.abort()
39
-        Base.metadata.drop_all(self.engine)
39
+        DeclarativeBase.metadata.drop_all(self.engine)
40 40
 
41 41
 
42 42
 class TestMyViewSuccessCondition(BaseTest):