Skip to content

Instantly share code, notes, and snippets.

@m0nte-cr1st0
Last active January 28, 2020 08:48
class CarViewTestCase(APITestCase):
def setUp(self) -> None:
self.user = UserFactory()
self.client.force_authenticate(user=self.user)
self.car = CarFactory()
def test_car_retrieve(self) -> None:
resp = self.client.get(reverse("Car-detail", kwargs={"uuid": self.car.uuid}))
self.assertEqual(resp.status_code, status.HTTP_200_OK)
class CarFactory(factory.django.DjangoModelFactory):
class Meta:
model = Car
# field = value
class CarsView(mixins.CreateModelMixin,
mixins.RetrieveModelMixin,
mixins.ListModelMixin,
GenericViewSet):
permission_classes = (IsAuthenticated,)
filter_backends = (DjangoFilterBackend,)
filterset_class = CarFilter
lookup_field = 'uuid'
serializer_class = CarSerializer
def get_queryset(self):
user = self.request.user
return Car.objects.filter(donor=user, paid_success=True)
def list(self, request, *args, **kwargs):
serializer = super().list(request, *args, **kwargs)
return Response({'cars': serializer.data})
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment