def test_delete_all(self): p = Portfolio(name='Avanza', currency='SEK').save() a_date = datetime.date(2001, 1, 3) pos = Position(symbol='AAPL', currency='SEK', currency_rate=1.0, enter_date=a_date, enter_price=5000.0, enter_commission=99.0, shares=1000.0, portfolio=p) pos.save() Position.delete_all() fetched_model = Position.load( symbol='AAPL', enter_date=a_date, portfolio=p) self.assertFalse(fetched_model)
def test_creation(self): p = Portfolio(name='Avanza', currency='SEK').save() a_date = datetime.date(2001, 1, 3) Position(symbol='AAPL', currency='SEK', currency_rate=1.0, enter_date=a_date, enter_price=5000.0, enter_commission=99.0, shares=1000.0, portfolio=p).save() fetched_model = Position.load( symbol='AAPL', enter_date=a_date, portfolio=p) self.assertEquals('AAPL', fetched_model.symbol) self.assertEquals(a_date, fetched_model.enter_date) self.assertEquals(p, fetched_model.portfolio)