def test_upload_data(self):
online = self.online
online._online = 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)
online._online.datas_service = mock.MagicMock(
return_value=data_services)
online.get_data = mock.MagicMock()
data1 = DataItem()
data1.status = Status.CREATED
data2 = DataItem()
data2.status = Status.CREATED
data3 = DataItem()
data3.status = Status.OK
online.get_data.side_effect = [data1, data2, data3]
online.get_data_upload_progress = mock.MagicMock()
online.get_data_upload_progress.side_effect = [(0, 100), (50, 100),
(100, 100)]
callback = mock.MagicMock()
data_content = mock.MagicMock()
data_content.__class__ = FileIO
result = online.upload_data('test.json', data_content,
DataItemType.JSON, callback)
self.assertEqual(result, 'data_id')
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')