def create_sample_data(self):
""" Add some data """
# == Create users
added_date = datetime.today()
access_end_date = utils.get_expiration_date(180)
user = UserEntity.create(email="*****@*****.**",
first="First",
last="Last",
minitial="M",
added_at=added_date,
modified_at=added_date,
email_confirmed_at=added_date,
access_expires_at=access_end_date)
# == Create roles
role_admin = RoleEntity.create(name=ROLE_ADMIN, description='role')
role_tech = RoleEntity.create(name=ROLE_TECHNICIAN, description='role')
role_res1 = RoleEntity.create(name=ROLE_RESEARCHER_ONE, description='')
role_res2 = RoleEntity.create(name=ROLE_RESEARCHER_TWO, description='')
user.roles.extend([role_admin, role_tech, role_res1, role_res2])
# == Create subject
subject = SubjectEntity.create(
redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=0)
# == Create events
evt = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 1',
day_offset=1,
added_at=added_date)
evt2 = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 2',
day_offset=2.3,
added_at=added_date)
self.assertIsNotNone(evt.id)
self.assertIsNotNone(evt2.id)
files = [
{'name': 'a.png', 'size': '123', 'event': evt.id},
{'name': 'b.png', 'size': '1234', 'event': evt.id},
{'name': 'c.png', 'size': '12345', 'event': evt2.id},
{'name': 'd.png', 'size': '123456', 'event': evt2.id},
{'name': 'e.png', 'size': '1234567', 'event': evt2.id},
]
# Create subject files
for fdata in files:
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.id)
self.assertIsNotNone(subject_file.id)
def create_sample_data(self):
""" Add some data """
# == Create users
added_date = datetime.today()
access_end_date = utils.get_expiration_date(180)
user = UserEntity.create(email="*****@*****.**",
first="First",
last="Last",
minitial="M",
added_at=added_date,
modified_at=added_date,
access_expires_at=access_end_date)
# == Create roles
role_admin = RoleEntity.create(name=ROLE_ADMIN, description='role')
role_tech = RoleEntity.create(name=ROLE_TECHNICIAN, description='role')
user.roles.extend([role_admin, role_tech])
# == Create subject
subject = SubjectEntity.create(
redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=0)
# == Create events
evt = EventEntity.create(redcap_arm='Arm 1', redcap_event='Event 1',
added_at=added_date)
evt2 = EventEntity.create(redcap_arm='Arm 1', redcap_event='Event 2',
added_at=added_date)
assert evt.id is not None
files = [
{'name': 'a.png', 'size': '1MB', 'event': evt.id},
{'name': 'b.png', 'size': '2MB', 'event': evt.id},
{'name': 'c.png', 'size': '3MB', 'event': evt2.id},
{'name': 'd.png', 'size': '4MB', 'event': evt2.id},
{'name': 'e.png', 'size': '5MB', 'event': evt2.id},
]
# Create subject files
for fdata in files:
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.id)
app.logger.debug("Init test case with: {}".format(subject_file))
def api_import_redcap_events():
"""
Refresh the list of events
"""
url = app.config['REDCAP_API_URL']
local_events = EventEntity.query.all()
redcap_events = utils.retrieve_redcap_events(
url=url,
token=app.config['REDCAP_API_TOKEN'],
max_time=app.config['REDCAP_CURL_API_MAX_TIME'])
new_events = find_new_events(local_events, redcap_events)
added_date = datetime.today()
inserted_events = []
for id, redcap_event in new_events.iteritems():
evt = EventEntity.create(
redcap_arm='arm_{}'.format(redcap_event['arm_num']),
redcap_event=redcap_event['event_name'],
day_offset=redcap_event['day_offset'],
added_at=added_date
)
inserted_events.append(evt)
details = [i for i in new_events]
LogEntity.redcap_events_imported(session['uuid'],
"Total: {} \n {}"
.format(len(details), details))
return utils.jsonify_success({
'local_events': [i.get_unique_event_name() for i in local_events],
'redcap_events': redcap_events,
'inserted_events': [i.get_unique_event_name() for i in inserted_events],
'api_url': url})
def api_import_redcap_events():
"""
Refresh the list of events
"""
url = app.config['REDCAP_API_URL']
local_events = EventEntity.query.all()
redcap_events = utils.retrieve_redcap_events(
url=url,
token=app.config['REDCAP_API_TOKEN'],
max_time=app.config['REDCAP_CURL_API_MAX_TIME'])
new_events = find_new_events(local_events, redcap_events)
added_date = datetime.today()
inserted_events = []
for id, redcap_event in new_events.iteritems():
evt = EventEntity.create(
redcap_arm='arm_{}'.format(redcap_event['arm_num']),
redcap_event=redcap_event['event_name'],
day_offset=redcap_event['day_offset'],
added_at=added_date
)
inserted_events.append(evt)
details = [i for i in new_events]
LogEntity.redcap_events_imported(session['uuid'],
"Total: {} \n {}"
.format(len(details), details))
return utils.jsonify_success({
'local_events': [i.get_unique_event_name() for i in local_events],
'redcap_events': redcap_events,
'inserted_events': [i.get_unique_event_name() for i in inserted_events],
'api_url': url})
def test_subject(self):
""" verify save and find operations """
added_date = datetime.today()
subject = SubjectEntity.create(
redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=False)
self.assertEquals(1, subject.id)
self.assertEquals("001", subject.redcap_id)
self.assertIsNotNone(subject.added_at)
self.assertIsNotNone(subject.last_checked_at)
self.assertFalse(subject.was_deleted)
self.assertEqual(subject.added_at, subject.last_checked_at)
subjects = SubjectEntity.query.all()
self.assertEquals(1, len(subjects))
# test usage of CRUDMixin.get_by_id() and create()
subject = SubjectEntity.update(subject, redcap_id="002")
self.assertEquals("002", subject.redcap_id)
evt = EventEntity.create(redcap_arm='Arm 1', redcap_event='Event 1',
added_at=added_date)
fdata = {'name': 'a.png', 'size': '1MB', 'event': evt.id}
user = UserEntity.create(email="*****@*****.**",
first="",
last="",
minitial="",
added_at=added_date,
modified_at=added_date,
access_expires_at=added_date)
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.get_id())
self.assertIsNotNone(subject_file.id)
actual_count = SubjectFileEntity.query.count()
self.assertEqual(1, actual_count)
actual_count = SubjectFileEntity.query.filter_by(
event_id=1).count()
self.assertEqual(1, actual_count)
sfile = SubjectFileEntity.query.first()
self.assertEqual(user.id, sfile.user_id)
def test_subject(self):
""" verify save and find operations """
added_date = datetime.today()
subject = SubjectEntity.create(redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=False)
self.assertEquals(1, subject.id)
self.assertEquals("001", subject.redcap_id)
self.assertIsNotNone(subject.added_at)
self.assertIsNotNone(subject.last_checked_at)
self.assertFalse(subject.was_deleted)
self.assertEqual(subject.added_at, subject.last_checked_at)
subjects = SubjectEntity.query.all()
self.assertEquals(1, len(subjects))
# test usage of CRUDMixin.get_by_id() and create()
subject = SubjectEntity.update(subject, redcap_id="002")
self.assertEquals("002", subject.redcap_id)
evt = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 1',
added_at=added_date)
fdata = {'name': 'a.png', 'size': '1MB', 'event': evt.id}
user = UserEntity.create(email="*****@*****.**",
first="",
last="",
minitial="",
added_at=added_date,
modified_at=added_date,
access_expires_at=added_date)
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.get_id())
self.assertIsNotNone(subject_file.id)
actual_count = SubjectFileEntity.query.count()
self.assertEqual(1, actual_count)
actual_count = SubjectFileEntity.query.filter_by(event_id=1).count()
self.assertEqual(1, actual_count)
sfile = SubjectFileEntity.query.first()
self.assertEqual(user.id, sfile.user_id)
def test_default_events(self):
evt = EventEntity.get_by_id(1)
self.assertIsNotNone(evt)
self.assertEquals("Event 1", evt.redcap_event)
with self.assertRaises(MultipleResultsFound):
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1').one()
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1',
redcap_event='Event 2').one()
self.assertEquals(2, evt2.id)
def test_default_events(self):
evt = EventEntity.get_by_id(1)
self.assertIsNotNone(evt)
self.assertEquals("Event 1", evt.redcap_event)
with self.assertRaises(MultipleResultsFound):
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1').one()
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1',
redcap_event='Event 2').one()
self.assertEquals(2, evt2.id)
def test_default_events(self):
""" Verify that we can insert rows to the `Event` table
Note: sample rows are inserted by the parent class
"""
evt = EventEntity.get_by_id(1)
self.assertIsNotNone(evt)
self.assertEquals("Arm 1", evt.redcap_arm)
self.assertEquals("Event 1", evt.redcap_event)
self.assertEquals(1, evt.day_offset)
# print(EventEntity.query.all())
with self.assertRaises(MultipleResultsFound):
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1').one()
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1',
redcap_event='Event 2').one()
self.assertEquals(2, evt2.id)
self.assertEquals(2.3, evt2.day_offset)
def test_default_events(self):
""" Verify that we can insert rows to the `Event` table
Note: sample rows are inserted by the parent class
"""
evt = EventEntity.get_by_id(1)
self.assertIsNotNone(evt)
self.assertEquals("Arm 1", evt.redcap_arm)
self.assertEquals("Event 1", evt.redcap_event)
self.assertEquals(1, evt.day_offset)
# print(EventEntity.query.all())
with self.assertRaises(MultipleResultsFound):
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1').one()
evt2 = EventEntity.query.filter_by(redcap_arm='Arm 1',
redcap_event='Event 2').one()
self.assertEquals(2, evt2.id)
self.assertEquals(2.3, evt2.day_offset)
def create_sample_data(self):
""" Add some data """
# == Create users
added_date = datetime.today()
access_end_date = utils.get_expiration_date(180)
user = UserEntity.create(email="*****@*****.**",
first="First",
last="Last",
minitial="M",
added_at=added_date,
modified_at=added_date,
email_confirmed_at=added_date,
access_expires_at=access_end_date)
# == Create roles
role_admin = RoleEntity.create(name=ROLE_ADMIN, description='role')
role_tech = RoleEntity.create(name=ROLE_TECHNICIAN, description='role')
role_res1 = RoleEntity.create(name=ROLE_RESEARCHER_ONE, description='')
role_res2 = RoleEntity.create(name=ROLE_RESEARCHER_TWO, description='')
user.roles.extend([role_admin, role_tech, role_res1, role_res2])
# == Create subject
subject = SubjectEntity.create(redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=0)
# == Create events
evt = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 1',
day_offset=1,
added_at=added_date)
evt2 = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 2',
day_offset=2.3,
added_at=added_date)
self.assertIsNotNone(evt.id)
self.assertIsNotNone(evt2.id)
files = [
{
'name': 'a.png',
'size': '123',
'event': evt.id
},
{
'name': 'b.png',
'size': '1234',
'event': evt.id
},
{
'name': 'c.png',
'size': '12345',
'event': evt2.id
},
{
'name': 'd.png',
'size': '123456',
'event': evt2.id
},
{
'name': 'e.png',
'size': '1234567',
'event': evt2.id
},
]
# Create subject files
for fdata in files:
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.id)
self.assertIsNotNone(subject_file.id)
def create_sample_data(self):
""" Add some data """
# == Create users
added_date = datetime.today()
access_end_date = utils.get_expiration_date(180)
user = UserEntity.create(email="*****@*****.**",
first="First",
last="Last",
minitial="M",
added_at=added_date,
modified_at=added_date,
access_expires_at=access_end_date)
# == Create roles
role_admin = RoleEntity.create(name=ROLE_ADMIN, description='role')
role_tech = RoleEntity.create(name=ROLE_TECHNICIAN, description='role')
user.roles.extend([role_admin, role_tech])
# == Create subject
subject = SubjectEntity.create(redcap_id="001",
added_at=added_date,
last_checked_at=added_date,
was_deleted=0)
# == Create events
evt = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 1',
added_at=added_date)
evt2 = EventEntity.create(redcap_arm='Arm 1',
redcap_event='Event 2',
added_at=added_date)
assert evt.id is not None
files = [
{
'name': 'a.png',
'size': '1MB',
'event': evt.id
},
{
'name': 'b.png',
'size': '2MB',
'event': evt.id
},
{
'name': 'c.png',
'size': '3MB',
'event': evt2.id
},
{
'name': 'd.png',
'size': '4MB',
'event': evt2.id
},
{
'name': 'e.png',
'size': '5MB',
'event': evt2.id
},
]
# Create subject files
for fdata in files:
subject_file = SubjectFileEntity.create(
subject_id=subject.id,
event_id=fdata['event'],
file_name=fdata['name'],
file_check_sum=utils.compute_text_md5(fdata['name']),
file_size=fdata['size'],
uploaded_at=added_date,
user_id=user.id)
app.logger.debug("Init test case with: {}".format(subject_file))
