webOS devices are mobile devices

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

diff --git a/bluechips/tests/__init__.py b/bluechips/tests/__init__.py
index cbc0ad459d5eade3873955a405ece11245b539e2..4eadf12630c18d453b1b32a2eb7e0a06cf0ad726 100644 (file)
--- a/bluechips/tests/__init__.py
+++ b/bluechips/tests/__init__.py
@@ -31,8 +31,15 @@ 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)
+    u1 = bluechips.model.User(u'root', u'Charlie Root', True)
+    u1.email = u'charlie@example.com'
+    u1.password = u'charliepass'
+    u2 = bluechips.model.User(u'ben', u'Ben Bitdiddle', True)
+    u3 = bluechips.model.User(u'gotta', u'Gotta Lisp', True)
+    u4 = bluechips.model.User(u'rich', u'Rich Scheme', True)
+
+    for u in (u1, u2, u3, u4):
+        meta.Session.add(u)
     meta.Session.commit()
 
 def tearDownPackage():
@@ -50,7 +57,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 +67,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()