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
def test_get_by_item_id(self):
item_link = ItemLink(type=self.link_type, item_id=self.item_1.id, linked_id=self.item_2.id)
db.session.add(item_link)
db.session.flush()
results = ItemLink.get_by_item_id(self.item_1.id)
self.assertListEqual(results, [item_link])
missing_results = ItemLink.get_by_item_id(self.item_2.id)
self.assertListEqual(missing_results, [])
def test_find_or_insert(self):
self.assertEqual(ItemLink.query.count(), 0)
link_1 = ItemLink.find_or_insert(self.item_1.id, self.item_2.id, self.link_type)
links = ItemLink.query.all()
self.assertEqual(len(links), 1)
self.assertEqual(links[0], link_1)
link_2 = ItemLink.find_or_insert(self.item_1.id, self.item_2.id, self.link_type)
links = ItemLink.query.all()
self.assertEqual(len(links), 1)
self.assertEqual(links[0], link_2)
self.assertEqual(link_1, link_2)
def test_find_or_insert(self):
links = ItemLink.query.all()
assert len(links) == 0
link_type = 'test'
test_item_1 = Item.create()
test_item_2 = Item.create()
link_1 = ItemLink.find_or_insert(test_item_1.id, test_item_2.id, link_type)
links = ItemLink.query.all()
assert len(links) == 1
assert links[0] == link_1
link_2 = ItemLink.find_or_insert(test_item_1.id, test_item_2.id, link_type)
links = ItemLink.query.all()
assert len(links) == 1
assert links[0] == link_2
assert link_1 == link_2
def test_to_dict(self):
item_link = ItemLink(type=self.link_type, item_id=self.item_1.id, linked_id=self.item_2.id)
db.session.add(item_link)
db.session.flush()
self.assertDictEqual(item_link.to_dict(),
{'type': item_link.type, 'item_id': item_link.item_id, 'linked_id': item_link.linked_id})
