def setUp(self):
super(TestHorseColic, self).setUp()
# Launch receivers
self.receiver = subprocess.Popen(['python', 'EmergencyReceiver.py'])
self.vetReceiver = subprocess.Popen(['python', 'VetEmergency.py'])
# Add a horse to db
HorseManager.add_horse_from_file(TEST_HORSE, TEST_DB)
def callback(ch, method, properties, body):
horse_id, database, condition = body.split()
try:
Owner_mail = HorseManager.find_horse(horse_id, database)['Mail']
except KeyError:
print("Owners mail not known")
raise
send_mail(Owner_mail, "Your horse has {}!".format(condition))
ch.basic_ack(delivery_tag=method.delivery_tag)
def testGroupingByCondition(self):
condition_horses = HorseManager.groupByCondition(TEST_DB)
for ch in condition_horses:
if ch['_id'] == None:
self.assertIn(u'Stormy', ch['value']['horses'])
if ch['_id'] == 'colic':
self.assertIn(u'Misty', ch['value']['horses'])
if ch['_id'] == 'heaves':
self.assertEqual(len(ch['value']['horses']), 2)
def callback(ch, method, properties, body):
horse_id, database, condition = body.split()
HorseManager.modify_horse_doc(horse_id, database, 'Condition', condition)
ch.basic_ack(delivery_tag=method.delivery_tag)
def callback(ch, method, properties, body):
horse_dir, database = body.split()
HorseManager.add_horse_from_file(horse_dir, database)
ch.basic_publish(exchange='',
routing_key=properties.reply_to,
body='True')
def setUp(self):
super(TestGroupConditions, self).setUp()
HorseManager.add_many_horses([MAGGIE, MISTY, BLUE, TEST_HORSE], TEST_DB)
def testAddingFromWrongDir(self):
with self.assertRaises(IOError):
HorseManager.add_horse_from_file(WRONG_DIR, TEST_DB)
def testAddingWithWrongData(self):
with self.assertRaises(ValueError):
HorseManager.add_horse_from_file(WRONG_HORSE, TEST_DB)
def testAddingHorse(self):
# Add horse to the database
HorseManager.add_horse_from_file(TEST_HORSE, TEST_DB)
# Check if stormy is in the database
self.is_horse_in_db(self.stormy)
