Esempio n. 1
0
 def test_remove_tmp(self):
     testItem = Item()
     testItem.createTmpRelations({})
     testItem.tmp_links = ['link1', 'link2']
     testItem.removeTmpRelations()
     with self.assertRaises(AttributeError):
         tmpLinks = testItem.tmp_links
Esempio n. 2
0
    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')