test_server.py 1.57 KB
Newer Older
di68kap's avatar
di68kap committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
#!/usr/bin/python3

"""test_server.py - tests of the server module of DHParser


Author: Eckhart Arnold <arnold@badw.de>

Copyright 2019 Bavarian Academy of Sciences and Humanities

Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at

    http://www.apache.org/licenses/LICENSE-2.0

Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
"""

import asyncio
import os
from multiprocessing import Process
import sys
from typing import Tuple

eckhart's avatar
eckhart committed
29
30
sys.path.extend(['../', './'])

di68kap's avatar
di68kap committed
31
32
33
34
35
36
37
38
39
from DHParser.server import CompilerServer


scriptdir = os.path.dirname(os.path.realpath(__file__))


def compiler_dummy(src: str, log_dir: str) -> Tuple[str, str]:
    return (src, log_dir)

eckhart's avatar
eckhart committed
40

di68kap's avatar
di68kap committed
41
42
43
class TestServer:
    def test_server(self):
        cs = CompilerServer(compiler_dummy)
eckhart's avatar
eckhart committed
44
        cs.run_as_process()
eckhart's avatar
eckhart committed
45

di68kap's avatar
di68kap committed
46
        async def compile(src, log_dir):
47
            reader, writer = await asyncio.open_connection('127.0.0.1', 8888)
di68kap's avatar
di68kap committed
48
49
50
51
            writer.write(src.encode())
            data = await reader.read(100)
            print(f'Received: {data.decode()!r}')
            writer.close()
52

di68kap's avatar
di68kap committed
53
        asyncio.run(compile('Test', ''))
eckhart's avatar
eckhart committed
54
        cs.terminate_server_process()
di68kap's avatar
di68kap committed
55
56
57
58
59
60



if __name__ == "__main__":
    from DHParser.testing import runner
    runner("", globals())