class TestNewsArticleViews(TestCase):
"""Tests the functions attached to news article views"""
def setUp(self):
self.article = ArticleFactory(publish=True)
self.request_factory = RequestFactory()
self.url = self.article.get_absolute_url()
self.view = NewsDetail.as_view()
def post_helper(self, data, user):
"""Returns a post response"""
request = self.request_factory.post(self.url, data)
request.user = user
request = mock_middleware(request)
pub_date = timezone.localtime(self.article.pub_date)
return self.view(request,
slug=self.article.slug,
year=pub_date.strftime('%Y'),
month=pub_date.strftime('%b').lower(),
day=pub_date.strftime('%d'))
def test_set_tags(self):
"""Posting a group of tags to an article should set the tags on that article."""
tags = "foo, bar, baz"
staff = UserFactory(is_staff=True)
response = self.post_helper({'tags': tags}, staff)
self.article.refresh_from_db()
ok_(response.status_code, 200)
ok_('foo' in [tag.name for tag in self.article.tags.all()])
ok_('bar' in [tag.name for tag in self.article.tags.all()])
ok_('baz' in [tag.name for tag in self.article.tags.all()])
def test_set_projects(self):
"""Posting a group of projects to an article should set that article's projects."""
project1 = ProjectFactory()
project2 = ProjectFactory()
staff = UserFactory(is_staff=True)
project_form = ProjectManagerForm(
{'projects': [project1.pk, project2.pk]},
user=staff,
)
ok_(project_form.is_valid(),
'We want to be sure we are posting valid data.')
data = {'action': 'projects'}
data.update(project_form.data)
response = self.post_helper(data, staff)
self.article.refresh_from_db()
project1.refresh_from_db()
project2.refresh_from_db()
ok_(response.status_code, 200)
ok_(self.article in project1.articles.all(),
'The article should be added to the project.')
ok_(self.article in project2.articles.all(),
'The article should be added to teh project.')
def test_staff_only(self):
"""Non-staff users cannot edit articles."""
user = UserFactory()
response = self.post_helper({'tags': 'hello'}, user)
eq_(response.status_code, 403,
'The server should return a 403 Forbidden error code.')
def test_manager_get_published(self):
"""Test the Article Manager's get_published method"""
article1 = ArticleFactory(publish=True)
article2 = ArticleFactory(publish=True)
published = Article.objects.get_published()
ok_(article1 in published and article2 in published)
ok_(all(a.publish and a.pub_date <= timezone.now() for a in published))
eq_(published.count(), 2)
def test_articles(self):
"""Projects should keep a list of relevant articles."""
article1 = ArticleFactory()
article2 = ArticleFactory()
self.project.articles.add(article1, article2)
ok_(article1 in self.project.articles.all())
ok_(article2 in self.project.articles.all())
self.project.articles.clear()
eq_(len(self.project.articles.all()), 0)
def setUp(self):
"""Create articles for the tests"""
ArticleFactory.create_batch(
2,
publish=True,
pub_date=datetime(1999, 1, 1, 12, 12, tzinfo=pytz.utc),
)
ArticleFactory(publish=True,
pub_date=datetime(1999, 1, 2, 12, tzinfo=pytz.utc))
ArticleFactory(publish=True,
pub_date=datetime(1999, 2, 3, 12, tzinfo=pytz.utc))
ArticleFactory(publish=True,
pub_date=datetime(2000, 6, 6, tzinfo=pytz.utc))
def test_news_archive_author(self):
"""Should return all articles for the given author"""
author = UserFactory()
ArticleFactory.create_batch(
3,
publish=True,
authors=[author],
pub_date=timezone.now() - timedelta(1),
)
response = get_allowed(
self.client,
reverse('news-author', kwargs={
'username': author.username
})
)
eq_(
len(response.context['object_list']),
Article.objects.filter(authors=author).count()
)
def setUp(self):
AgencyFactory()
ArticleFactory()
CrowdsourceResponseFactory()
FOIARequestFactory()
FlaggedTaskFactory()
NewAgencyTaskFactory()
OrphanTaskFactory()
QuestionFactory()
ResponseTaskFactory()
SnailMailTaskFactory()
UserFactory()
def test_news_detail(self):
"""News detail should display the given article"""
article = ArticleFactory(
publish=True,
pub_date=datetime(1999, 1, 1, 12, tzinfo=pytz.utc),
)
response = get_allowed(
self.client,
reverse('news-detail',
kwargs={
'year': 1999,
'month': 'jan',
'day': 1,
'slug': article.slug,
}))
eq_(response.context['object'], article)
class TestNewsUnit(TestCase):
"""Unit tests for news"""
def setUp(self):
"""Set up tests"""
self.article = ArticleFactory()
# models
def test_article_model_str(self):
"""Test the Article model's __str__ method"""
ok_(str(self.article))
def test_article_model_url(self):
"""Test the Article model's get_absolute_url method"""
pub_date = timezone.localtime(self.article.pub_date)
eq_(
self.article.get_absolute_url(),
reverse(
"news-detail",
kwargs={
"year": pub_date.strftime("%Y"),
"month": pub_date.strftime("%b").lower(),
"day": pub_date.strftime("%d"),
"slug": self.article.slug,
},
),
)
# manager
def test_manager_get_published(self):
"""Test the Article Manager's get_published method"""
article1 = ArticleFactory(publish=True)
article2 = ArticleFactory(publish=True)
published = Article.objects.get_published()
ok_(article1 in published and article2 in published)
ok_(all(a.publish and a.pub_date <= timezone.now() for a in published))
eq_(published.count(), 2)
def test_manager_get_drafts(self):
"""Test the Article Manager's get_drafts method"""
drafted = Article.objects.get_drafts()
ok_(self.article in drafted)
ok_(all(not a.publish for a in drafted))
eq_(drafted.count(), 1)
class TestNewsUnit(TestCase):
"""Unit tests for news"""
def setUp(self):
"""Set up tests"""
self.article = ArticleFactory()
# models
def test_article_model_unicode(self):
"""Test the Article model's __unicode__ method"""
ok_(unicode(self.article))
def test_article_model_url(self):
"""Test the Article model's get_absolute_url method"""
eq_(
self.article.get_absolute_url(),
reverse(
'news-detail',
kwargs={
'year': self.article.pub_date.strftime('%Y'),
'month': self.article.pub_date.strftime('%b').lower(),
'day': self.article.pub_date.strftime('%d'),
'slug': self.article.slug
}
)
)
# manager
def test_manager_get_published(self):
"""Test the Article Manager's get_published method"""
article1 = ArticleFactory(publish=True)
article2 = ArticleFactory(publish=True)
published = Article.objects.get_published()
ok_(article1 in published and article2 in published)
ok_(all(a.publish and a.pub_date <= timezone.now() for a in published))
eq_(published.count(), 2)
def test_manager_get_drafts(self):
"""Test the Article Manager's get_drafts method"""
drafted = Article.objects.get_drafts()
ok_(self.article in drafted)
ok_(all(not a.publish for a in drafted))
eq_(drafted.count(), 1)
def test_suggest_articles(self):
"""
Projects should recommend articles to be added to them.
They should recommend articles that intersect both the
project's set of contributors and the project's set of tags.
But projects should not recommend articles that they already contain.
"""
# set up data
tags = u'a'
user = UserFactory()
self.project.contributors.add(user)
self.project.tags.add(tags)
test_article = ArticleFactory()
test_article.authors.add(user)
test_article.tags.add(tags)
# since they have the same user and tags, the project should suggest the article.
ok_(test_article in self.project.suggest_articles())
# add the article to the project, then try again. it should not be suggested
self.project.articles.add(test_article)
ok_(test_article not in self.project.suggest_articles())
def test_news_detail(self):
"""News detail should display the given article"""
article = ArticleFactory(publish=True,
pub_date=datetime(1999,
1,
1,
12,
tzinfo=pytz.utc))
response = get_allowed(
self.client,
reverse(
"news-detail",
kwargs={
"year": 1999,
"month": "jan",
"day": 1,
"slug": article.slug
},
),
)
eq_(response.context["object"], article)
def setUp(self):
"""Set up tests"""
self.article = ArticleFactory()
def setUp(self):
self.article = ArticleFactory(publish=True)
self.request_factory = RequestFactory()
self.url = self.article.get_absolute_url()
self.view = NewsDetail.as_view()