require at least one non-zero share on an expenditure

[bluechips.git] / bluechips / tests / __init__.py

diff --git a/bluechips/tests/__init__.py b/bluechips/tests/__init__.py
index cbc0ad459d5eade3873955a405ece11245b539e2..f31ec7dff32eada2d5edc7aee772be4e3301ebb8 100644 (file)
--- a/bluechips/tests/__init__.py
+++ b/bluechips/tests/__init__.py
@@ -31,8 +31,11 @@ def setUpPackage():
     # Invoke websetup with the current config file
     SetupCommand('setup-app').run([config['__file__']])
     
-    test_user = bluechips.model.User(u'root', u'Charlie Root', False)
-    meta.Session.save(test_user)
+    meta.Session.add(bluechips.model.User(u'root', u'Charlie Root', True))
+    meta.Session.add(bluechips.model.User(u'ben', u'Ben Bitdiddle', True))
+    meta.Session.add(bluechips.model.User(u'gotta', u'Gotta Lisp', True))
+    meta.Session.add(bluechips.model.User(u'rich', u'Rich Scheme', True))
+
     meta.Session.commit()
 
 def tearDownPackage():
@@ -50,7 +53,7 @@ def createUsers(n=None):
         n = random.randint(2, 5)
     for i in xrange(n):
         u = bluechips.model.User(sample_users[i].lower(), resident=True)
-        meta.Session.save(u)
+        meta.Session.add(u)
     meta.Session.commit()
 
 def createExpenditures(n=None):
@@ -60,7 +63,7 @@ def createExpenditures(n=None):
     for i in xrange(n):
         e = bluechips.model.Expenditure(random.choice(users),
                                         Currency(random.randint(1000, 100000)))
-        meta.Session.save(e)
+        meta.Session.add(e)
         e.even_split()
     meta.Session.commit()