class DatabaseTest(unittest.TestCase):
def setUp(self):
self.db = Database("test_zoo.db")
self.a = Animal("lion", 24, "niya", "female", 150)
self.db.insert_animal(self.a)
self.c = self.db.zoo_conn.cursor()
def test_database_init(self):
self.assertEqual("test_zoo.db", self.db.name)
def test_insert_animal(self):
animal_from_db = self.c.execute('''select * from zoo''').fetchall()[0]
self.assertEqual((1, "lion", 24, "niya", "female", 150),
animal_from_db)
last_breed_from_db = self.c.execute('''select id, last_breed
from breeding''').fetchall()[0]
self.assertEqual((1, 0), last_breed_from_db)
def test_remove_animal(self):
self.db.remove_animal("lion", "niya")
animal_from_db = self.c.execute("select * from zoo").fetchall()
self.assertEqual(0, len(animal_from_db))
breed_from_db = self.c.execute("select * from breeding").fetchall()
self.assertEqual(0, len(breed_from_db))
def test_get_males_with_no_males(self):
self.assertEqual(0, len(self.db.get_males("lion")))
def test_get_males_with_one_male(self):
a2 = Animal("lion", 24, "sharik", "male", 150)
self.db.insert_animal(a2)
self.assertEqual([("sharik", )], self.db.get_males("lion"))
def test_has_male(self):
self.assertFalse(self.db.has_male("lion"))
def test_get_females(self):
self.assertEqual([("lion", "niya")], self.db.get_females())
def test_get_life_expectancy(self):
self.assertEqual(15, self.db.get_life_expectancy("lion"))
def test_get_food_type(self):
self.assertEqual("carnivore", self.db.get_food_type("lion"))
def test_get_gestation(self):
self.assertEqual(3, self.db.get_gestation("lion"))
def test_get_newborn_weight(self):
self.assertEqual(2.0, self.db.get_newborn_weight("lion"))
def test_get_average_weight(self):
self.assertEqual(200, self.db.get_average_weight("lion"))
def test_get_weight_age_ratio(self):
self.assertEqual(7.5, self.db.get_weight_age_ratio("lion"))
def test_get_food_weight_ration(self):
self.assertEqual(0.035, self.db.get_food_weight_ratio("lion"))
def test_get_last_breed(self):
self.assertEqual(0, self.db.get_last_breed("lion", "niya"))
def test_set_last_breed(self):
self.db.set_last_breed("lion", "niya", 3)
result = self.db.get_last_breed("lion", "niya")
self.assertEqual(result, 3)
def tearDown(self):
call("rm {}".format(self.db.name), shell=True)
class DatabaseTest(unittest.TestCase):
def setUp(self):
self.db = Database("test_zoo.db")
self.a = Animal("lion", 24, "Svetla", "female", 150)
self.db.insert_animal(self.a)
self.c = self.db.zoo_conn.cursor()
def test_get_name(self):
self.assertEqual("test_zoo.db", self.db.get_name())
def test_insert_animal(self):
animal_from_db = self.c.execute('''SELECT * FROM zoo''').fetchall()[0]
self.assertEqual((1, "lion", 24, "Svetla", "female", 150),
animal_from_db)
last_breed_from_db = self.c.execute('''SELECT id, last_breed
FROM breeding''').fetchall()[0]
self.assertEqual((1, 0), last_breed_from_db)
def test_insert_second_animal(self):
self.a2 = Animal("tiger", 24, "Tsveta", "female", 120)
self.db.insert_animal(self.a2)
animal_from_db = self.c.execute('''SELECT * FROM zoo''').fetchall()[1]
self.assertEqual((2, "tiger", 24, "Tsveta", "female", 120),
animal_from_db)
last_breed_from_db = self.c.execute('''SELECT id, last_breed
FROM breeding''').fetchall()[1]
self.assertEqual((2, 0), last_breed_from_db)
def test_remove_animal(self):
breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
self.assertEqual(1, len(breed_from_db))
self.db.remove_animal("lion", "Svetla")
animal_from_db = self.c.execute("SELECT * FROM zoo").fetchall()
self.assertEqual(0, len(animal_from_db))
breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
self.assertEqual(0, len(breed_from_db))
def test_initial_fill_with_animals(self):
self.db.remove_animal("lion", "Svetla")
self.db.initial_fill_with_animals()
animals_from_db = self.c.execute("SELECT * FROM zoo").fetchall()
self.assertEqual(10, len(animals_from_db))
breed_from_db = self.c.execute("SELECT * from breeding").fetchall()
self.assertEqual(5, len(breed_from_db))
def test_get_males_with_no_males(self):
self.assertEqual(0, len(self.db.get_males("lion")))
def test_get_males_with_one_male(self):
a2 = Animal("lion", 24, "Pencho", "male", 150)
self.db.insert_animal(a2)
self.assertEqual([("Pencho", )], self.db.get_males("lion"))
def test_has_a_male_species(self):
self.assertFalse(self.db.has_a_male_species("lion"))
def test_get_females(self):
self.assertEqual([("lion", "Svetla")], self.db.get_females())
def test_get_life_expectancy(self):
self.assertEqual(15, self.db.get_life_expectancy("lion"))
def test_get_food_type(self):
self.assertEqual("carnivore", self.db.get_food_type("lion"))
def test_get_gestation(self):
self.assertEqual(3, self.db.get_gestation("lion"))
def test_get_newborn_weight(self):
self.assertEqual(2.0, self.db.get_newborn_weight("lion"))
def test_get_average_weight(self):
self.assertEqual(200, self.db.get_average_weight("lion"))
def test_get_weight_age_ratio(self):
self.assertEqual(7.5, self.db.get_weight_age_ratio("lion"))
def test_get_food_weight_ration(self):
self.assertEqual(0.035, self.db.get_food_weight_ratio("lion"))
def test_get_last_breed(self):
self.assertEqual(0, self.db.get_last_breed("lion", "Svetla"))
def test_set_last_breed(self):
self.db.set_last_breed("lion", "Svetla", 3)
result = self.db.get_last_breed("lion", "Svetla")
self.assertEqual(result, 3)
def test_set_age(self):
self.db.set_age("lion", "Svetla", 30)
query = "SELECT age FROM zoo WHERE species = ? AND name = ?"
result = self.c.execute(query, ("lion", "Svetla")).fetchall()
self.assertEqual(30, result[0][0])
def test_set_weight(self):
self.db.set_weight("lion", "Svetla", 195)
query = "SELECT weight FROM zoo WHERE species = ? AND name = ?"
result = self.c.execute(query, ("lion", "Svetla")).fetchall()
self.assertEqual(195, result[0][0])
def tearDown(self):
call("rm {}".format(self.db.name), shell=True)
