in einem Kolben unit-test, wie kann ich die mock-Objekte auf Anfrage-global `g` - Objekt?

Ich habe eine flask Anwendung, die den Aufbau einer Datenbank-Verbindung in einem before_filter sehr ähnlich diese:

@app.before_request
def before_request():
    g.db = connect_db()

Nun: Schreibe ich einige unit-tests und ich nicht wollen, dass Sie für die hit-Datenbank. Ich will zu ersetzen g.db mit einem mock-Objekt, dass ich einstellen kann, Erwartungen auf.

Meine tests sind mit app.test_client(), wie gezeigt, in der flask Dokumentation hier. Ein Beispiel test sieht so etwas wie

def test(self):
    response = app.test_client().post('/endpoint', data={..})
    self.assertEqual(response.status_code, 200)
    ...

Den tests arbeiten und zu passieren, aber Sie sind zu schlagen, der Datenbank und als ich sagte ich will zu ersetzen, die den Zugriff auf die db mit mock-Objekten. Ich sehe keinen Weg in test_client Zugriff auf die g Objekt oder ändern Sie die before_filters.

InformationsquelleAutor Gabe Moothart | 2013-01-03
  1. 24

    Dies funktioniert

    test_app.py

    from flask import Flask, g

app = Flask(__name__)

def connect_db():
    print 'I ended up inside the actual function'
    return object()

@app.before_request
def before_request():
    g.db = connect_db()


@app.route('/')
def root():
    return 'Hello, World'

    test.py

    from mock import patch
import unittest

from test_app import app


def not_a_db_hit():
    print 'I did not hit the db'

class FlaskTest(unittest.TestCase):

    @patch('test_app.connect_db')
    def test_root(self, mock_connect_db):
        mock_connect_db.side_effect = not_a_db_hit
        response = app.test_client().get('/')
        self.assertEqual(response.status_code, 200)

if __name__ == '__main__':
    unittest.main()

    Also print out "habe ich nicht getroffen den db' statt 'landete ich im eigentlichen Funktion'. Natürlich werden Sie brauchen, um anzupassen, die verhöhnt, die Ihrem tatsächlichen Anwendungsfall.

    • Dies versuchen (ich hatte zu ändern, um 'flask_module.g.connect_db') gibt eine RuntimeError: working outside of request context
    • Ich schrieb gerade einem vollständigen Beispiel, das habe ich tatsächlich getestet und haben funktioniert.
    • Ich mag die Idee! Der einzige Teil, den ich nicht verstehe, ist g = G(), warum ist das notwendig?
    • Es ist nicht, ich brauchte nur ein Objekt mit einem Attribut db um die tests arbeiten. Vermutlich haben Sie schon-einer von denen 🙂
    • Die g Objekt speziell für Fläschchen und setzt seine Attribute nach jeder Anfrage. Ich glaube nicht, es wäre eine gute Idee, benutzen Sie einfach ein beliebiges Objekt, die benimmt sich nicht so.
    • Ich denke, Sie sind einfach nur Missverständnisse die Beispiel. Wenn ich einen Beitrag Antworten möchte ich eigentlich ausführen. Die global 'g' - Objekt ist irrelevant, weil es ist nicht gepatcht. Statt der Funktion wird gepatcht ist connect_db. Möglicherweise möchten Sie zu einem Spiel mit den mock-und patch-libraries, bevor zufällig downvoting akzeptierte Antworten.
    • Rückblickend auf diese kann ich nicht sehen, wie ich es geschafft zu missverstehen. Leider ist meine Stimme wird gesperrt jetzt. Sorry.
    • Keine Sorge! Vielen Dank für einen anderen Blick auf es trotzdem.
    • Ich Stimme mit @markus-unterwaditzer. Aber es ist sehr einfach zu klären deine Antwort: entfernen Sie Einfach Ihre benutzerdefinierten G Objekt, und fügen Sie g zu deiner Flasche importiert. So sollten Sie from flask import Flask, g. Es wird die Antwort viel klarer!
    • Ich denke also, wir werden sehen, ob es funktioniert.
    • irgendeine Idee, wie test_client umgeht die before_request Prozess?
    • Es ist nicht... das before_request Anruf immer noch durchgeführt wird. Aber der code läuft es gepatcht wurde.

    InformationsquelleAutor aychedee
Schreibe einen Kommentar