|
| 1 | +""" QGIS test for server services |
| 2 | +""" |
| 3 | +import os |
| 4 | +from qgis.PyQt.QtCore import QBuffer, QIODevice, QTextStream |
| 5 | +from qgis.testing import unittest |
| 6 | +from qgis.core import QgsApplication |
| 7 | +from qgis.server import (QgsServer, |
| 8 | + QgsServiceRegistry, |
| 9 | + QgsService, |
| 10 | + QgsServerRequest, |
| 11 | + QgsServerResponse) |
| 12 | + |
| 13 | +from utilities import unitTestDataPath |
| 14 | + |
| 15 | +class Response(QgsServerResponse): |
| 16 | + |
| 17 | + def __init__( self ): |
| 18 | + QgsServerResponse.__init__(self) |
| 19 | + self._buffer = QBuffer() |
| 20 | + self._buffer.open(QIODevice.ReadWrite) |
| 21 | + |
| 22 | + |
| 23 | + def setReturnCode( self, code ): |
| 24 | + pass |
| 25 | + |
| 26 | + def setHeader( self, key, val ): |
| 27 | + pass |
| 28 | + |
| 29 | + def sendError( self, code, message ): |
| 30 | + pass |
| 31 | + |
| 32 | + def io(self): |
| 33 | + return self._buffer |
| 34 | + |
| 35 | + |
| 36 | +class MyService(QgsService): |
| 37 | + |
| 38 | + def __init__(self, name, version, response): |
| 39 | + QgsService.__init__(self) |
| 40 | + self._response = response |
| 41 | + self._name = name |
| 42 | + self._version = version |
| 43 | + |
| 44 | + def name(self): |
| 45 | + return self._name |
| 46 | + |
| 47 | + def version(self): |
| 48 | + return self._version |
| 49 | + |
| 50 | + def executeRequest( self, request, response ): |
| 51 | + |
| 52 | + url = request.url() |
| 53 | + |
| 54 | + response.setReturnCode(201) |
| 55 | + response.write(self._response) |
| 56 | + |
| 57 | + |
| 58 | +class TestModules(unittest.TestCase): |
| 59 | + """ |
| 60 | + """ |
| 61 | + |
| 62 | + @classmethod |
| 63 | + def setUpClass(cls): |
| 64 | + cls.app = QgsApplication([], False) |
| 65 | + |
| 66 | + @classmethod |
| 67 | + def tearDownClass(cls): |
| 68 | + cls.app.exitQgis() |
| 69 | + |
| 70 | + def setUp(self): |
| 71 | + """Create the server instance""" |
| 72 | + self.testdata_path = unitTestDataPath('qgis_server') + '/' |
| 73 | + |
| 74 | + d = unitTestDataPath('qgis_server_accesscontrol') + '/' |
| 75 | + self.projectPath = os.path.join(d, "project.qgs") |
| 76 | + |
| 77 | + # Clean env just to be sure |
| 78 | + env_vars = ['QUERY_STRING', 'QGIS_PROJECT_FILE'] |
| 79 | + for ev in env_vars: |
| 80 | + try: |
| 81 | + del os.environ[ev] |
| 82 | + except KeyError: |
| 83 | + pass |
| 84 | + self.server = QgsServer() |
| 85 | + |
| 86 | + def test_modules(self): |
| 87 | + """ Tests that modules are loaded """ |
| 88 | + |
| 89 | + # Check that our 'SampleService is registered |
| 90 | + iface = self.server.serverInterface() |
| 91 | + service = iface.serviceRegistry().getService('SampleService') |
| 92 | + |
| 93 | + self.assertIsNotNone(service) |
| 94 | + |
| 95 | + |
| 96 | + |
| 97 | +if __name__ == '__main__': |
| 98 | + unittest.main() |
0 commit comments