def test_get_count(self): """ Tests that Luggages class counts luggages correctly. :return: """ luggages = Luggages() for luggage in self.luggage_list: luggages.add(luggage) self.assertEqual(luggages._get_count(), len(self.luggage_list))
def put(self, rideRequestId, uid): """ Add luggage to the ride request. --- tags: - luggage parameters: - name: id in: path description: ID of the ride request associated with the luggages required: true # schema: # $ref: "#/definitions/LuggageCollection" responses: '200': description: luggages response properties: newLuggageValues: type: array # items: # $ref: "#/definitions/LuggageItem" :param rideRequestId: :param uid: :return: """ # args = ride_request_parsers.luggage_parser.parse_args() json_input = request.get_json() # # Marshmallow version 3 # try: # data = luggages_schema.load(json_input) # except ValidationError as err: # return {'errors': err.messages}, 422 # # Marshmallow version 2 try: data = luggages_schema.load(json_input) except ValidationError as error: return error.messages, 422 luggage_list = data["luggages"] luggages = Luggages() luggages.add_from_list(luggage_list) # add_luggage_nontransactional(rideRequestId, luggages) luggage_actions.put_luggages(ride_request_id=rideRequestId, luggages=luggages) response_dict = {"newLuggageValues": luggages.to_dict()} return response_dict, 200
def test_init(self): """ Tests that Luggages class initializes successfully with no argument :return: """ luggages = Luggages() self.assertIsNotNone(luggages, msg="Luggages did not initialize. ")
def test_to_dict_two_pcs(self): luggages = Luggages() luggages._luggage_list = [self.luggage_list[0], self.luggage_list[1]] expected_dict = { "luggages": [ { "luggage_type": "large", "weight_in_lbs": 20 }, { "luggage_type": "medium", "weight_in_lbs": 15 } ], "total_weight": 35, "total_count": 2 } self.assertDictEqual(expected_dict, luggages.to_dict(), "Luggages did not return correct dict")
def test_to_dict_one_pcs(self): """ Tests that Luggages successfully returns dict with correct count and weight. :return: """ luggages = Luggages() luggages._luggage_list = [self.luggage_list[0]] expected_dict = { "luggages": [ { "luggage_type": "large", "weight_in_lbs": 20 } ], "total_weight": 20, "total_count": 1 } self.assertDictEqual(expected_dict, luggages.to_dict(), "Luggages did not return correct dict")
def put(self, rideRequestId, uid): """ :param rideRequestId: :param uid: :return: """ args = ride_request_parsers.luggage_parser.parse_args() luggage_list = args["luggages"] luggages = Luggages() luggages.add_from_list(luggage_list) # add_luggage_nontransactional(rideRequestId, luggages) luggage_actions.put_luggages(ride_request_id=rideRequestId, luggages=luggages) response_dict = {"newLuggageValues": luggages.to_dict()} return response_dict, 200
def test_from_dict(self): d = { "luggages": [ { "luggage_type": "large", "weight_in_lbs": 20 }, { "luggage_type": "medium", "weight_in_lbs": 15 } ], "total_weight": 35, "total_count": 2 } luggages = Luggages.from_dict(d) self.assertListEqual(luggages._luggage_list, [self.luggage_list[0], self.luggage_list[1]])
def test_get_weight_one(self): luggages = Luggages() luggages.add(self.luggage_list[0]) self.assertEqual(luggages._get_weight(), 20)
def test_get_count_one(self): luggages = Luggages() luggages.add(self.luggage_list[0]) self.assertEqual(luggages._get_count(), 1)
def test_add_many(self): luggages = Luggages() for luggage in self.luggage_list: luggages.add(luggage) self.assertListEqual(luggages._luggage_list, self.luggage_list)
def test_add_one(self): luggages = Luggages() luggages.add(self.luggage_list[0]) self.assertListEqual(luggages._luggage_list, [self.luggage_list[0]])