python - 使用特定 IP 的 Django REST 客户端测试

Question

Django REST docs从META读取地址。所以我必须创建Whitelist模型来存储它的IP地址/子网。没什么大不了的，我的数据库中已经有了所有 302 子网。

from rest_framework import permissions

class BlacklistPermission(permissions.BasePermission):
    """
    Global permission check for blacklisted IPs.
    """

    def has_permission(self, request, view):
        ip_addr = request.META['REMOTE_ADDR']
        blacklisted = Blacklist.objects.filter(ip_addr=ip_addr).exists()
        return not blacklisted

问题:
我想从其他 IP 地址测试白名单，而不是 127.0.0.1
我怎样才能在pytest中做到这一点？

Answer 1

感谢@rpkilby
https://github.com/encode/django-rest-framework/issues/5775#issuecomment-360734150

因为APICLient继承自相同的源。设置标题非常简单。

client = APIClient(REMOTE_ADDR='x.x.x.x')