Guénaël Muller 7 лет назад
Родитель
Сommit
45d9b3acde
2 измененных файлов: 6 добавлений и 4 удалений
  1. 2 0
      tracim/scripts/initializedb.py
  2. 4 4
      tracim/tests.py

+ 2 - 0
tracim/scripts/initializedb.py Просмотреть файл

43
 
43
 
44
         model = MyModel(name='one', value=1)
44
         model = MyModel(name='one', value=1)
45
         dbsession.add(model)
45
         dbsession.add(model)
46
+
47
+        ### Add global manager data, just for test

+ 4 - 4
tracim/tests.py Просмотреть файл

28
         self.session = get_tm_session(session_factory, transaction.manager)
28
         self.session = get_tm_session(session_factory, transaction.manager)
29
 
29
 
30
     def init_database(self):
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
     def tearDown(self):
34
     def tearDown(self):
35
-        from .models.meta import Base
35
+        from .models.meta import DeclarativeBase
36
 
36
 
37
         testing.tearDown()
37
         testing.tearDown()
38
         transaction.abort()
38
         transaction.abort()
39
-        Base.metadata.drop_all(self.engine)
39
+        DeclarativeBase.metadata.drop_all(self.engine)
40
 
40
 
41
 
41
 
42
 class TestMyViewSuccessCondition(BaseTest):
42
 class TestMyViewSuccessCondition(BaseTest):