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"))