def test_remove_tmp(self): testItem = Item() testItem.createTmpRelations({}) testItem.tmp_links = ['link1', 'link2'] testItem.removeTmpRelations() with self.assertRaises(AttributeError): tmpLinks = testItem.tmp_links
def test_add_link(self, mock_link): testItem = Item() testItem.tmp_links = [{'link': 'test_link'}] mock_val = MagicMock() mock_val.name = 'test_link' mock_link.return_value = mock_val testItem.addLinks() self.assertEqual(list(testItem.links)[0].name, 'test_link')