def test_get_from_sourceDB(self): lesson = Lesson.get_from_sourceDB('New York', self.user) self.assertIsInstance(lesson, Lesson) self.assertEqual(lesson.addedBy, self.user) self.assertEqual(lesson.title, 'New York') self.assertEqual(lesson.sourceDB, 'wikipedia') self.assertIsNotNone(lesson.commitTime) self.assertTrue(Lesson.objects.filter(addedBy=self.user).exists())
def test_get_from_sourceDB_noSave_wiki_user(self): user = User.objects.create_user(username='******', password='******') lesson = Lesson.get_from_sourceDB('New York', user, doSave=False) self.assertIsInstance(lesson, Lesson) self.assertEqual(lesson.addedBy, user) self.assertEqual(lesson.title, 'New York') self.assertEqual(lesson.sourceDB, 'wikipedia') self.assertIsNotNone(lesson.commitTime) self.assertFalse(Lesson.objects.filter(addedBy=self.user).exists())
def test_sourceDB_temp(self): """ Check wikipedia temporary document retrieval. """ lesson = Lesson.get_from_sourceDB( 'New York City', self.user, doSave=False ) self.assertIn('City of New York', lesson.text) # got the text? self.assertEqual(Lesson.objects.count(), 0) # nothing saved?