Пример #1
0
 def test_remove_tmp(self):
     testWork = Work()
     testWork.createTmpRelations({})
     testWork.tmp_links = ['link1', 'link2']
     testWork.removeTmpRelations()
     with self.assertRaises(AttributeError):
         tmpLinks = testWork.tmp_links
Пример #2
0
 def test_create_tmp(self):
     workData = {
         'links': ['link1', 'link2'],
         'identifiers': ['id1', 'id2', 'id3']
     }
     testWork = Work()
     testWork.createTmpRelations(workData)
     self.assertEqual(testWork.tmp_identifiers[1], 'id2')
     self.assertEqual(len(testWork.tmp_links), 2)
     self.assertEqual(testWork.links, set())