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(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 setUp(self):
s = success()
self.messages = [#starting on Thursday until Friday between 19-21
('THU FRI 19:00 21:00', s),
#starting today until next Friday between 19-21
('FRI 19 21', s),
#starting today until next Friday after 19
('FRI 19', s),
#only today 19 - 21
('19 21', s),
#only today after 19
('19', s),
#today starting now
('', s),
('17:30 16:30', 'Squash - start date greater than end date'),
('29:5', 'Squash - incorrect parameters')
]
def execute(self, context):
logging.info('Creating squash schedule for user ', self.username)
session = context.session
try:
user_product = query_user_product(self.username, 'squashcity', session)
if user_product is None:
return user_not_registered()
squashcity_req = SquashCityRequest(request_type = 'SQUASHCITY',
user_product=user_product,
start_date=self.d_from,
end_date=self.d_to)
session.add(squashcity_req)
session.commit()
return success()
except:
print "Exception in SquashCity.execute: ", sys.exc_info()
return retry_later()
def execute(self, context):
logging.info('Creating squash schedule for user ', self.username)
session = context.session
try:
user_product = query_user_product(self.username, 'squashcity',
session)
if user_product is None:
return user_not_registered()
squashcity_req = SquashCityRequest(request_type='SQUASHCITY',
user_product=user_product,
priority=user_product.priority,
start_date=self.d_from,
end_date=self.d_to)
session.add(squashcity_req)
session.commit()
return success()
except:
print "Exception in SquashCity.execute: ", sys.exc_info()
return retry_later()
