def test_add_fund_duplicate_name(self):
"""Test to raise error on adding a duplicate fund name"""
fund = Fund(fundname='test_fund')
db.session.add(fund)
db.session.commit()
duplicate_fund = Fund(fundname='test_fund')
db.session.add(duplicate_fund)
self.assertRaises(IntegrityError, db.session.commit)
def test_add_fund(self):
"""Test to commit a new fund to database"""
fund = Fund(fundname='test_fund')
db.session.add(fund)
db.session.commit()
self.assertTrue(fund.id)
self.assertEqual(fund.fundname, 'test_fund')
def test_add_committment(self):
"""Test to commit a new committment to database"""
fund = Fund(fundname='test_fund')
db.session.add(fund)
db.session.commit()
committment = Committment(fund_id=fund.id, amount=1000)
db.session.add(committment)
db.session.commit()
self.assertTrue(committment.id)
self.assertEqual(committment.amount, 1000)
self.assertEqual(committment.fund_id, fund.id)
def test_check_committent_fund_relationship(self):
"""Test to check that Fund & Committment relationship work"""
fund = Fund(fundname='test_fund')
db.session.add(fund)
db.session.commit()
committment = Committment(fund_id=fund.id, amount=1000)
db.session.add(committment)
db.session.commit()
self.assertTrue(fund.committments)
self.assertTrue(hasattr(fund.committments, '__iter__'))
self.assertTrue(isinstance(fund.committments[0], Committment))
self.assertEqual(committment.id, fund.committments[0].id)
def test_add_two_committments_from_same_fund(self):
"""
Test to check that One-to_Many relationship works between
Fund and Committment models.
"""
fund = Fund(fundname='fund1')
db.session.add(fund)
db.session.commit()
committment1 = Committment(fund_id=fund.id, amount=1500)
committment2 = Committment(fund_id=fund.id, amount=2500)
db.session.add(committment1)
db.session.add(committment2)
db.session.commit()
self.assertEqual(committment1.id, fund.committments[0].id)
self.assertEqual(committment2.id, fund.committments[1].id)
self.assertEqual(committment1.fund_id, committment2.fund_id)
def test_add_fund_investments(self):
"""Test to check Many-to-Many relationship among models"""
fund = Fund('fund_1')
db.session.add(fund)
db.session.commit()
committment = Committment(fund.id, 10000.0)
db.session.add(committment)
db.session.commit()
call = CapitalCall('investment_1', 5000)
db.session.add(call)
db.session.commit()
fund_invest = FundInvestment(5000)
db.session.add(fund_invest)
db.session.commit()
fund_invest.fund.append(fund)
fund_invest.committment.append(committment)
fund_invest.capitalcall.append(call)
db.session.commit()
self.assertTrue(fund.fundinvestments)
self.assertTrue(committment.fundinvestments)
self.assertTrue(call.fundinvestments)
self.assertEqual(fund.fundinvestments[0].id, fund_invest.id)
self.assertEqual(committment.fundinvestments[0].id, fund_invest.id)
self.assertEqual(call.fundinvestments[0].id, fund_invest.id)
self.assertEqual(fund.fundinvestments[0].investment_amount,
fund_invest.investment_amount)
self.assertEqual(committment.fundinvestments[0].investment_amount,
fund_invest.investment_amount)
self.assertEqual(call.fundinvestments[0].investment_amount,
fund_invest.investment_amount)
def test_to_json(self):
"""Test to to check if fund JSON is recieved as dict"""
fund = Fund(fundname='test_fund')
db.session.add(fund)
db.session.commit()
self.assertTrue(isinstance(fund.to_json(), dict))