from pylons import config
from routes import url_for
-from bluechips import model
+import bluechips.model
from bluechips.model import meta
from bluechips.model.types import Currency
# Invoke websetup with the current config file
SetupCommand('setup-app').run([config['__file__']])
- test_user = model.User()
- test_user.username = u'root'
- test_user.name = u'Charlie Root'
- test_user.resident = True
+ test_user = bluechips.model.User(u'root', u'Charlie Root', False)
meta.Session.save(test_user)
meta.Session.commit()
self.app = TestApp(wsgiapp)
TestCase.__init__(self, *args, **kwargs)
-def createUsers():
- for i in xrange(random.randint(2, 5)):
- u = model.User()
- u.username = sample_users[i].lower()
- u.name = sample_users[i]
- u.resident = 1
+def createUsers(n=None):
+ if n is 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.commit()
-def createExpenditures():
- users = meta.Session.query(model.User).all()
- for i in xrange(random.randint(5, 20)):
- e = model.Expenditure()
+def createExpenditures(n=None):
+ if n is None:
+ n = random.randint(5, 20)
+ users = meta.Session.query(bluechips.model.User).all()
+ for i in xrange(n):
+ e = bluechips.model.Expenditure()
e.spender = random.choice(users)
e.amount = Currency(random.randint(1000, 100000))
meta.Session.save(e)
meta.Session.commit()
def deleteUsers():
- map(meta.Session.delete, meta.Session.query(model.User))
+ map(meta.Session.delete, meta.Session.query(bluechips.model.User))
def deleteExpenditures():
- map(meta.Session.delete, meta.Session.query(model.Expenditure))
+ map(meta.Session.delete, meta.Session.query(bluechips.model.Expenditure))