test_models.py 文件源码

def test_can_delete_images(self):
        media_file = SimpleUploadedFile("test.png", b"\x00\x01\x02\x03")
        image = MediaFile(mediatitle="test", mediafile=media_file)
        image.save()
        self.assertEqual(MediaFile.objects.all().count(), 1)
        self.assertIn(
         datetime.now().strftime("%Y%m%d%H%M%S") + ".png",
         os.listdir(MEDIA_ROOT)
        )

        image.delete()
        self.assertEqual(MediaFile.objects.all().count(), 0)
        time.sleep(1)
        self.assertNotIn(
         datetime.now().strftime("%Y%m%d%H%M%S") + ".png",
         os.listdir(MEDIA_ROOT)
        )