test_views.py

class LembagaTestViews(APITestCase):

    def setUp(self):
        user_full = User.objects.create_user(
            username="username2",
            email="username2@email.com",
            password="password",
            first_name="User",
            last_name="Name"
        )
        self.client.force_authenticate(user_full)
        Lembaga.objects.create(nama_lembaga="Shopee", deskripsi="Shopee-pee-pee")
        Lembaga.objects.create(nama_lembaga="Tokopedia", deskripsi="Toped-pedia-pedia")
        Lembaga.objects.create(nama_lembaga="Bukalapak", deskripsi="Bulakapak-bulakapak")

    def test_get_spesific_lembaga_by_nama(self):
        response = self.client.get(reverse('get_lembaga_view_by_nama', args=['Shopee']))
        status_code = status.HTTP_200_OK
        # Serialize the data from db
        lembaga = Lembaga.objects.get(nama_lembaga="Shopee")
        serializer = LembagaSerializer(lembaga)
        expected_response = {
            'success': 'true',
            'message': 'Lembaga {} fetched successfully'.format(lembaga.get_nama()),
            'data': serializer.data
        }
        self.assertEqual(response.data, expected_response)
        self.assertEqual(response.status_code, status_code)

    def test_get_spesific_lembaga_by_nama_not_found(self):
        response = self.client.get(reverse('get_lembaga_view_by_nama', args=['Blibli']))
        lembaga = Lembaga.objects.filter(nama_lembaga="Blibli")
        self.assertEqual(response.status_code, status.HTTP_404_NOT_FOUND)

    def test_get_all_lembaga(self):
        response = self.client.get(reverse('get_all_lembaga'))
        status_code = status.HTTP_200_OK

        # Serialize the data from db
        lembaga = Lembaga.objects.all()
        serializer = LembagaSerializer(lembaga, many=True)

        expected_response = {
            'success': 'true',
            'message': 'All lembaga\'s fetched successfully',
            'data': serializer.data
        }

        self.assertEqual(response.data, expected_response)
        self.assertEqual(response.status_code, status_code)