Total Complexity | 2 |
Total Lines | 51 |
Duplicated Lines | 0 % |
Changes | 0 |
1 | import aiohttp_rpc |
||
2 | |||
3 | |||
4 | async def test_adding_method(): |
||
5 | def method(): |
||
6 | pass |
||
7 | |||
8 | class TestClass: |
||
9 | def method_1(self): |
||
10 | pass |
||
11 | |||
12 | @classmethod |
||
13 | def method_2(cls): |
||
14 | pass |
||
15 | |||
16 | @staticmethod |
||
17 | def method_3(): |
||
18 | pass |
||
19 | |||
20 | rpc_server = aiohttp_rpc.JsonRpcServer() |
||
21 | |||
22 | rpc_server.add_method(method) |
||
23 | assert rpc_server.methods['method'].func == method |
||
24 | |||
25 | rpc_server.add_method(aiohttp_rpc.JsonRpcMethod(method, name='test')) |
||
26 | assert rpc_server.methods['test'].func == method |
||
27 | |||
28 | test_class = TestClass() |
||
29 | rpc_server.add_method(test_class.method_1) |
||
30 | assert rpc_server.methods['method_1'].func == test_class.method_1 |
||
31 | |||
32 | rpc_server.add_method(TestClass.method_2) |
||
33 | assert rpc_server.methods['method_2'].func == TestClass.method_2 |
||
34 | |||
35 | rpc_server.add_method(TestClass.method_3) |
||
36 | assert rpc_server.methods['method_3'].func == TestClass.method_3 |
||
37 | |||
38 | |||
39 | async def test_adding_methods(): |
||
40 | def method_1(): |
||
41 | pass |
||
42 | |||
43 | def method_2(): |
||
44 | pass |
||
45 | |||
46 | rpc_server = aiohttp_rpc.JsonRpcServer() |
||
47 | |||
48 | rpc_server.add_methods([method_1, method_2]) |
||
49 | assert rpc_server.methods['method_1'].func == method_1 |
||
50 | assert rpc_server.methods['method_2'].func == method_2 |
||
51 |