def test_create_es_work(self, mock_work):
mock_session = MagicMock()
mock_session.query.return_value.get.return_value = TestDict(uuid=0)
testDoc = ESDoc(('1', ), mock_session)
newWork = testDoc.createWork()
self.assertEqual(testDoc.dbRec.uuid, 0)
self.assertEqual(newWork, {'title': 'test', 'uuid': '000'})
def test_add_measure(self):
testMeasure = TestDict(**{
'quantity': 'test',
'value': 1,
})
measureRec = ESDoc.addMeasurement(testMeasure)
self.assertEqual(measureRec.quantity, 'test')
self.assertEqual(measureRec.value, 1)
def test_add_identifier(self):
testID = TestDict(**{
'type': None,
'generic': [TestDict(**{'value': 'hello'})]
})
idRec = ESDoc.addIdentifier(testID)
self.assertEqual(idRec.id_type, 'generic')
self.assertEqual(idRec.identifier, 'hello')
def test_add_cover_other_link(self):
testCover = TestDict(**{
'url': 'testURL',
'media_type': 'image/test',
'flags': {'cover': False}
})
coverRec = ESDoc.addCover(testCover)
assert coverRec is None
def test_add_measure(self):
testMeasure = TestDict(**{
'quantity': 'test',
'value': 1,
})
measureRec = ESDoc.addMeasurement(testMeasure)
assert measureRec.quantity == 'test'
assert measureRec.value == 1
def test_add_language(self):
testLang = TestDict(**{
'language': 'test',
'iso_2': 'te',
'iso_3': 'tes'
})
langRec = ESDoc.addLanguage(testLang)
self.assertEqual(langRec.language, 'test')
self.assertEqual(langRec.iso_3, 'tes')
def test_add_cover_json_object(self):
testCover = TestDict(**{
'url': 'testURL',
'media_type': 'image/test',
'flags': {'cover': True}
})
coverRec = ESDoc.addCover(testCover)
assert coverRec.url == 'testURL'
assert coverRec.media_type == 'image/test'
def test_add_cover_json_string(self):
testCover = TestDict(**{
'url': 'testURL',
'media_type': 'image/test',
'flags': '{"cover": true}'
})
coverRec = ESDoc.addCover(testCover)
assert coverRec.url == 'testURL'
assert coverRec.media_type == 'image/test'
def test_add_language(self):
testLang = TestDict(**{
'language': 'test',
'iso_2': 'te',
'iso_3': 'tes'
})
langRec = ESDoc.addLanguage(testLang)
assert langRec.language == 'test'
assert langRec.iso_3 == 'tes'
def test_add_cover_json_string(self):
testCover = TestDict(
**{
'url': 'testURL',
'media_type': 'image/test',
'flags': '{"cover": true}'
})
coverRec = ESDoc.addCover(testCover)
self.assertEqual(coverRec.url, 'testURL')
self.assertEqual(coverRec.media_type, 'image/test')
def test_add_link(self):
testLink = TestDict(**{
'url': 'test/url',
'media_type': 'test',
'flags': '{\"local\": false}',
'id': 1
})
linkRec = ESDoc.addLink(testLink)
assert linkRec.url == 'test/url'
assert linkRec.media_type == 'test'
assert linkRec.local == False
def test_add_cover_json_object(self):
testCover = TestDict(**{
'url': 'testURL',
'media_type': 'image/test',
'flags': {
'cover': True
}
})
coverRec = ESDoc.addCover(testCover)
self.assertEqual(coverRec.url, 'testURL')
self.assertEqual(coverRec.media_type, 'image/test')
def test_add_link(self):
testLink = TestDict(
**{
'url': 'test/url',
'media_type': 'test',
'flags': '{\"local\": false}',
'id': 1
})
linkRec = ESDoc.addLink(testLink)
self.assertEqual(linkRec.url, 'test/url')
self.assertEqual(linkRec.media_type, 'test')
self.assertEqual(linkRec.local, False)
def test_add_identifier(self):
testID = TestDict(**{
'type': None,
'generic': [
TestDict(**{
'value': 'hello'
})
]
})
idRec = ESDoc.addIdentifier(testID)
assert idRec.id_type == 'generic'
assert idRec.identifier == 'hello'
def test_insert_instance_w_pub_date(self):
testInstance = MagicMock()
testDate = MagicMock()
testDate.lower = '2019-01-01'
testDate.upper = '2019-12-31'
dateObj = MagicMock()
dateObj.date_type = 'pub_date'
dateObj.display_date = '2019'
dateObj.date_range = testDate
testInstance.title = 'Test Title'
testInstance.dates = [dateObj]
newInstance = ESDoc.addInstance(testInstance)
self.assertEqual(newInstance.title, 'Test Title')
self.assertEqual(newInstance.pub_date_sort, '2019-01-01')
self.assertEqual(newInstance.pub_date_sort_desc, '2019-12-31')
def test_init_esdoc(self, mock_create):
newDoc = ESDoc(('work1', ), 'session')
self.assertEqual(newDoc.workID, 'work1')
self.assertEqual(newDoc.session, 'session')
self.assertEqual(newDoc.dbRec, None)
self.assertEqual(newDoc.work, 'testWork')
def test_init_esdoc(self, mock_create):
newDoc = ESDoc(('work1',), 'session')
assert newDoc.workID == 'work1'
assert newDoc.session == 'session'
assert newDoc.dbRec == None
assert newDoc.work == 'testWork'
