def test_credit(self):
self.create_user()
data = { 'username': "******", 'password': "******" }
# Let's create a service
response = self.app.post('/start/php', data=data)
servicedict = simplejson.loads(response.data)
self.assertEquals(1, servicedict['sid'])
# No sid and decrement
data = {}
response = self.app.post('/callback/decrementUserCredit.php', data=data)
self.assertEquals({ 'error': True }, simplejson.loads(response.data))
# Right sid but not enough credit
data = { 'sid': 1, 'decrement': 10000 }
response = self.app.post('/callback/decrementUserCredit.php', data=data)
self.assertEquals({ 'error': True }, simplejson.loads(response.data))
user = app.auth_user("ema", "properpass")
self.assertEquals(120, user.credit)
# Right sid and enough credit
data = { 'sid': 1, 'decrement': 1 }
response = self.app.post('/callback/decrementUserCredit.php', data=data)
self.assertEquals({ 'error': False }, simplejson.loads(response.data))
user = app.auth_user("ema", "properpass")
self.assertEquals(119, user.credit)
def test_create_service(self):
self.create_user()
user = app.auth_user("ema", "properpass")
service = app.Service(name="New selenium service", type="selenium",
user=user)
app.db.session.add(service)
app.db.session.commit()
# Testing service->user backref
self.assertEquals(120, service.user.credit)
def test_decrement_user_credit(self):
user = self.create_user()
user.credit -= 10
if user.credit > -1:
app.db.session.commit()
else:
app.db.session.rollback()
user = app.auth_user("ema", "properpass")
self.assertEquals(110, user.credit)
user.credit -= 5000
if user.credit > -1:
app.db.session.commit()
else:
app.db.session.rollback()
user = app.auth_user("ema", "properpass")
self.assertEquals(110, user.credit)
def test_create_user(self):
self.create_user()
self.assertFalse(app.auth_user("ema", "wrongpass"))
self.assert_(app.auth_user("ema", "properpass") is not None)
self.assertFalse(app.auth_user("wronguname", "properpass"))
