def test_create_whisky_empty_field(self):
whisky_fields = self.whisky_dict.keys()
for field in whisky_fields:
copy_dict = self.whisky_dict.copy()
copy_dict.pop(field)
try:
Whisky.create_whisky(**copy_dict)
self.fail()
except:
pass
def load_whiskies(self, user):
for i in range(5):
whisky = self.whisky_dict.copy()
for key, value in whisky.items():
if key != 'created_by':
if type(value) != str:
whisky[key] = value + i
else:
whisky[key] = value + str(i)
Whisky.create_whisky(whisky['name'], whisky['company'],
whisky['type'], whisky['age'],
whisky['origin'], whisky['flavor'],
whisky['description'], whisky['rating'],
user.uid)
def test_update_whisky(self):
whisky = self.create_test_whisky()
self.whisky_dict['wid'] = whisky.wid
self.whisky_dict['name'] = 'updated'
updated_whisk = Whisky.update_whisky(self.whisky_dict, self.user.uid)
self.assertEqual(whisky.wid, updated_whisk.wid)
self.assertEqual(whisky.name, updated_whisk.name)
for key in self.whisky_dict:
self.assertEqual(whisky.__getattribute__(key), updated_whisk.__getattribute__(key))
def create_test_whisky(self, uid=1, whisky=None, database=db):
if whisky is None:
whisky = self.whisky_dict
whisky_to_create = whisky.copy()
if uid != 1:
whisky_to_create['created_by'] = uid
whisky = Whisky(**whisky_to_create)
database.session.add(whisky)
database.session.commit()
return whisky
def test_get_whiskies_search(self):
self.create_test_whisky()
for key, value in self.whisky_dict.items():
if key != 'age' and key != 'rating' and key != 'created_by':
searched_whisky = Whisky.get_whiskies(value, self.user.uid)
whiskies = [whisky for whisky in searched_whisky]
self.assertEqual(1, len(whiskies), msg=key)
self.create_test_whisky()
for key, value in self.whisky_dict.items():
if key != 'age' and key != 'rating' and key != 'created_by':
searched_whisky = Whisky.get_whiskies(value, self.user.uid)
whiskies = [whisky for whisky in searched_whisky]
self.assertEqual(2, len(whiskies))
self.whisky_dict['name'] = 'bork'
self.create_test_whisky()
self.create_test_user('username2')
self.create_test_whisky(uid=2)
searched_whisky = Whisky.get_whiskies('bork', self.user.uid)
whiskies = [whisky for whisky in searched_whisky]
self.assertEqual(1, len(whiskies))
def test_search_whisky(self):
user = User.create_user('username', 'password')
Whisky.create_whisky('unique', 'unique', 'unique', 12, 'unique',
'unique', 'unique', 5, user.uid)
self.load_whiskies(user)
self.login_user('username', 'password')
time.sleep(5)
items = []
for i in range(5):
items.append(self.driver.find_element_by_id("id_item_%s" % i))
search_field = self.driver.find_element_by_id("id_search_input")
search_field.send_keys("unique")
time.sleep(5)
new_items = []
for i in range(5):
if i == 0:
new_items.append(
self.driver.find_element_by_id("id_item_%s" % i))
else:
try:
self.driver.find_element_by_id("id_item_%s" % i)
self.fail()
except:
pass
self.assertEqual(len(new_items), 1)
self.assertNotEqual(len(new_items), len(items))
whisk = self.get_whisky_card_attr()
self.assertEqual(whisk['name'], 'unique')
search_field.clear()
search_field.send_keys('545')
time.sleep(5)
try:
self.driver.find_element_by_id("id_name_0")
self.fail()
except:
pass
def test_get_whisky_not_existing(self):
searched_whisky = Whisky.get_wisky(0)
self.assertIsNone(searched_whisky)
whisky = self.create_test_whisky()
searched_whisky = Whisky.get_wisky(whisky.wid + 1)
self.assertIsNone(searched_whisky)
def test_get_whisky(self):
added_whisky = self.create_test_whisky()
searched_whisky = Whisky.get_wisky(added_whisky.wid)
for key in self.whisky_dict:
self.assertEqual(added_whisky.__getattribute__(key), searched_whisky.__getattribute__(key))
def test_create_whisky(self):
whisky = Whisky.create_whisky('name', 'company', 'type', 12, 'origin', 'flavor', 'description', 5,
self.user.uid)
whisky_added = Whisky.query.filter_by(wid=whisky.wid).first()
for key in self.whisky_dict:
self.assertEqual(whisky.__getattribute__(key), whisky_added.__getattribute__(key))
def test_update_whisky_only_owner(self):
whisky = self.create_test_whisky()
self.whisky_dict['wid'] = 1
failed_whisk = Whisky.update_whisky(self.whisky_dict, 2)
self.assertIsNone(failed_whisk)