單元測試問題的小瓶應用程序

我有一個非常原始的瓶應用程序，它可以像我期待的那樣工作，但我沒有爲它編寫單元測試。該應用程序的代碼如下（我省略微不足道的一部分）：單元測試問題的小瓶應用程序

app.py 

from flask import * 
import random 
import string 

app = Flask(__name__) 
keys = [] 
app.testing = True 

@app.route('/keygen/api/keys', methods=['POST']) 
def create(): 
    symbol = string.ascii_letters + string.digits 
    key = ''.join(random.choice(symbol) for _ in range(4)) 
    key_instance = {'key': key, 'is_used': False} 
    keys.append(key_instance) 

    return jsonify({'keys': keys}), 201

測試是：

tests.py 

import unittest 
from flask import * 
import app 

class TestCase(unittest.TestCase): 
    def test_number_one(self): 
     test_app = Flask(app) 
     with test_app.test_client() as client: 
      rv = client.post('/keygen/api/keys') 

     ...something... 

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

回溯：

ERROR: test_number_one (__main__.TestCase) 
---------------------------------------------------------------------- 
Traceback (most recent call last): 
    File "tests.py", line 12, in test_number_one 
    test_app = Flask(app) 
    File "/Users/bulrathi/Yandex.Disk.localized/Virtualenvs/ailove/lib/python3.5/site-packages/flask/app.py", line 346, in __init__ 
    root_path=root_path) 
    File "/Users/bulrathi/Yandex.Disk.localized/Virtualenvs/ailove/lib/python3.5/site-packages/flask/helpers.py", line 807, in __init__ 
    root_path = get_root_path(self.import_name) 
    File "/Users/bulrathi/Yandex.Disk.localized/Virtualenvs/ailove/lib/python3.5/site-packages/flask/helpers.py", line 668, in get_root_path 
    filepath = loader.get_filename(import_name) 
    File "<frozen importlib._bootstrap_external>", line 384, in _check_name_wrapper 
ImportError: loader for app cannot handle <module 'app' from '/Users/bulrathi/Yandex.Disk.localized/Обучение/Code/Django practice/ailove/keygen/app.py'> 

---------------------------------------------------------------------- 
Ran 1 test in 0.003s 

FAILED (errors=1)

感謝您的時間。

來源

2016-09-28 mr_bulrathi

在測試中，當'app'已經是'Flask'的一個實例時，''test_app''初始化爲'Flask（app）'，實質上是'test_app = Flask（Flask（app））'。嘗試刪除這一行，並用app.test_client（）作爲客戶端替換'with'行。'。 – kfb

我得到了'AttributeError：模塊'app'沒有'test_client'屬性 –

你也需要在你的測試中添加'app ['TESTING'] = True'。 – kfb

你必須與發佈代碼的幾個問題（壓痕除外）：

首先，在tests.py你import app和使用它，但app是模塊，而不是從app.py的app對象。你應該作爲一個參數導入使用

from app import app

，其次實際app對象，您使用的是app對象（假設我們固定進口）到另一個Flask()構造，基本上等於在說：

app = Flask(Flask(app))

由於我們已經從app.py導入app，我們可以直接使用它，所以我們刪除了app = Flask(app)行（以及相關聯的import聲明，因爲我們不再需要它），並且您的測試文件變爲：

import unittest 
from app import app 

class TestCase(unittest.TestCase): 
    def test_number_one(self): 
     with app.test_client() as client: 
     rv = client.post('/keygen/api/keys') 

    ...something... 

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

你也應該注意到，from flask import *形式有利於導入模塊的特定部分的勸阻，所以

from flask import Flask, jsonify

將是您的app.py更好。

來源

2016-09-28 09:32:38 kfb

單元測試問題的小瓶應用程序

回答

相關問題