From 9f9629edb389278eb935663d2111d894613c44de Mon Sep 17 00:00:00 2001 From: Oliver Walters Date: Mon, 1 Mar 2021 11:21:18 +1100 Subject: [PATCH] Fix for unit test --- InvenTree/part/test_api.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/InvenTree/part/test_api.py b/InvenTree/part/test_api.py index 23865aa0f3..faadf26c15 100644 --- a/InvenTree/part/test_api.py +++ b/InvenTree/part/test_api.py @@ -243,10 +243,12 @@ class PartAPITest(InvenTreeAPITestCase): for n in [1, 5, 10]: response = self.get(reverse('api-part-list'), {'limit': n}) - self.assertIn('count', response) - self.assertIn('results', response) + data = response.data + + self.assertIn('count', data) + self.assertIn('results', data) - self.assertEqual(len(response['results']), n) + self.assertEqual(len(data['results']), n) class PartAPIAggregationTest(InvenTreeAPITestCase):