diff --git a/rhodecode/tests/functional/test_admin_notifications.py b/rhodecode/tests/functional/test_admin_notifications.py --- a/rhodecode/tests/functional/test_admin_notifications.py +++ b/rhodecode/tests/functional/test_admin_notifications.py @@ -3,6 +3,7 @@ from rhodecode.model.db import Notificat from rhodecode.model.user import UserModel from rhodecode.model.notification import NotificationModel +from rhodecode.model.meta import Session class TestNotificationsController(TestController): @@ -10,8 +11,8 @@ class TestNotificationsController(TestCo def tearDown(self): for n in Notification.query().all(): inst = Notification.get(n.notification_id) - self.Session().delete(inst) - self.Session().commit() + Session().delete(inst) + Session().commit() def test_index(self): self.log_user() @@ -29,7 +30,7 @@ class TestNotificationsController(TestCo NotificationModel().create(created_by=u1, subject=u'test_notification_1', body=u'notification_1', recipients=[cur_user]) - self.Session().commit() + Session().commit() response = self.app.get(url('notifications')) response.mustcontain(u'test_notification_1') @@ -67,7 +68,7 @@ class TestNotificationsController(TestCo subject=u'test', body=u'hi there', recipients=[cur_user, u1, u2]) - self.Session().commit() + Session().commit() u1 = User.get(u1.user_id) u2 = User.get(u2.user_id)