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)
