En utilisant pytest pour m'assurer qu'un fichier est créé et écrit en

Question

J'utilise pytest et je veux tester qu'une fonction écrit du contenu dans un fichier. J'ai donc writer.py qui comprend:

MY_DIR = '/my/path/' 

def my_function(): 
    with open('{}myfile.txt'.format(MY_DIR), 'w+') as file: 
     file.write('Hello') 
     file.close() 

Je veux tester /my/path/myfile.txt est créé et a le contenu correct:

import writer 

class TestFile(object): 

    def setup_method(self, tmpdir): 
     self.orig_my_dir = writer.MY_DIR 
     writer.MY_DIR = tmpdir 

    def teardown_method(self): 
     writer.MY_DIR = self.orig_my_dir 

    def test_my_function(self): 
     writer.my_function() 

     # Test the file is created and contains 'Hello' 

Mais je suis coincé avec la façon de le faire. Tout ce que j'essaie, comme quelque chose comme:

 import os 
     assert os.path.isfile('{}myfile.txt'.format(writer.MYDIR)) 

qui me génère des erreurs conduisent à penser que je ne suis pas comprendre ou à utiliser tmpdir correctement.

Comment dois-je tester cela? (Si le reste de la façon dont j'utilise pytest est aussi horrible, n'hésitez pas à me le dire aussi!)

Answer 1

J'ai un test pour travailler en modifiant la fonction que je teste afin qu'elle accepte un chemin écrire à. Cela rend plus facile à tester. Alors writer.py est:

MY_DIR = '/my/path/' 

def my_function(my_path): 
    # This currently assumes the path to the file exists. 
    with open(my_path, 'w+') as file: 
     file.write('Hello') 

my_function(my_path='{}myfile.txt'.format(MY_DIR)) 

Et le test:

import writer 

class TestFile(object): 

    def test_my_function(self, tmpdir): 

     test_path = tmpdir.join('/a/path/testfile.txt') 

     writer.my_function(my_path=test_path) 

     assert test_path.read() == 'Hello'