def dummy_category():
"""
Generate a dummy category.
"""
category = Category(name=uuid6())
category.save()
return category
def dummy_category():
"""
Generate a dummy category.
"""
category = Category(name=uuid6())
category.save()
return category
class TestArticlePermission(TestCase):
"""
Test that the correct users have access to the correct articles.
"""
def setUp(self):
# Create the request factory
self.factory = RequestFactory()
# Create three users - two normal users and an admin (staff)
self.user1 = User.objects.create_user('user1', '*****@*****.**', 'user1')
self.user2 = User.objects.create_user('user2', '*****@*****.**', 'user2')
self.admin = User.objects.create_user('admin', '*****@*****.**', 'admin')
self.admin.is_staff = True
self.admin.save()
# Create a dummy category for the articles
self.category = Category(name='Test')
self.category.save()
# Create a dummy article for testing
self.article = Article(author=self.user1, category=self.category, title='Test', body='Test', date=now())
def request(self, article, user):
request = self.factory.get(article.get_absolute_url())
request.user = user
return view_article(request, article.id)
def check_permission(self, user, access):
if access:
self.assertEqual(self.request(self.article, user).status_code, 200)
else:
with self.assertRaises(Http404):
self.request(self.article, user)
def test_access_to_draft_article(self):
self.article.status = Article.DRAFT
self.article.save()
self.check_permission(AnonymousUser(), False)
self.check_permission(self.user1, True)
self.check_permission(self.user2, False)
self.check_permission(self.admin, True)
def test_access_to_unapproved_article(self):
self.article.status = Article.UNAPPROVED
self.article.save()
self.check_permission(AnonymousUser(), False)
self.check_permission(self.user1, True)
self.check_permission(self.user2, False)
self.check_permission(self.admin, True)
def test_access_to_published_article(self):
self.article.status = Article.PUBLISHED
self.article.save()
self.check_permission(AnonymousUser(), True)
self.check_permission(self.user1, True)
self.check_permission(self.user2, True)
self.check_permission(self.admin, True)
def setUp(self):
# Create a dummy user, category, and article
user = User.objects.create_user('user', '*****@*****.**', 'user')
category = Category(name='Test')
category.save()
self.article = Article(author=user, category=category, title='Test', body='Test', status=Article.PUBLISHED, date=now())
self.article.save()
# Create a dummy image
o = BytesIO(smart_bytes('x\x9cb`\x01\x00\x00\x00\xff\xff\x03\x00\x00\x06\x00\x05'))
self.image = Image(caption='Test', image=File(o, name='test.png'))
self.image.save()
def setUp(self):
# Create a dummy user, category, and article
user = User.objects.create_user('user', '*****@*****.**', 'user')
category = Category(name='Test')
category.save()
self.article = Article(author=user, category=category, title='Test', body='Test', status=Article.PUBLISHED, date=now())
self.article.save()
# Create a dummy image
o = BytesIO(smart_bytes('x\x9cb`\x01\x00\x00\x00\xff\xff\x03\x00\x00\x06\x00\x05'))
self.image = Image(caption='Test', image=File(o, name='test.png'))
self.image.save()
def setUp(self):
# Create the request factory
self.factory = RequestFactory()
# Create three users - two normal users and an admin (staff)
self.user1 = User.objects.create_user('user1', '*****@*****.**', 'user1')
self.user2 = User.objects.create_user('user2', '*****@*****.**', 'user2')
self.admin = User.objects.create_user('admin', '*****@*****.**', 'admin')
self.admin.is_staff = True
self.admin.save()
# Create a dummy category for the articles
self.category = Category(name='Test')
self.category.save()
# Create a dummy article for testing
self.article = Article(author=self.user1, category=self.category, title='Test', body='Test', date=now())
