from kallithea.model.user import UserModel

from kallithea.model.meta import Session

from kallithea.model.notification import NotificationModel

class TestNotifications(TestController):

    def setup_method(self, method):

        Session.remove()

        u1 = UserModel().create_or_update(username=u'u1',

                                        password=u'qweqwe',

                                        email=u'u1@example.com',

                                        firstname=u'u1', lastname=u'u1')

        Session().commit()

        self.u1 = u1.user_id

        u2 = UserModel().create_or_update(username=u'u2',

                                        password=u'qweqwe',

                                        email=u'u2@example.com',

                                        firstname=u'u2', lastname=u'u3')

        Session().commit()

        self.u2 = u2.user_id

                                        password=u'qweqwe',

                                        email=u'u3@example.com',

                                        firstname=u'u3', lastname=u'u3')

        Session().commit()

        self.remove_all_notifications()

        assert [] == Notification.query().all()

        assert [] == UserNotification.query().all()

    def test_create_notification(self):

        usrs = [self.u1, self.u2]

        notification = NotificationModel().create(created_by=self.u1,

                                           subject=u'subj', body=u'hi there',

                                           recipients=usrs)

        Session().commit()

        u1 = User.get(self.u1)

        u2 = User.get(self.u2)

        u3 = User.get(self.u3)

        notifications = Notification.query().all()

        assert len(notifications) == 1

        assert notifications[0].recipients == [u1, u2]

        assert notification.notification_id == notifications[0].notification_id

        unotification = UserNotification.query() \

            .filter(UserNotification.notification == notification).all()

        assert len(unotification) == len(usrs)