def create_and_test_matches(self): dbman.create_product("SquashCity", session=self.session) for m in self.messages: dbman.register_user("SquashCity", user_name=m[1][0], priority=m[1][1], session=self.session) server = CommandServer([ Message(i, 'CMD SquashCity ' + msg[0], msg[1][0]) for (i, msg) in enumerate(self.messages) ]) #print list(server) context = Context(self.session) for c in list(server): _ = c.execute(context) requests = dbman.active_product_requests('squashcity', session=self.session) self.assertEqual(3, len(requests)) matches = match(requests, self.search_results) req_george = dbman.active_user_requests('George', 'SquashCity', session=self.session)[0] self.assertEqual(self.search_results[0], matches[req_george]) req_ina = dbman.active_user_requests('Ina', 'SquashCity', session=self.session)[0] self.assertEqual(self.search_results[1], matches[req_ina]) req_lucian = dbman.active_user_requests('Luci', 'SquashCity', session=self.session)[0] self.assertFalse(req_lucian in matches) return matches
def create_user(self): #create a category res = dbman.create_product(self.products[0], session=self.session) self.assertEqual(com.success(), res) #create a user assigned to that category res = dbman.register_user(self.products[0], self.username, session=self.session) self.assertEqual(com.success(), res) #create the user again res = dbman.register_user(self.products[0], self.username, session=self.session) self.assertEqual(com.error(com.WARN_ALREADY_REGISTERED), res) #create another category res = dbman.create_product(self.products[1], session=self.session) self.assertEqual(com.success(), res) #assign existing user to another category res = dbman.register_user(self.products[1], self.username, session=self.session) self.assertEqual(com.success(), res)
def validate_commands(self): dbman.create_product("SquashCity", session=self.session) dbman.register_user("SquashCity", "George", session=self.session) s = success() server = CommandServer([Message(i, 'CMD SquashCity ' + msg[0], 'George') for (i,msg) in enumerate(self.messages)]) #print list(server) context = Context(self.session) for (i,c) in enumerate(list(server)): if isinstance(c, Exception): self.assertEqual(c.message, self.messages[i][1]) else: res = c.execute(context) self.assertEqual(s, res)
def test_squash(self): session = dbman.new_session("memory") dbman.create_product("SquashCity", session) dbman.register_user("SquashCity", "George", session) s = success() server = CommandServer([Message(i, 'CMD SquashCity ' + msg[0], 'George') for (i,msg) in enumerate(self.messages)]) #print list(server) context = Context(session) for (i,c) in enumerate(list(server)): if isinstance(c, Exception): self.assertEqual(c.message, self.messages[i][1]) else: res = c.execute(context) self.assertEqual(s, res)
def delete_or_put_user(userid): method = request.form['param1'] if method == 'delete': db.session.query(User).filter(User.id == userid).delete() db.session.commit() return redirect(url_for('get_users')) elif method == 'put': prodid = request.form['param2'] dbman.register_user(prodid, userid, None, db.session) return redirect(url_for('get_user', userid=userid)) elif method == 'put_active': active = request.form['param2'] active = True if active == 'true' else False user = db.session.query(User).filter(User.id == userid).one() user.active = active db.session.commit() return redirect(url_for('get_user', userid=userid))
def delete_or_put_user(userid): method = request.form['param1'] if method == 'delete': db.session.query(User).filter(User.id==userid).delete() db.session.commit() return redirect(url_for('get_users')) elif method == 'put': prodid = request.form['param2'] dbman.register_user(prodid, userid, None, db.session) return redirect(url_for('get_user', userid=userid)) elif method == 'put_active': active = request.form['param2'] active = True if active == 'true' else False user = db.session.query(User).filter(User.id==userid).one() user.active = active db.session.commit() return redirect(url_for('get_user', userid=userid))
def create_user(self): #create a category res = dbman.create_product(self.products[0], session=self.session) self.assertEqual(com.success(), res) #create a user assigned to that category res = dbman.register_user(self.products[0], self.username, session=self.session) self.assertEqual(com.success(), res) #create the user again res = dbman.register_user(self.products[0], self.username, session=self.session) self.assertEqual(com.error(com.WARN_ALREADY_REGISTERED), res) #create another category res = dbman.create_product(self.products[1], session=self.session) self.assertEqual(com.success(), res) #assign existing user to another category res = dbman.register_user(self.products[1], self.username, session=self.session) self.assertEqual(com.success(), res)
def create_user_invalid_product(self): result = dbman.register_user('DUMMY_CATEG', self.username, session=self.session) self.assertEqual(com.error(com.ERR_INVALID_PRODUCT), result)
def create_users_and_products(self): dbman.create_product(self.product, self.session) dbman.register_user(self.product, self.user, session=self.session)
def create_user_invalid_product(self): result = dbman.register_user('DUMMY_CATEG', self.username, session=self.session) self.assertEqual(com.error(com.ERR_INVALID_PRODUCT), result)