def get_or_create_user(auth_user):
user = User()
# Returns None or List[User] objects - could be more than one
users = user.get_user_by_username(str(auth_user.username))
if users is None:
user.set_user_id(str(uuid.uuid4()))
user.set_user_name(auth_user.name)
user.set_lf_email(auth_user.email.lower())
user.set_lf_username(auth_user.username)
user.set_lf_sub(auth_user.sub)
user.save()
event_data = f'CLA user added for {auth_user.username}'
Event.create_event(
event_data=event_data,
event_type=EventType.CreateUser,
contains_pii=True,
)
return user
# Just return the first matching record
return users[0]
def test_add_project_manager(mock_event, project):
""" Tests event logging where LFID is added to the project ACL """
event_type = EventType.AddProjectManager
username = '******'
lfid = 'manager'
Project.load = Mock()
Project.get_project_name = Mock(return_value=project.get_project_name())
Project.save = Mock()
user = User()
user.set_user_name('foo')
Project.get_managers_by_project_acl = Mock(return_value=[user])
Project.add_project_acl = Mock()
Project.get_project_acl = Mock(return_value=('foo'))
project_controller.add_project_manager(
username,
project.get_project_id(),
lfid
)
event_data = '{} added {} to project {}'.format(username,lfid,project.get_project_name())
mock_event.assert_called_with(
event_type=event_type,
event_data=event_data,
event_summary=event_data,
event_project_id=project.get_project_id(),
contains_pii=True,
)
def user(user_table):
""" create user """
with patch(PATCH_METHOD) as req:
req.return_value = {}
user_instance = User()
user_instance.set_user_id("user_foo_id")
user_instance.set_user_email("*****@*****.**")
user_instance.set_user_name("foo_username")
user_instance.save()
yield user_instance
def user_instance():
"""
Mock user instance
"""
with patch(PATCH_METHOD) as req:
req.return_value = USER_TABLE_DATA
instance = User()
instance.set_user_id("foo")
instance.set_user_name("username")
instance.set_user_external_id("bar")
instance.save()
yield instance
def get_or_create_user(auth_user):
user = User()
existing_user = user.get_user_by_username(str(auth_user.username))
if existing_user is None:
user.set_user_id(str(uuid.uuid4()))
user.set_user_name(auth_user.name)
user.set_lf_email(auth_user.email.lower())
user.set_lf_username(auth_user.username)
user.set_lf_sub(auth_user.sub)
user.save()
return user
return existing_user
def get_or_create_user(auth_user):
user = User()
# Returns None or List[User] objects - could be more than one
users = user.get_user_by_username(str(auth_user.username))
if users is None:
user.set_user_id(str(uuid.uuid4()))
user.set_user_name(auth_user.name)
user.set_lf_email(auth_user.email.lower())
user.set_lf_username(auth_user.username)
user.set_lf_sub(auth_user.sub)
user.save()
return user
# Just return the first matching record
return users[0]