def test_add_sourcefile_relative_path(self, stat): m = mock.Mock() stat.return_value = m m.st_size = 100 final_filename = "audio/11/1122-3333-4444/mp3/1122-3333-4444-mp3.mp3" sf, created = util.docserver_add_sourcefile(self.doc.id, self.sft.id, final_filename) self.assertEqual("audio/11/1122-3333-4444/mp3/1122-3333-4444-mp3.mp3", sf.path)
def test_add_sourcefile(self, stat): m = mock.Mock() stat.return_value = m m.st_size = 100 sft = models.SourceFileType.objects.get_by_slug("mp3") final_filename = "/root/directory/audio/11/1122-3333-4444/mp3/1122-3333-4444-mp3.mp3" sf, created = util.docserver_add_sourcefile(self.doc.id, self.sft.id, final_filename) self.assertTrue(created) stat.assert_called_with(final_filename) self.assertEqual(self.doc, sf.document) self.assertEqual(sft, sf.file_type) self.assertEqual("11/1122-3333-4444/mp3/1122-3333-4444-mp3.mp3", sf.path)
def test_add_sourcefile_already_exists(self, stat): final_filename = "audio/11/1122-3333-4444/mp3/1122-3333-4444-mp3.mp3" sf = models.SourceFile.objects.create(path=final_filename, size=100, document=self.doc, file_type=self.sft) sfid = sf.id m = mock.Mock() stat.return_value = m m.st_size = 200 new_filename = "/root/directory/audio/something-else.mp3" newsf, created = util.docserver_add_sourcefile(self.doc.id, self.sft.id, new_filename) self.assertFalse(created) self.assertEqual("something-else.mp3", newsf.path) self.assertEqual(200, newsf.size) self.assertEqual(sfid, newsf.id)