单元测试Flask应用程序类

时间:2012-06-19 23:48:14

标签: python unit-testing flask nose

我对Python很陌生,所以如果对于基本的东西来说这很麻烦,我会提前道歉。

我的情况与How do you set up a Flask application with SQLAlchemy for testing?类似。对我而言,最大的区别在于,与我在互联网上看到的大多数其他Flask示例不同,我的应用程序的大多数代码都在一个类中。出于某种原因,这导致我的单元测试无法正常工作。以下是我的应用程序和测试的基本设置:

应用

from Flask import Flask

app = Flask(__name__)

class MyApplication():
    def __init__(self, param1, param2):
        app.add_url("/path/<methodParam>", "method1", self.method1, methods=["POST"])
        # Initialize the app

    def getApplication(self):
       options = # application configuration options
       middleware = ApplicationMiddleware(app, options)
       return middleware

    def method1(self, methodParam):
        # Does useful stuff that should be tested
    # More methods, etc.

应用程序测试:

import unittest
from apppackage import MyApplication

class ApplicationTestCase(unittest.TestCase):

    def setUp(self):
        self.tearDown()
        param1 = # Param values
        param2 = # Param values
        # Other local setup 
        self.app = MyApplication(param1, param2).app.test_client()

    def tearDown(self):
       # Clean up tests

    def test_method1(self):
        methodParam = # Param value
        response = self.app.post("path/methodParam")
        assert(reponse.status_code == 200)

当我通过

运行测试时
  

nosetests --with-coverage --cover-package apppackage   ./test/test_application.py

我收到以下错误:

  

param2).app.test_client()AttributeError:MyApplication实例有   没有属性'app'

我已经尝试在类声明中移动应用程序,但这没有任何好处,并且不是我见过的所有其他单元测试指南的做法。为什么我的单元测试找不到“app”属性?

1 个答案:

答案 0 :(得分:2)

您的单元测试无法找到“app”属性,因为MyApplication没有。模块中有一个“app”属性,其中定义了MyApplication。但这是两个不同的地方。

也许尝试以下方法:

class MyApplication(object):
    def __init__(self, param1, param2):
        self.app = Flask(__name__)
        self.app.add_url("/path/<methodParam>", "method1", self.method1, methods=["POST"])
        # Initialize the app

或者,你似乎也有一个“getApplication”方法,你并没有真正做任何事情,但我想你正在使用它。也许你真的想在你的测试中想到这个......

def setUp(self):
        self.tearDown()
        param1 = # Param values
        param2 = # Param values
        # Other local setup 
        self.app = MyApplication(param1, param2).getApplication().test_client()