在Flask中测试经过修饰的API路由

Question

I have the following simple controller

@companies.route('/companies/<int:id>', methods=['GET'])
@inject_repository(CompanyRepository)
def get_company(repository, id):
    """
    GET /companies/id
    """
    company = repository.get(pk=id)
    return jsonify(company.to_json())

The DAO is being injected by @inject_repository(CompanyRepository)

I do not, however, get what is happening at test time:

def test_get_company(self):

    response = self.client.get(url_for('companies.get_company', id=1))
    data = json.loads(response.get_data(as_text=True))

    assert_equal(response.status_code, 200)

I get the following error

TypeError: get_company() missing 1 required positional argument: 'id'

Isn't the controller being decorated?

---

Decorator code

def inject_repository(repository_class):

    def decorator(func):

        @wraps(func)
        def wrapper(*args, **kwargs):

            if 'psql' in class_init_fields(repository_class):
                repository = psql_repository_for(
                    repository_model=repository_class,
                    connection_meta=current_app.config['POSTGRESQL'])

            else:
                raise ValueError("'{}' repository cannot be injected".format(repository_class.__name__))

            return func(repository=repository)

        return wrapper

    return decorator

Answer 1

You're not passing the arguments on from your decorator to the decorated function.

def wrapper(*args, **kwargs):
    ...
    kwargs['repository'] = repository
    return func(*args, **kwargs)