ソースを参照

fix tests sample

Guénaël Muller 7 年 前
コミット
45d9b3acde
共有2 個のファイルを変更した6 個の追加4 個の削除を含む
  1. 2 0
      tracim/scripts/initializedb.py
  2. 4 4
      tracim/tests.py

+ 2 - 0
tracim/scripts/initializedb.py ファイルの表示

@@ -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 ファイルの表示

@@ -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):