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 get_role(self, role):
""" helper for creating roles """
roles = {
ROLE_ADMIN: RoleEntity(name=ROLE_ADMIN, description='role'),
ROLE_TECHNICIAN: RoleEntity(name=ROLE_TECHNICIAN,
description='role'),
}
return roles[role]
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 test_role(self):
""" verify save and find operations """
role_admin = self.get_role(ROLE_ADMIN)
role_tech = self.get_role(ROLE_TECHNICIAN)
db.session.add(role_admin)
# before commit the primary key does not exist
self.assertIsNone(role_admin.id)
# after commit the primary key is generated
db.session.commit()
assert 1 == role_admin.id
roles = RoleEntity.query.all()
assert len(roles) == 1
# test usage of CRUDMixin.get_by_id() and create()
role_tech = RoleEntity.save(role_tech)
assert 2 == role_tech.id
ordered = RoleEntity.query.order_by(RoleEntity.name).all()
# verify that we have two rows now
assert 2 == len(ordered)
# verify that the attribute is saved
first = ordered[0]
assert ROLE_ADMIN == first.name
# verify that sorting worked
first = RoleEntity.query.order_by(db.desc(RoleEntity.name)).first()
assert ROLE_TECHNICIAN == first.name
def test_role(self):
""" verify save and find operations """
role_admin = self.get_role(ROLE_ADMIN)
role_tech = self.get_role(ROLE_TECHNICIAN)
db.session.add(role_admin)
# before commit the primary key does not exist
self.assertIsNone(role_admin.id)
# after commit the primary key is generated
db.session.commit()
assert 1 == role_admin.id
roles = RoleEntity.query.all()
assert len(roles) == 1
# test usage of CRUDMixin.get_by_id() and create()
role_tech = RoleEntity.save(role_tech)
assert 2 == role_tech.id
ordered = RoleEntity.query.order_by(RoleEntity.name).all()
# verify that we have two rows now
assert 2 == len(ordered)
# verify that the attribute is saved
first = ordered[0]
assert ROLE_ADMIN == first.name
# verify that sorting worked
first = RoleEntity.query.order_by(db.desc(RoleEntity.name)).first()
assert ROLE_TECHNICIAN == first.name
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))
