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