def update_borrower(borrower, idx, borrowers, deed_token): borrower_service = BorrowerService() borrower_json = { "id": "", "token": "", "forename": borrower['forename'], "surname": borrower['surname'] } if 'middle_name' in borrower: borrower_json["middle_name"] = borrower["middle_name"] if 'id' not in borrower: created_borrower = borrower_service.saveBorrower(borrower, deed_token) borrower_json["id"] = created_borrower.id borrower_json["token"] = created_borrower.token else: borrower_updater = BorrowerModel() updated_borrower = borrower_updater.update_borrower_by_id(borrower) borrower_json["id"] = int(borrower["id"]) borrower_json["token"] = updated_borrower.token return borrower_json
def test_update_borrower_by_id_no_borrower(self, mock_db, mock_query): updated_borrower = copy.deepcopy( DeedHelper._valid_single_borrower_update) mock_query.return_value = None borrower = Borrower() res = borrower.update_borrower_by_id(updated_borrower) self.assertEqual(res, "Error No Borrower")
def test_update_borrower_by_id(self, mock_db, mock_query): updated_borrower = copy.deepcopy( DeedHelper._valid_single_borrower_update) testBorrower = borrower_object_helper(updated_borrower) mock_query.return_value = testBorrower updated_borrower["forename"] = "Frank" borrower = Borrower() res = borrower.update_borrower_by_id(updated_borrower) self.assertEqual(res.forename, "Frank")