def test_delete_by_item_id(self):
self.assertEqual(ItemLink.query.count(), 0)
linked_item = Item.create()
link_1 = ItemLink(type=self.link_type, item_id=self.item_1.id, linked_id=linked_item.id)
link_2 = ItemLink(type=self.link_type, item_id=self.item_2.id, linked_id=linked_item.id)
db.session.add(link_1)
db.session.add(link_2)
db.session.flush()
links = ItemLink.query.all()
self.assertEqual(len(links), 2)
self.assertIn(link_1, links)
self.assertIn(link_2, links)
ItemLink.delete_by_item_id(self.item_1.id)
links = ItemLink.query.all()
self.assertEqual(len(links), 1)
self.assertNotIn(link_1, links)
self.assertIn(link_2, links)
ItemLink.delete_by_item_id(self.item_2.id)
self.assertEqual(ItemLink.query.count(), 0)
def test_delete_by_item_id(self):
links = ItemLink.query.all()
assert len(links) == 0
link_type = 'test'
item_1 = Item.create()
item_2 = Item.create()
linked_item = Item.create()
link_1 = ItemLink(type=link_type, item_id=item_1.id, linked_id=linked_item.id)
link_2 = ItemLink(type=link_type, item_id=item_2.id, linked_id=linked_item.id)
db.session.add(link_1)
db.session.add(link_2)
db.session.flush()
links = ItemLink.query.all()
assert len(links) == 2
assert link_1 in links and link_2 in links
ItemLink.delete_by_item_id(item_1.id)
links = ItemLink.query.all()
assert len(links) == 1
assert link_1 not in links and link_2 in links
ItemLink.delete_by_item_id(item_2.id)
links = ItemLink.query.all()
assert len(links) == 0
