def test_cluster_qs_user_with_userperms(self):
user = self.create_user('cluster_admin')
user.grant('admin', self.cluster)
qs = cluster_qs_for_user(user)
self.assertQuerysetEqual(qs, [repr(self.cluster)])
user.delete()
def test_cluster_qs_on_readonly_cluster(self):
"""
Read only clusters should not be returned when using
cluster_qs_for_user unless you pass readonly=True as a kwarg.
"""
user = self.superuser
# create a read only cluster (no username)
ro_cluster = Cluster.objects.create(hostname="foo.example.org",
slug="foo")
qs = cluster_qs_for_user(user, readonly=False)
self.assertNotIn(ro_cluster, qs)
qs = cluster_qs_for_user(user, readonly=True)
expected_clusters = [repr(self.cluster), repr(ro_cluster)]
self.assertQuerysetEqual(qs, expected_clusters, ordered=False)
# Cleanup
ro_cluster.delete()
def test_cluster_qs_for_user_with_groupperms(self):
group = Group.objects.create(name='cluster_admin')
user = self.create_user('user')
group.grant('admin', self.cluster)
group.user_set.add(user)
qs = cluster_qs_for_user(user)
self.assertQuerysetEqual(qs, [repr(self.cluster)])
user.delete()
group.delete()
def test_cluster_qs_user_without_perms(self):
user = self.create_user('cluster_admin')
qs = cluster_qs_for_user(user)
self.assertFalse(qs)
user.delete()
def test_cluster_qs_for_superuser(self):
user = self.superuser
qs = cluster_qs_for_user(user)
self.assertQuerysetEqual(qs, [repr(self.cluster)], ordered=False)
def test_cluster_qs_for_user_anon(self):
user = AnonymousUser()
qs = cluster_qs_for_user(user)
self.assertFalse(qs)
def get_queryset(self):
self.queryset = cluster_qs_for_user(self.request.user)
qs = super(ClusterListView, self).get_queryset()
qs = qs.select_related("nodes", "virtual_machines")
return qs
