Update BlueChips for WebHelpers 1.0 and beyond

[bluechips.git] / bluechips / tests / functional / test_spend.py

diff --git a/bluechips/tests/functional/test_spend.py b/bluechips/tests/functional/test_spend.py
index 7c510f91968f182746dd6dd7ef9b7313649c98ee..4228f0fc9c22b8ac00c94ede6caf134a5ca86c5d 100644 (file)
--- a/bluechips/tests/functional/test_spend.py
+++ b/bluechips/tests/functional/test_spend.py
@@ -1,8 +1,15 @@
 from datetime import date
+from formencode import Invalid
+
+from webhelpers.pylonslib.secure_form import token_key
+
 from bluechips.tests import *
 
 from bluechips import model
 from bluechips.model import meta
+from bluechips.model.types import Currency
+
+from bluechips.controllers.spend import ExpenditureSchema
 
 class TestSpendController(TestController):
 
@@ -16,15 +23,17 @@ class TestSpendController(TestController):
                 filter_by(name=u'Charlie Root').one()
         
         form['spender_id'] = user.id
-        form['amount'] = '66.78'
+        form['amount'] = '74.04'
         # Make sure date is today.
         today = date.today()
         assert form['date'].value == today.strftime('%m/%d/%Y')
         form['description'] = 'A test expenditure'
         form['shares-0.amount'] = '1'
         form['shares-1.amount'] = '2'
-        form['shares-2.amount'] = '3'
-        form['shares-3.amount'] = '4'
+        form['shares-2.amount'] = '2'
+        form['shares-3.amount'] = '1'
+        for ii in range(4):
+            assert int(form['shares-%d.user_id' % ii].value) == ii + 1
 
         response = form.submit()
         response = response.follow()
@@ -33,14 +42,20 @@ class TestSpendController(TestController):
         e = meta.Session.query(model.Expenditure).\
                 order_by(model.Expenditure.id.desc()).first()
         assert e.spender.name == u'Charlie Root'
-        assert e.amount == 6678
+        assert e.amount == 7404
         assert e.date == today
         assert e.description == u'A test expenditure'
 
         # Test the split.
+        shares = dict(((sp.user_id, sp.share)
+                       for sp in e.splits))
+        assert shares[1] == Currency('12.34')
+        assert shares[2] == Currency('24.68')
+        assert shares[3] == Currency('24.68')
+        assert shares[4] == Currency('12.34')
 
 
-    def test_edit(self):
+    def test_edit_and_delete(self):
         user = meta.Session.query(model.User).\
                 filter_by(name=u'Charlie Root').one()
         e = model.Expenditure(user, 53812, u'Lemon bundt cake', None)
@@ -71,6 +86,104 @@ class TestSpendController(TestController):
                 order_by(model.Expenditure.id.desc()).first()
         assert e.description == u'Updated bundt cake'
 
+        response = self.app.get(url_for(controller='spend',
+                                        action='delete',
+                                        id=e.id))
+        response = response.form.submit('delete').follow()
+        response.mustcontain('Expenditure', 'deleted')
+
+    def test_delete_nonexistent(self):
+        self.app.get(url_for(controller='spend',
+                             action='delete',
+                             id=124344),
+                     status=404)
+
+    def test_destroy_nonexistent(self):
+        response = self.app.get(url_for(controller='spend',
+                                        action='edit'))
+        params = self.sample_params.copy()
+        params[token_key] = response.form[token_key].value
+        self.app.post(url_for(controller='spend',
+                              action='destroy',
+                              id=124344), 
+                      params=params,
+                      status=404)
+
+    def test_delete_xsrf_protection(self):
+        self.app.post(url_for(controller='spend',
+                              action='destroy',
+                              id=1),
+                      params={'delete': 'Delete'},
+                      status=403)
+
+    def test_edit_zero_value(self):
+        user = meta.Session.query(model.User).\
+                filter_by(name=u'Charlie Root').one()
+        e = model.Expenditure(user, 0, u'A zero value expenditure', None)
+        e.even_split()
+        meta.Session.add(e)
+        meta.Session.commit()
+
+        response = self.app.get(url_for(controller='spend',
+                                        action='edit',
+                                        id=e.id))
+        response.mustcontain('Edit an Expenditure')
+        form = response.form
+
+        assert int(form['spender_id'].value) == user.id
+        assert form['amount'].value == '0.00'
+        assert form['date'].value == date.today().strftime('%m/%d/%Y')
+        assert form['description'].value == u'A zero value expenditure'
+        for ii in range(4):
+            assert form['shares-%d.amount' % ii].value == '0'
+
+    def test_edit_nonexistent(self):
+        response = self.app.get(url_for(controller='spend',
+                                        action='edit',
+                                        id=124234), status=404)
+
+    def test_update_nonexistent(self):
+        response = self.app.get(url_for(controller='spend',
+                                        action='edit'))
+        params = self.sample_params.copy()
+        params[token_key] = response.form[token_key].value
+        self.app.post(url_for(controller='spend',
+                              action='update',
+                              id=14234), 
+                      params=params,
+                      status=404)
+
+    def test_xsrf_protection(self):
+        self.app.post(url_for(controller='spend',
+                              action='update'),
+                      params=self.sample_params,
+                      status=403)
+
+    def test_all_zero_shares_fails(self):
+        params = self.sample_params.copy()
+        for ii in range(4):
+            params['shares-%d.amount' % ii] = '0'
+        v = ExpenditureSchema()
+        try:
+            v.to_python(params)
+        except Invalid:
+            pass
+
+    def setUp(self):
+        self.sample_params = {
+            'spender_id': '1',
+            'amount': '44.12',
+            'date': '10/5/2008',
+            'description': 'Example expenditure post data.',
+            'shares-0.user_id': '1',
+            'shares-0.amount': '1',
+            'shares-1.user_id': '2',
+            'shares-1.amount': '1',
+            'shares-2.user_id': '3',
+            'shares-2.amount': '1',
+            'shares-3.user_id': '4',
+            'shares-3.amount': '1'}
+
     def tearDown(self):
         expenditures = meta.Session.query(model.Expenditure).all()
         for e in expenditures: