Ejemplo n.º 1
0
    def test_saving_and_retrieving_articles(self):

        #article has title | url | description | newspaper | category
        article_1 = Article()
        article_1.newspaper = "Python Daily"
        article_1.category = "Programming"
        article_1.title = "Python in 10 minutes"
        article_1.url = "http://realpython.com"
        article_1.description = "Learn the Python programming language in 10 minutes"
        article_1.save()

        article_2 = Article()
        article_2.newspaper = "NYT"
        article_2.category = "Programming"
        article_2.title = "Kung Fu React"
        article_2.url = "http://flask.com"
        article_2.description = "Best tutorial to learning react on the web"
        article_2.save()

        saved_articles = Article.objects.all()
        self.assertEqual(saved_articles.count(), 2)

        self.assertEqual(article_1, Article.objects.all()[0])
        self.assertEqual(article_1.newspaper, 'Python Daily')
        self.assertEqual(article_1.category, 'Programming')
        self.assertEqual(article_1.title, 'Python in 10 minutes')
        self.assertEqual(article_1.url, "http://realpython.com")
        self.assertEqual(article_1.description, "Learn the Python programming language in 10 minutes")

        self.assertEqual(article_2, Article.objects.all()[1])
        self.assertEqual(article_2.newspaper, "NYT")
        self.assertEqual(article_2.category, "Programming")
        self.assertEqual(article_2.title, "Kung Fu React")
        self.assertEqual(article_2.url, "http://flask.com")
        self.assertEqual(article_2.description, "Best tutorial to learning react on the web")