tests.test_adding_methods

Complexity

Total Complexity

2

Size/Duplication

Total Lines	60
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	test_adding_methods()	0	12	1
A	test_adding_method()	0	42	1

import aiohttp_rpc


async def test_adding_method():
    def method():
        pass

    class TestClass:
        def method_1(self):
            pass

        @classmethod
        def method_2(cls):
            pass

        @staticmethod
        def method_3():
            pass

    rpc_server = aiohttp_rpc.JsonRpcServer()

    rpc_server.add_method(method)
    assert rpc_server.methods['method'].func == method

    rpc_server.add_method(['test', method])
    assert rpc_server.methods['test__method'].func == method

    rpc_server.add_method(aiohttp_rpc.JsonRpcMethod('test_2', method))
    assert rpc_server.methods['test_2__method'].func == method

    rpc_server.add_method(aiohttp_rpc.JsonRpcMethod('', method, custom_name='test'))
    assert rpc_server.methods['test'].func == method

    rpc_server.add_method(aiohttp_rpc.JsonRpcMethod('test', method, custom_name='test'))
    assert rpc_server.methods['test__test'].func == method

    test_class = TestClass()
    rpc_server.add_method(test_class.method_1)
    assert rpc_server.methods['method_1'].func == test_class.method_1

    rpc_server.add_method(TestClass.method_2)
    assert rpc_server.methods['method_2'].func == TestClass.method_2

    rpc_server.add_method(TestClass.method_3)
    assert rpc_server.methods['method_3'].func == TestClass.method_3


async def test_adding_methods():
    def method_1():
        pass

    def method_2():
        pass

    rpc_server = aiohttp_rpc.JsonRpcServer()

    rpc_server.add_methods([method_1, method_2])
    assert rpc_server.methods['method_1'].func == method_1
    assert rpc_server.methods['method_2'].func == method_2