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'