def test_model_invitations(self):
#
# Create objects.
#
group = Model.DbGroup("InvGroup")
role = Model.DbRole("student")
user = Model.DbUser("InvUser", "Name of Invitation User", "*****@*****.**", None, role)
invitation = Model.DbInvitation(group, 5, True)
invitation_accept = Model.DbAcceptedInvitation(invitation, user, True)
def test_model(self):
#
# Create objects
#
auth_type = Model.DbAuthType("DB")
permission_type = permissions.EXPERIMENT_ALLOWED
permission_type_p1 = permissions.EXPERIMENT_PERMANENT_ID
permission_type_p2 = permissions.EXPERIMENT_CATEGORY_ID
permission_type_p3 = permissions.TIME_ALLOWED
auth = Model.DbAuth(auth_type, "WebLab DB", 1)
role = Model.DbRole("administrator")
user = Model.DbUser("admin1", "Name of administrator 1",
"*****@*****.**", None, role)
group = Model.DbGroup("5A")
user_permission = Model.DbUserPermission(
user, permission_type, "student2::weblab-pld",
datetime.datetime.utcnow(),
"Permission for student2 to use WebLab-PLD")
user_permission_p1 = Model.DbUserPermissionParameter(
user_permission, permission_type_p1, "ud-pld")
user_permission_p2 = Model.DbUserPermissionParameter(
user_permission, permission_type_p2, "PLD experiments")
user_permission_p3 = Model.DbUserPermissionParameter(
user_permission, permission_type_p3, "100")
group_permission = Model.DbGroupPermission(
group, permission_type, "5A::weblab-dummy",
datetime.datetime.utcnow(),
"Permission for group 5A to use WebLab-Dummy")
group_permission_p1 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p1, "ud-dummy")
group_permission_p2 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p2, "Dummy experiments")
group_permission_p3 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p3, "300")
role_permission = Model.DbRolePermission(
role, permission_type, "administrator::weblab-dummy",
datetime.datetime.utcnow(),
"Permission for administrator to use WebLab-Dummy")
role_permission_p1 = Model.DbRolePermissionParameter(
role_permission, permission_type_p1, "ud-dummy")
role_permission_p2 = Model.DbRolePermissionParameter(
role_permission, permission_type_p2, "Dummy experiments")
role_permission_p3 = Model.DbRolePermissionParameter(
role_permission, permission_type_p3, "300")
#
# Method __repr__()
#
variables = locals()
variables.pop("self")
for k in variables.keys():
repr(eval(k))
#
# Method DbAuth.get_config_value()
#
auth.configuration = "param1=value1;param2=value2"
self.assertEquals("value1", auth.get_config_value("param1"))
self.assertEquals("value2", auth.get_config_value("param2"))
#
# Method Db(User|Group|Role)Permission.get_permission_type()
#
self.assertEquals(permission_type,
user_permission.get_permission_type())
self.assertEquals(permission_type,
group_permission.get_permission_type())
self.assertEquals(permission_type,
role_permission.get_permission_type())
#
# Method Db(User|Group|Role)Permission.get_parameter()
#
self.assertEquals(
user_permission_p1,
user_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
user_permission_p2,
user_permission.get_parameter("experiment_category_id"))
self.assertEquals(user_permission_p3,
user_permission.get_parameter("time_allowed"))
self.assertEquals(
group_permission_p1,
group_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
group_permission_p2,
group_permission.get_parameter("experiment_category_id"))
self.assertEquals(group_permission_p3,
group_permission.get_parameter("time_allowed"))
self.assertEquals(
role_permission_p1,
role_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
role_permission_p2,
role_permission.get_parameter("experiment_category_id"))
self.assertEquals(role_permission_p3,
role_permission.get_parameter("time_allowed"))
#
# Method Db(User|Group|Role)PermissionParameter.get_name()
#
self.assertEquals("experiment_permanent_id",
user_permission_p1.get_name())
self.assertEquals("experiment_category_id",
user_permission_p2.get_name())
self.assertEquals("time_allowed", user_permission_p3.get_name())
self.assertEquals("experiment_permanent_id",
group_permission_p1.get_name())
self.assertEquals("experiment_category_id",
group_permission_p2.get_name())
self.assertEquals("time_allowed", group_permission_p3.get_name())
self.assertEquals("experiment_permanent_id",
role_permission_p1.get_name())
self.assertEquals("experiment_category_id",
role_permission_p2.get_name())
self.assertEquals("time_allowed", role_permission_p3.get_name())
#
# Method Db(User|Group|Role)PermissionParameter.get_datatype()
#
self.assertEquals("string", user_permission_p1.get_datatype())
self.assertEquals("string", user_permission_p2.get_datatype())
self.assertEquals("float", user_permission_p3.get_datatype())
self.assertEquals("string", group_permission_p1.get_datatype())
self.assertEquals("string", group_permission_p2.get_datatype())
self.assertEquals("float", group_permission_p3.get_datatype())
self.assertEquals("string", role_permission_p1.get_datatype())
self.assertEquals("string", role_permission_p2.get_datatype())
self.assertEquals("float", role_permission_p3.get_datatype())
def test_model(self):
#
# Create objects
#
auth_type = Model.DbAuthType("DB")
permission_type = Model.DbPermissionType(
'experiment_allowed',
'This type has a parameter which is the permanent ID (not a INT) of an Experiment. Users which have this permission will have access to the experiment defined in this parameter',
user_applicable=True,
group_applicable=True,
role_applicable=True,
ee_applicable=True)
permission_type_p1 = Model.DbPermissionTypeParameter(
permission_type, 'experiment_permanent_id', 'string',
'the unique name of the experiment')
permission_type_p2 = Model.DbPermissionTypeParameter(
permission_type, 'experiment_category_id', 'string',
'the unique name of the category of experiment')
permission_type_p3 = Model.DbPermissionTypeParameter(
permission_type, 'time_allowed', 'float',
'Time allowed (in seconds)')
auth = Model.DbAuth(auth_type, "WebLab DB", 1)
role = Model.DbRole("administrator")
user = Model.DbUser("admin1", "Name of administrator 1",
"*****@*****.**", None, role)
ee = Model.DbExternalEntity("ee1", "Country of ee1",
"Description of ee1", "*****@*****.**",
"password")
group = Model.DbGroup("5A")
user_permission = Model.DbUserPermission(
user, permission_type.group_applicable, "student2::weblab-pld",
datetime.datetime.utcnow(),
"Permission for student2 to use WebLab-PLD")
user_permission_p1 = Model.DbUserPermissionParameter(
user_permission, permission_type_p1, "ud-pld")
user_permission_p2 = Model.DbUserPermissionParameter(
user_permission, permission_type_p2, "PLD experiments")
user_permission_p3 = Model.DbUserPermissionParameter(
user_permission, permission_type_p3, "100")
group_permission = Model.DbGroupPermission(
group, permission_type.group_applicable, "5A::weblab-dummy",
datetime.datetime.utcnow(),
"Permission for group 5A to use WebLab-Dummy")
group_permission_p1 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p1, "ud-dummy")
group_permission_p2 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p2, "Dummy experiments")
group_permission_p3 = Model.DbGroupPermissionParameter(
group_permission, permission_type_p3, "300")
role_permission = Model.DbRolePermission(
role, permission_type.role_applicable,
"administrator::weblab-dummy", datetime.datetime.utcnow(),
"Permission for administrator to use WebLab-Dummy")
role_permission_p1 = Model.DbRolePermissionParameter(
role_permission, permission_type_p1, "ud-dummy")
role_permission_p2 = Model.DbRolePermissionParameter(
role_permission, permission_type_p2, "Dummy experiments")
role_permission_p3 = Model.DbRolePermissionParameter(
role_permission, permission_type_p3, "300")
ee_permission = Model.DbExternalEntityPermission(
ee, permission_type.ee_applicable, "ee1::weblab-dummy",
datetime.datetime.utcnow(),
"Permission for ee1 to use WebLab-Dummy")
ee_permission_p1 = Model.DbExternalEntityPermissionParameter(
ee_permission, permission_type_p1, "ud-dummy")
ee_permission_p2 = Model.DbExternalEntityPermissionParameter(
ee_permission, permission_type_p2, "Dummy experiments")
ee_permission_p3 = Model.DbExternalEntityPermissionParameter(
ee_permission, permission_type_p3, "300")
#
# Method __repr__()
#
variables = locals()
variables.pop("self")
for k in variables.keys():
repr(eval(k))
#
# Method DbAuth.get_config_value()
#
auth.configuration = "param1=value1;param2=value2"
self.assertEquals("value1", auth.get_config_value("param1"))
self.assertEquals("value2", auth.get_config_value("param2"))
#
# Method DbPermissionType.get_parameter()
#
self.assertEquals(
permission_type_p1,
permission_type.get_parameter("experiment_permanent_id"))
self.assertEquals(
permission_type_p2,
permission_type.get_parameter("experiment_category_id"))
self.assertEquals(permission_type_p3,
permission_type.get_parameter("time_allowed"))
#
# Method Db(User|Group|Role|ExternalEntity)Permission.get_permission_type()
#
self.assertEquals(permission_type,
user_permission.get_permission_type())
self.assertEquals(permission_type,
group_permission.get_permission_type())
self.assertEquals(permission_type,
role_permission.get_permission_type())
self.assertEquals(permission_type, ee_permission.get_permission_type())
#
# Method Db(User|Group|Role|ExternalEntity)Permission.get_parameter()
#
self.assertEquals(
user_permission_p1,
user_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
user_permission_p2,
user_permission.get_parameter("experiment_category_id"))
self.assertEquals(user_permission_p3,
user_permission.get_parameter("time_allowed"))
self.assertEquals(
group_permission_p1,
group_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
group_permission_p2,
group_permission.get_parameter("experiment_category_id"))
self.assertEquals(group_permission_p3,
group_permission.get_parameter("time_allowed"))
self.assertEquals(
role_permission_p1,
role_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
role_permission_p2,
role_permission.get_parameter("experiment_category_id"))
self.assertEquals(role_permission_p3,
role_permission.get_parameter("time_allowed"))
self.assertEquals(
ee_permission_p1,
ee_permission.get_parameter("experiment_permanent_id"))
self.assertEquals(
ee_permission_p2,
ee_permission.get_parameter("experiment_category_id"))
self.assertEquals(ee_permission_p3,
ee_permission.get_parameter("time_allowed"))
#
# Method Db(User|Group|Role|ExternalEntity)PermissionParameter.get_name()
#
self.assertEquals("experiment_permanent_id",
user_permission_p1.get_name())
self.assertEquals("experiment_category_id",
user_permission_p2.get_name())
self.assertEquals("time_allowed", user_permission_p3.get_name())
self.assertEquals("experiment_permanent_id",
group_permission_p1.get_name())
self.assertEquals("experiment_category_id",
group_permission_p2.get_name())
self.assertEquals("time_allowed", group_permission_p3.get_name())
self.assertEquals("experiment_permanent_id",
role_permission_p1.get_name())
self.assertEquals("experiment_category_id",
role_permission_p2.get_name())
self.assertEquals("time_allowed", role_permission_p3.get_name())
self.assertEquals("experiment_permanent_id",
ee_permission_p1.get_name())
self.assertEquals("experiment_category_id",
ee_permission_p2.get_name())
self.assertEquals("time_allowed", ee_permission_p3.get_name())
#
# Method Db(User|Group|Role|ExternalEntity)PermissionParameter.get_datatype()
#
self.assertEquals("string", user_permission_p1.get_datatype())
self.assertEquals("string", user_permission_p2.get_datatype())
self.assertEquals("float", user_permission_p3.get_datatype())
self.assertEquals("string", group_permission_p1.get_datatype())
self.assertEquals("string", group_permission_p2.get_datatype())
self.assertEquals("float", group_permission_p3.get_datatype())
self.assertEquals("string", role_permission_p1.get_datatype())
self.assertEquals("string", role_permission_p2.get_datatype())
self.assertEquals("float", role_permission_p3.get_datatype())
self.assertEquals("string", ee_permission_p1.get_datatype())
self.assertEquals("string", ee_permission_p2.get_datatype())
self.assertEquals("float", ee_permission_p3.get_datatype())
def insert_required_initial_data(engine):
session = sessionmaker(bind=engine)
session._model_changes = {}
session = session()
# Roles
federated = model.DbRole("federated")
session.add(federated)
administrator = model.DbRole("administrator")
session.add(administrator)
instructor = model.DbRole("instructor")
session.add(instructor)
student = model.DbRole("student")
session.add(student)
db = model.DbAuthType("DB")
session.add(db)
ldap = model.DbAuthType("LDAP")
session.add(ldap)
iptrusted = model.DbAuthType("TRUSTED-IP-ADDRESSES")
session.add(iptrusted)
facebook = model.DbAuthType("FACEBOOK")
session.add(facebook)
openid = model.DbAuthType("OPENID")
session.add(openid)
weblab_db = model.DbAuth(db, "WebLab DB", 1)
session.add(weblab_db)
session.commit()
weblab_openid = model.DbAuth(openid, "OPENID", 7)
session.add(weblab_openid)
session.commit()
federated_access_forward = model.DbRolePermission(
federated,
permissions.ACCESS_FORWARD,
"federated_role::access_forward",
datetime.datetime.utcnow(),
"Access to forward external accesses to all users with role 'federated'"
)
session.add(federated_access_forward)
session.commit()
administrator_admin_panel_access = model.DbRolePermission(
administrator,
permissions.ADMIN_PANEL_ACCESS,
"administrator_role::admin_panel_access",
datetime.datetime.utcnow(),
"Access to the admin panel for administrator role with full_privileges"
)
session.add(administrator_admin_panel_access)
administrator_admin_panel_access_p1 = model.DbRolePermissionParameter(administrator_admin_panel_access, permissions.FULL_PRIVILEGES, True)
session.add(administrator_admin_panel_access_p1)
administrator_access_all_labs = model.DbRolePermission(
administrator,
permissions.ACCESS_ALL_LABS,
"administrator_role::access_all_labs",
datetime.datetime.utcnow(),
"Access all the laboratories"
)
session.add(administrator_access_all_labs)
upgrader = DbRegularUpgrader(str(engine.url))
session.execute(
model.Base.metadata.tables['alembic_version'].insert().values(version_num = upgrader.head)
)
session.commit()
def apply(self, session):
federated = Model.DbRole("federated")
session.add(federated)