plans_blueprint_test.py 文件源码

def test_plan_add_valid_body(app, manager):
    url = url_for("plans.add_plan")

    # add a planner and an attack
    manager.attacks_store.add(attack1_module)
    manager.planners_store.add(planner1_module)

    with app.test_request_context(url):

        res = app.test_client().post(url_for("plans.add_plan"),
                                     content_type="application/json",
                                     data=json.dumps(valid_request_body))

        assert res.status_code == 200
        assert res.mimetype == "application/json"
        assert res.json == {"msg": "ok"}