bluechips/tests/functional/test_spend.py

   1 from datetime import date
   2 from bluechips.tests import *
   3
   4 from bluechips import model
   5 from bluechips.model import meta
   6
   7 class TestSpendController(TestController):
   8
   9     def test_index(self):
  10         response = self.app.get(url_for(controller='spend'))
  11         # Test response...
  12         response.mustcontain('Add a New Expenditure')
  13         form = response.form
  14
  15         user = meta.Session.query(model.User).\
  16                 filter_by(name=u'Charlie Root').one()
  17
  18         form['spender_id'] = user.id
  19         form['amount'] = '66.78'
  20         # Make sure date is today.
  21         today = date.today()
  22         assert form['date'].value == today.strftime('%m/%d/%Y')
  23         form['description'] = 'A test expenditure'
  24         form['shares-0.amount'] = '1'
  25         form['shares-1.amount'] = '2'
  26         form['shares-2.amount'] = '3'
  27         form['shares-3.amount'] = '4'
  28
  29         response = form.submit()
  30         response = response.follow()
  31         response.mustcontain('Expenditure updated.')
  32
  33         e = meta.Session.query(model.Expenditure).\
  34                 order_by(model.Expenditure.id.desc()).first()
  35         assert e.spender.name == u'Charlie Root'
  36         assert e.amount == 6678
  37         assert e.date == today
  38         assert e.description == u'A test expenditure'
  39         meta.Session.delete(e)
  40         meta.Session.commit()
  41
  42         # Test the split.
  43
  44
  45     def test_edit(self):
  46         user = meta.Session.query(model.User).\
  47                 filter_by(name=u'Charlie Root').one()
  48         e = model.Expenditure(user, 53812, u'Lemon bundt cake', None)
  49         e.even_split()
  50         meta.Session.add(e)
  51         meta.Session.commit()
  52
  53         response = self.app.get(url_for(controller='spend',
  54                                         action='edit',
  55                                         id=e.id))
  56         response.mustcontain('Edit an Expenditure')
  57         form = response.form
  58
  59         assert int(form['spender_id'].value) == user.id
  60         assert form['amount'].value == '538.12'
  61         assert form['date'].value == date.today().strftime('%m/%d/%Y')
  62         assert form['description'].value == u'Lemon bundt cake'
  63
  64         form['description'] = u'Updated bundt cake'
  65
  66         # Update the split too.
  67
  68         response = form.submit()
  69         response = response.follow()
  70         response.mustcontain('Expenditure updated.')
  71
  72         e = meta.Session.query(model.Expenditure).\
  73                 order_by(model.Expenditure.id.desc()).first()
  74         assert e.description == u'Updated bundt cake'
  75         meta.Session.delete(e)
  76         meta.Session.commit()