def test_get_data(self):
portal = Portal('http://localhost:8090/iportal', 'admin', 'Supermap123')
portal._portal = mock.MagicMock()
data = DataItem()
data_services = mock.MagicMock()
data_services.get_data = mock.MagicMock(return_value=data)
portal._portal.datas_service = mock.MagicMock(return_value=data_services)
result = portal.get_data('data_id')
self.assertEqual(result, data)
def test_upload_data(self):
portal = Portal('http://localhost:8090/iportal', 'admin', 'Supermap123')
portal._portal = mock.MagicMock()
data_services = mock.MagicMock()
mdmr = MyDatasMethodResult()
mdmr.childID = 'data_id'
data_services.post_datas = mock.MagicMock(return_value=mdmr)
data_services.upload_data = mock.MagicMock(return_value=mdmr)
portal._portal.datas_service = mock.MagicMock(return_value=data_services)
portal.get_data = mock.MagicMock()
data1 = DataItem()
data1.status = Status.CREATED
data2 = DataItem()
data2.status = Status.CREATED
data3 = DataItem()
data3.status = Status.OK
portal.get_data.side_effect = [data1, data2, data3]
portal.get_data_upload_progress = mock.MagicMock()
portal.get_data_upload_progress.side_effect = [(0, 100), (50, 100), (100, 100)]
callback = mock.MagicMock()
data_content = mock.MagicMock()
data_content.__class__ = FileIO
result = portal.upload_data('test.json', data_content, DataItemType.JSON, callback)
self.assertEqual(result, 'data_id')
