test_mirador_resource.py

import json

from Mirador_backend.tests.tester import TestCase


class MiradorResourceTest(TestCase):
    def getData(self):
        return self.getFixtureRecordsByModel(self.fixtures[0], 'Mirador_backend.models.MiradorResource')

    def testGetOne(self):
        id = 1
        payload = json.dumps({})
        response = self.client.get(f'{self.base}/{id}', headers={"Content-Type": "application/json"}, data=payload)

        self.assertEqual(200, response.status_code)

        expected = next(x for x in self.getData() if x['id'] == id)
        self.assertEqual(expected, response.json)

    def testGetAll(self):
        payload = json.dumps({})
        response = self.client.get(self.base, headers={"Content-Type": "application/json"}, data=payload)
        self.assertEqual(200, response.status_code)
        self.assertEqual({str(x['id']): x for x in self.getData()}, response.json)