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_dir, database = body.split()
HorseManager.add_horse_from_file(horse_dir, database)
ch.basic_publish(exchange='',
routing_key=properties.reply_to,
body='True')
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)
