class TestGetService(TestCase):
def setUp(self):
# Getting started by adding an airplane instance.
self.factory = RequestFactory()
self.view = views.AirplaneList.as_view()
self.uri = 'airplanes/'
self.data = {"airplane_id": 1, "passengers": 100}
self.request = self.factory.post(self.uri,
data=self.data,
format='json')
self.response = self.view(self.request)
# For testing get service.
self.get_obj = GetService()
self.pk = 1
self.get_obj.set_pk(self.pk)
self.result = {
"airplane_id": 1,
"passengers": 100,
"fuel_consumption": 0.20,
"flight_time": 1000.00
}
def test_get_and_set_pk(self):
self.assertEqual(self.get_obj.get_pk(), self.pk)
def test_get_instance_service(self):
self.assertIsInstance(self.get_obj.get_instance_service(), Airplane)
def test_get_service(self):
self.assertEqual(len(self.get_obj.get_service()), 1)
def test_get_detail_service(self):
self.assertEqual(self.get_obj.get_detail_service()['airplane_id'], 1)
def update_service(self):
get_service_obj = GetService()
get_service_obj.set_pk(self.pk)
airplane = get_service_obj.get_instance_service()
serializer = AirplaneSerializer(airplane,
data=self.request.data,
partial=True)
if serializer.is_valid():
serializer.save()
return serializer.data, status.HTTP_200_OK
return serializer.errors, status.HTTP_400_BAD_REQUEST
def delete_service(self):
get_service_obj = GetService()
get_service_obj.set_pk(self.pk)
airplane = get_service_obj.get_instance_service()
airplane.delete()
return status.HTTP_204_NO_CONTENT