Pārlūkot izejas kodu

fix tests sample

Guénaël Muller 7 gadus atpakaļ
vecāks
revīzija
45d9b3acde
2 mainītis faili ar 6 papildinājumiem un 4 dzēšanām
  1. 2 0
      tracim/scripts/initializedb.py
  2. 4 4
      tracim/tests.py

+ 2 - 0
tracim/scripts/initializedb.py Parādīt failu

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 Parādīt failu

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