使用url_路径中的参数测试drf genericviewset

dauxcl2d  于 2021-08-20  发布在  Java
关注(0)|答案(1)|浏览(441)

我正在尝试使用unittest测试drf端点。测试视图:

class HwView(mixins.ListModelMixin, viewsets.GenericViewSet):

    @action(detail=False, methods=['get'], url_path='switch_state/(?P<switch_id>\d+)')
    def switch_state(self, request, switch_id):
        print(f'sw: {switch_id}')

        results = {"state": 'ok'}

        return Response(results)

url.py中的条目

from rest_framework import routers
router = routers.DefaultRouter()

router.register(r'hw', hw_views.HwView, basename='HwModel')

以及测试代码

from rest_framework.test import APIRequestFactory, APITestCase

class TestCommandProcessor(APITestCase):
    def setUp(self):
        pass

    def test_switch_state(self):
        factory = APIRequestFactory()
        request = factory.get(
            '/api/hw/switch_state/3',
        )

        view = HwView.as_view({'get': 'switch_state'}, basename='HwModel')

        response = view(request)

        self.assertIn('state', response.data)
        self.assertEqual('ok', response.data['state'])

当我运行测试时,我得到一个错误:

TypeError: switch_state() missing 1 required positional argument: 'switch_id'

此视图中的任何其他方法(get不带参数,post带参数)都可以使用相同的测试方法。
你能帮我找出为什么我的视图不能解析url中的参数吗?或者关于如何重写测试以成功测试代码的任何建议。

axzmvihb

axzmvihb1#

尝试

response = view(request, switch_id=3)

response = view(request, {"switch_id": 3})

相关问题