Skip to content
GitLab
Projects
Groups
Snippets
Help
Loading...
Help
What's new
10
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Sign in
Toggle navigation
Open sidebar
lscsoft
gracedb
Commits
f96b0397
Commit
f96b0397
authored
Jun 24, 2019
by
Tanner Prestegard
Committed by
GraceDB
Jun 26, 2019
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Comment out robotuser group management tests
parent
ef5a108b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
100 additions
and
100 deletions
+100
-100
gracedb/ligoauth/tests/test_middleware.py
gracedb/ligoauth/tests/test_middleware.py
+100
-100
No files found.
gracedb/ligoauth/tests/test_middleware.py
View file @
f96b0397
...
...
@@ -490,106 +490,106 @@ class TestShibbolethWebAuthMiddleware(GraceDbTestBase):
self
.
assertFalse
(
request
.
user
.
groups
.
filter
(
pk
=
new_group
.
pk
).
exists
())
def
test_robotuser_group_addition
(
self
):
"""
Shib group header content is not used to add groups for a robotuser
"""
# Create a robot user account
r_user
=
User
.
objects
.
create
(
username
=
'robot.user'
)
r_user
.
groups
.
add
(
self
.
internal_group
)
r_user
.
groups
.
add
(
self
.
robot_group
)
# Create new group for testing
new_group
=
AuthGroup
.
objects
.
create
(
name
=
'new_group'
,
ldap_name
=
'new_ldap_group'
)
# Compile group header
delim
=
ShibbolethWebAuthMiddleware
.
group_delimiter
groups_str
=
delim
.
join
([
self
.
internal_group
.
ldap_name
,
new_group
.
ldap_name
])
# Set up request
request
=
self
.
factory
.
get
(
self
.
url
)
request
.
META
.
update
(
**
{
settings
.
SHIB_USER_HEADER
:
r_user
.
username
,
settings
.
SHIB_GROUPS_HEADER
:
groups_str
,
})
# Make sure user just has internal and robot groups initially
self
.
assertEqual
(
r_user
.
groups
.
count
(),
2
)
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
internal_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
robot_group
.
pk
).
exists
())
# Necessary pre-processing middleware
SessionMiddleware
().
process_request
(
request
)
AuthenticationMiddleware
().
process_request
(
request
)
# Process request
self
.
mw_instance
.
process_request
(
request
)
# Make sure user is authenticated and was authenticated by
# the shibboleth backend and that the group memberships are
# unchanged
self
.
assertTrue
(
request
.
user
.
is_authenticated
)
self
.
assertEqual
(
request
.
user
.
backend
,
'ligoauth.backends.ShibbolethRemoteUserBackend'
)
self
.
assertEqual
(
r_user
.
groups
.
count
(),
2
)
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
internal_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
robot_group
.
pk
).
exists
())
self
.
assertFalse
(
r_user
.
groups
.
filter
(
pk
=
new_group
.
pk
).
exists
())
def
test_robotuser_group_removal
(
self
):
"""
Shib group header content is not used to remove groups for a robotuser
"""
# Create a robot user account
r_user
=
User
.
objects
.
create
(
username
=
'robot.user'
)
r_user
.
groups
.
add
(
self
.
internal_group
)
r_user
.
groups
.
add
(
self
.
robot_group
)
# Create new group and add robotuser
new_group
=
AuthGroup
.
objects
.
create
(
name
=
'new_group'
,
ldap_name
=
'new_ldap_group'
)
r_user
.
groups
.
add
(
new_group
)
# Set up request
# Shib session doesn't have new_group in it
request
=
self
.
factory
.
get
(
self
.
url
)
request
.
META
.
update
(
**
{
settings
.
SHIB_USER_HEADER
:
r_user
.
username
,
settings
.
SHIB_GROUPS_HEADER
:
self
.
internal_group
.
ldap_name
,
})
# Make sure user has three groups initially
self
.
assertEqual
(
r_user
.
groups
.
count
(),
3
)
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
internal_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
robot_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
new_group
.
pk
).
exists
())
# Necessary pre-processing middleware
SessionMiddleware
().
process_request
(
request
)
AuthenticationMiddleware
().
process_request
(
request
)
# Process request
self
.
mw_instance
.
process_request
(
request
)
# Make sure user is authenticated and was authenticated by
# the shibboleth backend and that the group memberships are
# unchanged
self
.
assertTrue
(
request
.
user
.
is_authenticated
)
self
.
assertEqual
(
request
.
user
.
backend
,
'ligoauth.backends.ShibbolethRemoteUserBackend'
)
self
.
assertEqual
(
r_user
.
groups
.
count
(),
3
)
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
internal_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
self
.
robot_group
.
pk
).
exists
())
self
.
assertTrue
(
r_user
.
groups
.
filter
(
pk
=
new_group
.
pk
).
exists
())
#
def test_robotuser_group_addition(self):
#
"""
#
Shib group header content is not used to add groups for a robotuser
#
"""
#
# Create a robot user account
#
r_user = User.objects.create(username='robot.user')
#
r_user.groups.add(self.internal_group)
#
r_user.groups.add(self.robot_group)
#
# Create new group for testing
#
new_group = AuthGroup.objects.create(name='new_group',
#
ldap_name='new_ldap_group')
#
# Compile group header
#
delim = ShibbolethWebAuthMiddleware.group_delimiter
#
groups_str = delim.join([self.internal_group.ldap_name,
#
new_group.ldap_name])
#
# Set up request
#
request = self.factory.get(self.url)
#
request.META.update(**{
#
settings.SHIB_USER_HEADER: r_user.username,
#
settings.SHIB_GROUPS_HEADER: groups_str,
#
})
#
# Make sure user just has internal and robot groups initially
#
self.assertEqual(r_user.groups.count(), 2)
#
self.assertTrue(r_user.groups.filter(
#
pk=self.internal_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=self.robot_group.pk).exists())
#
# Necessary pre-processing middleware
#
SessionMiddleware().process_request(request)
#
AuthenticationMiddleware().process_request(request)
#
# Process request
#
self.mw_instance.process_request(request)
#
# Make sure user is authenticated and was authenticated by
#
# the shibboleth backend and that the group memberships are
#
# unchanged
#
self.assertTrue(request.user.is_authenticated)
#
self.assertEqual(request.user.backend,
#
'ligoauth.backends.ShibbolethRemoteUserBackend')
#
self.assertEqual(r_user.groups.count(), 2)
#
self.assertTrue(r_user.groups.filter(
#
pk=self.internal_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=self.robot_group.pk).exists())
#
self.assertFalse(r_user.groups.filter(
#
pk=new_group.pk).exists())
#
def test_robotuser_group_removal(self):
#
"""
#
Shib group header content is not used to remove groups for a robotuser
#
"""
#
# Create a robot user account
#
r_user = User.objects.create(username='robot.user')
#
r_user.groups.add(self.internal_group)
#
r_user.groups.add(self.robot_group)
#
# Create new group and add robotuser
#
new_group = AuthGroup.objects.create(name='new_group',
#
ldap_name='new_ldap_group')
#
r_user.groups.add(new_group)
#
# Set up request
#
# Shib session doesn't have new_group in it
#
request = self.factory.get(self.url)
#
request.META.update(**{
#
settings.SHIB_USER_HEADER: r_user.username,
#
settings.SHIB_GROUPS_HEADER: self.internal_group.ldap_name,
#
})
#
# Make sure user has three groups initially
#
self.assertEqual(r_user.groups.count(), 3)
#
self.assertTrue(r_user.groups.filter(
#
pk=self.internal_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=self.robot_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=new_group.pk).exists())
#
# Necessary pre-processing middleware
#
SessionMiddleware().process_request(request)
#
AuthenticationMiddleware().process_request(request)
#
# Process request
#
self.mw_instance.process_request(request)
#
# Make sure user is authenticated and was authenticated by
#
# the shibboleth backend and that the group memberships are
#
# unchanged
#
self.assertTrue(request.user.is_authenticated)
#
self.assertEqual(request.user.backend,
#
'ligoauth.backends.ShibbolethRemoteUserBackend')
#
self.assertEqual(r_user.groups.count(), 3)
#
self.assertTrue(r_user.groups.filter(
#
pk=self.internal_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=self.robot_group.pk).exists())
#
self.assertTrue(r_user.groups.filter(
#
pk=new_group.pk).exists())
def
test_user_update
(
self
):
"""Test user information update in middleware"""
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment