def test__date_validation(self): story = Story() story.day = 29 story.month = 2 story.year = 2013 with self.assertRaises(ValidationError): story.clean()
def test_unknown_date_fields(self): story = Story() story.day = 0 story.month = 0 story.year = 2000 story.clean() self.assertEquals(story.day, 0) self.assertEquals(story.month, 0) self.assertEquals(story.year, 2000)
def add_to_db(self, entry, media_folder): new_story = Story() if (entry[6].lower) in ['audio', 'video', 'image', 'pdf']: try: new_story.media_file = File(open(media_folder + "/" + entry[0])) except IOError: self.stderr.write(RED_C + "Media %s not found for story %s." % (entry[0], entry[1]) + \ DEFAULT_C + "\n" ) return else: if entry[6].lower() == "link": new_story.link_url = entry[0] new_story.title = entry[1] new_story.year = entry[2] try: new_story.branch = Branch.objects.get(name=entry[3]) except ObjectDoesNotExist: self.stderr.write(RED_C + "Branch %s not found for story %s." % (entry[3], new_story.title) + DEFAULT_C + "\n") return new_story.user = User.objects.get(pk=1) new_story.content_type = CONTENT_TYPE[entry[6].lower()] new_story.description = entry[7] new_story.public_approved = True print new_story new_story.save() for keyword in entry[5].split(','): new_story.keywords.add(keyword)