Homec4science
Diffusion invenio-infoscience (maint-2.1) Browse

R3600/modules/webmessage/lib/webmessage_flask_tests.py
a80bc4185b7emaint-2.1

Last Change
Raw File

webmessage_flask_tests.py
View Options

# -*- coding: utf-8 -*-
#
## This file is part of Invenio.
## Copyright (C) 2012, 2013 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
from invenio.sqlalchemyutils import db
from invenio.webmessage_config import CFG_WEBMESSAGE_STATUS_CODE
from invenio.testutils import make_flask_test_suite, run_test_suite, \
FlaskSQLAlchemyTest, InvenioFixture
from fixture import SQLAlchemyFixture
from invenio.webaccount_fixtures import UserData
def fixture_builder():
from invenio.websession_model import User
return SQLAlchemyFixture(env={'UserData': User}, engine=db.metadata.bind,
session=db.session)
fixture = InvenioFixture(fixture_builder)
class MsgMESSAGETest(FlaskSQLAlchemyTest):
@fixture.with_data(UserData)
def test_index(data, self):
from invenio.websession_model import User
from invenio.webmessage_model import MsgMESSAGE, UserMsgMESSAGE
users = data.UserData
admin = User.query.filter(User.nickname.like('admin')).one()
romeo = User.query.filter(User.nickname.like('romeo')).one()
juliet = User.query.filter(User.nickname.like('juliet')).one()
m = MsgMESSAGE()
m.subject = 'Message 1 Subject'
m.body = 'body'
try:
m.sent_to_user_nicks = 'admin, NOTEXISTS'
except:
assert m.sent_to_user_nicks is None
m.sent_to_user_nicks = ', '.join([users.admin.nickname,
users.romeo.nickname])
try:
m.sent_to_user_nicks = 'admin, NOTEXISTS'
except:
# Remove unnecessary space
assert m.sent_to_user_nicks == ','.join([users.admin.nickname,
users.romeo.nickname])
m.sent_to_group_names = ''
m.id_user_from = users.admin.id
db.session.add(m)
db.session.commit()
m = MsgMESSAGE.query.first()
assert admin in m.recipients
assert romeo in m.recipients
assert juliet not in m.recipients
response = self.login(users.admin.nickname, users.admin.password)
assert users.admin.nickname in response.data
response = self.client.get("/yourmessages/", follow_redirects=True)
assert 'Message 1 Subject' in response.data
self.logout()
response = self.login(users.romeo.nickname, users.romeo.password)
assert users.romeo.nickname in response.data
response = self.client.get("/yourmessages/", follow_redirects=True)
assert 'Message 1 Subject' in response.data
self.logout()
response = self.login(users.juliet.nickname, users.juliet.password)
assert users.juliet.nickname in response.data
response = self.client.get("/yourmessages/", follow_redirects=True)
assert 'Message 1 Subject' not in response.data
self.logout()
# Avoid problems with foreign keys.
UserMsgMESSAGE.query.delete()
MsgMESSAGE.query.delete()
db.session.commit()
@fixture.with_data(UserData)
def test_send_later(data, self):
from invenio.websession_model import User
from invenio.webmessage_model import MsgMESSAGE, UserMsgMESSAGE
users = data.UserData
romeo = User.query.filter(User.nickname.like('romeo')).one()
#response = self.login('romeo', '')
import time
from datetime import datetime, timedelta
sent = datetime.now()
received = sent + timedelta(seconds=5)
m = MsgMESSAGE()
m.subject = 'sent_now'
m.body = 'body'
m.sent_to_user_nicks = 'admin'
m.sent_to_group_names = ''
m.id_user_from = users.romeo.id
m.sent_date = sent
db.session.add(m)
#db.session.commit()
m = MsgMESSAGE()
m.subject = 'sent_later'
m.body = 'body'
m.sent_to_user_nicks = 'admin'
m.sent_to_group_names = ''
m.id_user_from = romeo.id
m.sent_date = sent
m.received_date=received#.strftime('%Y-%m-%d %H:%M:%S')
for um in m.sent_to_users:
um.status = CFG_WEBMESSAGE_STATUS_CODE['REMINDER']
db.session.add(m)
db.session.commit()
#response = self.client.post('/yourmessages/add',
# #base_url=request.base_url.replace('http:','https:'),
# data=dict(
# sent_to_user_nicks='admin',
# sent_to_group_names=' ',
# subject='sent_later',
# body='test',
# received_date=received.strftime('%Y-%m-%d %H:%M:%S')
# ), follow_redirects=True,
# headers=[('X-Requested-With', 'XMLHttpRequest')])
#self.logout()
response = self.login(users.admin.nickname, users.admin.password)
response = self.client.get("/yourmessages/", follow_redirects=True)
assert 'sent_now' in response.data
assert 'sent_later' not in response.data
while datetime.now() <= received:
time.sleep(1)
time.sleep(1)
response = self.client.get("/yourmessages/", follow_redirects=True)
assert 'sent_now' in response.data
assert 'sent_later' in response.data
# Avoid problems with foreign keys.
UserMsgMESSAGE.query.delete(synchronize_session='fetch')
MsgMESSAGE.query.delete(synchronize_session='fetch')
db.session.commit()
@fixture.with_data(UserData)
def test_with_fixture(data, self):
from invenio.websession_model import User
u = User.query.all()
assert len(u) == len(dict(data.UserData))
db.session.commit()
def test_without_fixture(self):
from invenio.websession_model import User
u = User.query.all()
assert len(u) == 0
TEST_SUITE = make_flask_test_suite(MsgMESSAGETest)
if __name__ == "__main__":
run_test_suite(TEST_SUITE)
c4science · Help