Question

我正在为Flask API编写测试，用户可以在其中创建一个帐户。运行测试（单元测试）后，我不断收到此错误。有没有更好的方法来测试该API，即我也可以用来测试GET，PUT和DELETE请求？

TypeError：self.assertEqual（result [“ id”]，“ 4”） “方法”对象不可下标

这是我的数据库设置：

    class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    public_id = db.Column(db.String(50), unique = True)
    name = db.Column(db.String(50))
    password = db.Column(db.String(80))
    admin = db.Column(db.Boolean)

我要测试的API：

@APP.route('/user', methods = ['POST'])
def create_user():

    data = request.get_json(force = True)
    hashedpassword = generate_password_hash(data['password'], method= 'sha256')
    new_user = User(public_id = str(uuid.uuid4), name = data['name'], password = hashedpassword, admin = False)

    db.session.add(new_user)
    db.session.commit()

我的测试：

class testMainmodule(unittest.TestCase):

    def setUp(self):
        APP.testing = True
        self.app = APP.test_client()
        self.data = {"admin": False,
                "id": "4",
                "name": "njati",
                "password": "sha256$xfIUTEIX$6973717971585c3b7ebb593876def4124ff3eb4f8e30c3b43e2c8af20fe64952",
                "public_id": "<function uuid4 at 0x7fb631f93d08>"}

    def post_create_user(self, data = {}):
        if not data:
            data = self.data
        result= self.app.post(path = "/user", data = json.dumps(self.data), content_type = "application/json")
        json_response = json.loads(result.get_data(as_text=True))
        return jsonify(json_response)

    def test_create_user(self):
        result = self.post_create_user

        self.assertEqual(result["id"], "4")
        self.assertEqual(result["public_id"], "<function uuid4 at 0x7fb631f93d08>")
        self.assertTrue(result["admin"], False)
        self.assertEqual(result["name"], "njati")
        self.assertEqual(result["password"], "sha256$xfIUTEIX$6973717971585c3b7ebb593876def4124ff3eb4f8e30c3b43e2c8af20fe64952")
        self.assertEqual(response.status_code, 201)

Answer 1

您的代码未调用create_user，它指向方法create_user。这就是为什么你得到

TypeError: self.assertEqual(result["id"], "4") 'method' object is not subscriptable

尝试

result = self.post_create_user()
self.assertDictEqual(result,self.data)
# You will face an issue with the line below since response is not defined
self.assertEqual(response.status_code, 201)

不是

result = self.post_create_user

Answer 2

也许对@balderman已经给出的答案有一些进一步的阐述，但这是原因：

result["id"]

result是一个保存类方法的变量，方法不是用这种方式。

当您声明result = self.post_create_user时，python假定post_create_user是一个不能用参数调用的属性。但是由于它是一种方法，所以应该改用self.post_create_user（），并且在将其作为参数传递时（不带括号），应该提供第二个参数作为其参数（请参见上述balderman的回答）

测试使用sqlite数据库的Flask API

2 个答案: