class ArticleTestClass(TestCase):
def setUp(self):
# Creating a new editor and saving it
self.sammy = Editor(first_name='Sammy',
last_name='Mutua',
email='*****@*****.**')
self.sammy.save_editor()
# Creating a new tag and saving it
self.new_tag = tags(name='testing')
self.new_tag.save()
self.new_article = Article(title='Test Article',
post='This is a random test Post',
editor=self.sammy)
self.new_article.save()
self.new_article.tags.add(self.new_tag)
def tearDown(self):
Editor.objects.all().delete()
tags.objects.all().delete()
Article.objects.all().delete()
def test_get_news_today(self):
today_news = Article.todays_news()
self.assertTrue(len(today_news) > 0)
def test_get_news_by_date(self):
test_date = '2017-03-17'
date = dt.datetime.strptime(test_date, '%Y-%m-%d').date()
news_by_date = Article.days_news(date)
self.assertTrue(len(news_by_date) == 0)
class EditorTestClass(TestCase):
#set up method
def setUp(self):
self.sammy = Editor(first_name='Sammy',
last_name='Mutua',
email='*****@*****.**')
#Test Instance
def test_instance(self):
self.assertTrue(isinstance(self.sammy, Editor))
#Testing Save Method
def test_save_method(self):
self.sammy.save_editor()
editors = Editor.objects.all()
self.assertTrue(len(editors) > 0)
