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')