def test_update_instance(self, mockInst):
testWork = Work()
testWork.epubsToLoad = []
testWork.session = 'session'
testWork.updateInstance(mockInst, 'newInst')
mockInst.update.assert_called_once_with('session', 'newInst')
def test_work_insert(self):
testData = {'title': 'Test Title'}
mock_session = MagicMock()
testWork = Work(session=mock_session)
testWork.epubsToLoad = []
with patch.multiple(Work,
createTmpRelations=DEFAULT,
addImportJson=DEFAULT,
addIdentifiers=DEFAULT,
addInstances=DEFAULT,
addAgents=DEFAULT,
addAltTitles=DEFAULT,
addSubjects=DEFAULT,
addMeasurements=DEFAULT,
addLinks=DEFAULT,
addDates=DEFAULT,
addLanguages=DEFAULT,
removeTmpRelations=DEFAULT) as item_mocks:
newEpubs = testWork.insert(testData)
self.assertEqual(testWork.title, 'Test Title')
self.assertEqual(newEpubs, [])
def test_item_update(self):
testData = {'title': 'Title', 'series': 'newSeries'}
testWork = Work()
testWork.title = 'Title'
testWork.series = 'oldSeries'
testWork.epubsToLoad = []
with patch.multiple(Work,
createTmpRelations=DEFAULT,
addImportJson=DEFAULT,
addTitles=DEFAULT,
updateIdentifiers=DEFAULT,
updateInstances=DEFAULT,
updateAgents=DEFAULT,
updateSubjects=DEFAULT,
updateMeasurements=DEFAULT,
updateLinks=DEFAULT,
updateDates=DEFAULT,
updateLanguages=DEFAULT,
removeTmpRelations=DEFAULT) as item_mocks:
newEpubs = testWork.update(testData)
self.assertEqual(testWork.series, 'newSeries')
self.assertEqual(newEpubs, [])