django Einstellung Umgebungsvariablen in Unittest Tests

Question

Ich möchte in meiner Django-App Umgebungsvariablen für Tests setzen können, um ausgeführt werden zu können. Zum Beispiel sind meine Ansichten auf mehrere API-Schlüssel angewiesen.

Es gibt Möglichkeiten zu override settings during testing, aber ich möchte nicht, dass sie in settings.py als ein Sicherheitsproblem definiert sind.

Ich habe in meiner Setup-Funktion versucht, diese Umgebungsvariablen zu setzen, aber das funktioniert nicht, um der Django-Anwendung die Werte zu geben.

class MyTests(TestCase): 
    def setUp(self): 
     os.environ['TEST'] = '123' # doesn't propogate to app 

Als ich vor Ort testen, habe ich einfach eine .env Datei, die ich mit

foreman start -e .env web 

laufen die os.environ mit Werten versorgt. Aber in Djangos unittest.TestCase hat es keinen Weg (den ich kenne), das zu setzen.

Wie kann ich das umgehen?

Answer 1

Wie @schillingt in den Kommentaren notiert wurde, war EnvironmentVarGuard der richtige Weg.

from test.test_support import EnvironmentVarGuard 
from django.test import TestCase 

class MyTestCase(TestCase): 
    def setUp(self): 
     self.env = EnvironmentVarGuard() 
     self.env.set('VAR', 'value') 

    def test_something(self): 
     with self.env: 
      # ... perform tests here ... # 
      pass 

Dies setzt die richtigen Umgebungsvariablen für die Dauer der Kontextobjekt with Aussage.

Answer 2

Ich verwende py.test als meine Test-Runner, und es ermöglicht Ihnen, eine pytest.ini Datei zu erstellen, in der Sie eine bestimmte Einstellungsdatei angeben können, während Tests ausgeführt werden.

Siehe Dokumentation zu diesem Thema hier:

http://pytest-django.readthedocs.org/en/latest/configuring_django.html#pytest-ini-settings

Ich empfehle py.test allgemein als Testläufer, weil es verschiedene Arten von Testklassen und sogar einfacher Funktionen unterstützt, und es ist ziemlich einfach zu installieren up Fixtures oder anderen Code, der vor und nach Tests ausgeführt wird.

Answer 3

Alte Frage, aber es erschien in einer Google-Suche und keine der vorhandenen Antworten sind geeignet. Wenn Sie pytest verwenden, kann env vars mit pytest's monkeypatching functionality eingestellt/wiederhergestellt werden.

Answer 4

Die test.support.EnvironmentVarGuard ist eine interne API, die von Version zu Version mit brechenden (rückwärtskompatiblen) Änderungen geändert werden kann. In der Tat ist das gesamte test Paket nur für den internen Gebrauch. Auf der Dokumentationsseite des Testpakets wurde ausdrücklich angegeben, dass es sich um ein internes Testen von Kernbibliotheken und KEINE öffentliche API handelt. (siehe Links unten)

Sie sollten patch.dict() in Python Standard lib unittest.mock verwenden. Es kann als Kontextmanager, Dekorator oder Klassendekorator verwendet werden. Siehe den folgenden Beispielcode, der aus der offiziellen Python-Dokumentation kopiert wurde.

import os 
from unittest.mock import patch 
with patch.dict('os.environ', {'newkey': 'newvalue'}): 
    print(os.environ['newkey']) # should print out 'newvalue' 
    assert 'newkey' in os.environ # should be True 
assert 'newkey' not in os.environ # should be True 

Update: für diejenigen, die nicht in der Dokumentation gründlich liest und könnte die Note verpasst haben, lesen Sie mehr test Paket Noten

https://docs.python.org/2/library/test.html oder

https://docs.python.org/3/library/test.html

Answer 5

Wenn Sie Laden Sie Ihre Umgebungsvariablen in Django Datei wie folgt:
import os ENV_NAME = os.environ.get('ENV_NAME', 'default')
Sie könnten so:
from django.test import TestCase, override_settings @override_settings(ENV_NAME="super_setting") def test_...(self):