單元測試完成後如何刪除圖像文件？

在單元測試中，我在開始時創建了3個對象（文章）。測試完成後，我注意到media_root文件夾中有3個圖像。單元測試完成後如何刪除圖像文件？

問題：如何刪除測試完成後創建的圖像？

P.S.我試圖使用下一個代碼，但它刪除media_root文件夾。

def tearDown(self): 
     rmtree(settings.MEDIA_ROOT, ignore_errors=True)

注：方法test_article_form_valid和test_article_crud問題。

tests.py：

class ArticleTestCase(TestCase): 
    def setUp(self): 
     self.image = open(os.path.join(BASE_DIR, 'static/images/tests/image.jpg'), "r") 

    def test_article_form_valid(self): 
     data = { 
      'head': 'TEXT', 
     } 
     files_data = { 
      'image': SimpleUploadedFile(
       name=self.image.name, 
       content=self.image.read(), 
       content_type='image/jpeg' 
      ) 
     } 
     form = ArticleForm(data=data, files=files_data) 
     self.assertTrue(form.is_valid()) <-- ERROR 

    def test_article_crud(self): 
     response = self.client.get(reverse("article:article_create")) 
     self.assertEquals(response.status_code, 200) 
     response = self.client.post(
      reverse("article:article_create"), 
      data={ 
       'head': 'TEST', 
       'image': self.image 
      }, 
      follow=True, 
      format='multipart' 
     ) 
     self.assertEqual(response.status_code, 200) 
     self.assertEqual(Article.objects.all().count(), 1) <-- ERROR 

    def test_article_view(self): 
     first_article = Article.objects.create(
      pk=150, 
      head='First', 
      image=SimpleUploadedFile(
       name=self.image.name, 
       content=self.image.read(), 
       content_type='image/jpeg' 
      ) 
     ) 

     second_article = Article.objects.create(
      pk=160, 
      head='Second', 
      image=SimpleUploadedFile(
       name=self.image.name, 
       content=self.image.read(), 
       content_type='image/jpeg' 
      ) 
     ) 

     third_article = Article.objects.create(
      pk=170, 
      head='Third', 
      image=SimpleUploadedFile(
       name=self.image.name, 
       content=self.image.read(), 
       content_type='image/jpeg' 
      ) 
     ) 
     [***]

ERROR：

FAIL: test_article_crud (article.tests.ArticleTestCase) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/home/nurzhan/CA/article/tests.py", line 55, in test_article_crud 
    self.assertEqual(Article.objects.all().count(), 1) 
AssertionError: 0 != 1 

====================================================================== 
FAIL: test_article_form_valid (article.tests.ArticleTestCase) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "/home/nurzhan/CA/article/tests.py", line 117, in test_article_form_valid 
    self.assertTrue(form.is_valid()) 
AssertionError: False is not true

來源

2017-09-18 Nurzhan Nogerbek

使用tempfile在python模塊，使用它作爲在TestCasesetUp()方法settings.MEDIA_ROOT，

from django.conf import settings 
import tempfile 

def setUp(self): 
    settings.MEDIA_ROOT = tempfile.mkdtemp()

然後，當測試完成時，測試中創建的文件將被自動刪除。

來源

2017-09-18 07:08:15 zaidfazil

但是OP想要保留目錄，只是刪除其中的文件，我想。 – SiHa

@SiHa絕對正確！我只需要刪除在我的測試中創建的圖像文件。不要觸摸文件夾。你還有其他建議嗎？ –

測試點是模擬某個功能而不影響實際的數據庫和測試，不是嗎？如果使用'tempfile'模塊，則不必創建實際的圖像對象，但爲它們創建一個模擬對象，然後在測試之後進行測試，這些對象將自動刪除。甚至沒有提出刪除文件的問題，也沒有創建文件，它們在運行時被創建和銷燬。 – zaidfazil

單元測試完成後如何刪除圖像文件？

回答

相關問題