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
-__all__ = ['url_for', 'TestController', 'sample_users']
+import random
+
+__all__ = ['url_for', 'TestController',
+ 'createUsers', 'createExpenditures',
+ 'deleteUsers', 'deleteExpenditures']
sample_users = [u'Alice', u'Bob', u'Charlie', u'Dave', u'Eve']
-def setup():
+def setUpPackage():
# 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
- meta.Session.save(test_user)
+ test_user = bluechips.model.User(u'root', u'Charlie Root', False)
+ meta.Session.add(test_user)
meta.Session.commit()
-def teardown():
+def tearDownPackage():
meta.metadata.drop_all()
class TestController(TestCase):
wsgiapp = loadapp('config:%s' % config['__file__'])
self.app = TestApp(wsgiapp)
TestCase.__init__(self, *args, **kwargs)
+
+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.add(u)
+ meta.Session.commit()
+
+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(random.choice(users),
+ Currency(random.randint(1000, 100000)))
+ meta.Session.add(e)
+ e.even_split()
+ meta.Session.commit()
+
+def deleteUsers():
+ map(meta.Session.delete, meta.Session.query(bluechips.model.User))
+
+def deleteExpenditures():
+ map(meta.Session.delete, meta.Session.query(bluechips.model.Expenditure))