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")
