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)
