def test_delete_fund(self):
"""Ensure fund is deleted"""
fund = UTILS.add_fund('fund_1')
with self.client as client:
res = client.delete(f'/funds/{fund.id}')
data = json.loads(res.data.decode())
self.assertEqual(res.status_code, 200)
self.assertIn('success', data['status'])
self.assertIn(UTILS.DELETED(TYPE, fund.id), data['message'])
def test_delete_committment(self):
"""Ensure committment is deleted"""
committment = UTILS.add_committment(self.fund_1.id, 1000)
with self.client as client:
res = client.delete(f'/committments/{committment.id}')
data = json.loads(res.data.decode())
self.assertEqual(res.status_code, 200)
self.assertIn('success', data['status'])
self.assertIn(UTILS.DELETED(TYPE, committment.id), data['message'])
def test_delete_capitalcall(self):
"""Ensure capitalcall is deleted"""
capitalcall = UTILS.add_capitalcall('investment_1', 1000)
with self.client as client:
res = client.delete(f'/capitalcalls/{capitalcall.id}')
data = json.loads(res.data.decode())
self.assertEqual(res.status_code, 200)
self.assertIn('success', data['status'])
self.assertIn(UTILS.DELETED(TYPE, capitalcall.id), data['message'])
def delete(self, fund_id):
"""Delete single fund details"""
try:
fund = Fund.query.filter_by(id=int(fund_id)).first()
if not fund:
return UTILS.api_response(
msg=UTILS.NOT_EXISTS(TYPE, fund_id),
code=404,
)
else:
UTILS.delete(fund)
return UTILS.api_response(msg=UTILS.DELETED(TYPE, fund.id),
code=200)
except ValueError as e:
return UTILS.api_response(msg=f'{UTILS.VALUE_ERR} {self.__name__}',
code=404,
data=f'{str(e)}')
def delete(self, investment_id):
"""Delete single investment details"""
try:
inv = Investment.query.filter_by(id=int(investment_id)).first()
if not inv:
return UTILS.api_response(
msg=UTILS.NOT_EXISTS(TYPE, investment_id),
code=404,
)
else:
UTILS.delete(inv)
return UTILS.api_response(msg=UTILS.DELETED(TYPE, inv.id),
code=200)
except ValueError as e:
return UTILS.api_response(msg=f'{UTILS.VALUE_ERR} {self.__name__}',
code=404,
data=f'{str(e)}')
def delete(self, capitalcall_id):
"""Delete single capitalcall details"""
try:
capitalcall = CapitalCall.query \
.filter_by(id=int(capitalcall_id)).first()
if not capitalcall:
return UTILS.api_response(
msg=UTILS.NOT_EXISTS(TYPE, capitalcall_id),
code=404,
)
else:
UTILS.delete(capitalcall)
return UTILS.api_response(msg=UTILS.DELETED(
TYPE, capitalcall.id),
code=200)
except ValueError as e:
return UTILS.api_response(msg=f'{UTILS.VALUE_ERR} {self.__name__}',
code=404,
data=f'{str(e)}')
def delete(self, committment_id):
"""Delete single committment details"""
try:
committment = Committment.query \
.filter_by(id=int(committment_id)).first()
if not committment:
return UTILS.api_response(
msg=UTILS.NOT_EXISTS(TYPE, committment_id),
code=404,
)
else:
UTILS.delete(committment)
return UTILS.api_response(msg=UTILS.DELETED(
TYPE, committment.id),
code=200)
except ValueError as e:
return UTILS.api_response(msg=f'{UTILS.VALUE_ERR} {self.__name__}',
code=404,
data=f'{str(e)}')