DRF test if JSON list of objects contains a specific object

user_3465744 0
Viewed 403

I'm testing an endpoint from Django DRF, which produces a JSON list of objects. I'm trying to check if a specific object is in the returned list. I've tried assertIn and assertContains, but they produce errors.

My endpoint returns list of objects:

[
  {
    'id' : 1,
    'name': 'HelpMe'
  },
  {
    'id' : 2,
    'name': 'WhatToDo'
  },
  {
    'id' : 3,
    'name': 'ToSolveThis'
  },
]

Test code for the assertIn:

    def test_list(self):      
        client = APIClient()

        response = client.get('/api/some_list/', format='json')

        self.assertEqual(
            len(response.data),
            19
        )

        self.assertIn(
            response.data,
            {
                "id": 2,
                "name": "WhatToDo"
            }
        )

assertIn produces TypeError: unhashable type: 'ReturnList'

Test code for the assertContains:

    def test_list(self):      
        client = APIClient()

        response = client.get('/api/some_list/', format='json')

        self.assertEqual(
            len(response.data),
            19
        )

        self.assertContains(
            response,
            {
                "id": 2,
                "name": "WhatToDo"
            }
        )

assertContain just failed the test.

What is the best approach to test if a specific object exists in the JSON list in response?

django-rest-framework pytest testing django python
1 Answers

You have inverted the arguments of: assertIn, try this:

self.assertIn({"id": 2, "name": "WhatToDo"}, response.data)

# Or you can also write:
self.assertTrue({"id": 2, "name": "WhatToDo"} in response.data)
user_167328980
Related