def test_get_positions(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()
self.assertEquals('AAPL', p.get_positions().get().symbol)
self.assertEquals(1, p.get_positions().count())
def fixture(request):
Position.delete_all()
Quote.delete_all()
Portfolio.delete_all()
avanza = Portfolio(name='Avanza', currency='SEK').save()
Portfolio(name='XO', currency='GBP').save()
Position(symbol='AAPL',
currency='SEK',
currency_rate=1.0,
enter_date=date.today(),
enter_price=5000.0,
enter_commission=99.0,
shares=1000.0,
portfolio=avanza).save()
Quote(symbol='AAPL',
close=1234.5,
high=1234.5,
low=1234.5,
open=1234.5,
date=date.today()).save()
return HttpResponseRedirect('/')
def get_portfolios(request):
portfolios = Portfolio.all()
show_closed_positions = request.GET.get('show_closed', False) == 'true'
for p in portfolios:
p.set_show_closed(show_closed_positions)
return portfolios
def test_creation(self):
Portfolio(name='Avanza', currency='SEK').save()
fetched_model = Portfolio.load('Avanza')
self.assertEquals('Avanza', fetched_model.name)
self.assertEquals('SEK', fetched_model.currency)
def test_delete_all(self):
Portfolio(name='Avanza', currency='SEK').save()
Portfolio.delete_all()
self.assertFalse(Portfolio.load('Avanza'))
