Comment fournir stdin, aux fichiers et variable d'environnement entrées de Python tests unitaires?

Tous les trois situations que vous avez décrites sont là où vous en avez besoin spécifiquement de sortir de votre chemin pour vous assurer que vous utilisez le couplage lâche dans votre conception.

Ne vous vraiment besoin de test de l'unité Python raw_input méthode? Le open méthode? os.environ.get? No.

Vous devez configurer votre conception de sorte que vous pouvez les remplacer par d'autres moyens de récupération de l'entrée. Ensuite, lors de vos tests unitaires, vous allez lancer dans une souche de la sorte, qui ne fait pas appel raw_input ou open.

Par exemple, votre code peut être quelque chose comme:

import os
def say_hello(input_func):
    name = input_func()
    return "Hello " + name

def prompt_for_name():
    return raw_input("What is your name? ")

print say_hello(prompt_for_name)
# Normally would pass in methods, but lambdas can be used for brevity
print say_hello(lambda: open("a.txt").readline())
print say_hello(lambda: os.environ.get("USER"))

La session ressemble:

Quel est votre nom? quelqu'un 
Bonjour quelqu'un 
Bonjour [texte] 

Bonjour marc 

Alors votre test sera comme:

def test_say_hello():
    output = say_hello(lambda: "test")
    assert(output == "Hello test")

Garder à l'esprit que vous ne devriez pas avoir à tester une langue IO installations (sauf si vous êtes l'un de la conception de la langue, qui est une autre situation tout à fait).

Si vous pouvez vous en sortir sans l'aide d'un processus externe, le faire.

Cependant, il y a des situations où c'est compliqué, et vous voulez vraiment utiliser, par exemple, vous souhaitez tester l'interface de ligne de commande d'un C exécutable.

La saisie de l'utilisateur

Utilisation subprocess.Popen comme dans:

process = subprocess.Popen(
    command,
    shell  = False,
    stdin  = subprocess.PIPE,
    stdout = subprocess.PIPE,
    stderr = subprocess.PIPE,
    universal_newlines = True
)
stdout, stderr = process.communicate("the user input\nline 2")
exit_status = process.wait()

Il n'y a pas de différence entre la prise d'entrée d'un utilisateur et de le prendre à partir d'un tuyau pour la saisie effectuée avec des méthodes comme raw_input ou sys.stdin.read().

Fichiers

• Créer un répertoire temporaire et de créer les fichiers que vous souhaitez lire à partir de là sur votre test setUp méthodes:

  tdir = tempfile.mkdtemp(
      prefix = 'filetest_', 
  )
  fpath = os.path.join(tdir,'filename')
  fp = open(fpath, 'w')
  fp.write("contents")
  fp.close()

• Faire la lecture du fichier dans les tests.

• Supprimer la temp dir par la suite.

  shutil.rmtree(tdir)

• Il est assez compliqué de lecture de fichiers, et la plupart des programmes peuvent lire soit à partir de fichiers ou à partir de l'entrée standard (par exemple, avec fileinput). Donc, si vous voulez tester c'est ce qui arrive quand une certaine contenu est entrée, et votre programme accepte l'entrée standard STDIN, il suffit d'utiliser Popen pour tester le programme.

Variables d'environnement

• Définir les variables d'environnement avec os.environ["THE_VAR"] = "the_val"
• Unset avec del os.environ["THE_VAR"]
• os.environ = {'a':'b'} ne fonctionne pas
• Alors appel subprocess.Popen. L'environnement est hérité du processus appelant.

Code Du Modèle

J'ai un module sur mon github que les tests STDOUT, STDERR et le statut de sortie donné STDIN, arguments de ligne de commande et de l'environnement. Vérifiez également les tests de ce module dans le cadre du "tests" dir. Il doit y avoir beaucoup mieux de modules pour ce, afin de prendre le mien, juste à des fins d'apprentissage.