| # This file is dual licensed under the terms of the Apache License, Version |
| # 2.0, and the BSD License. See the LICENSE file in the root of this repository |
| # for complete details. |
| |
| import abc |
| |
| import pytest |
| |
| from cryptography.utils import ( |
| InterfaceNotImplemented, |
| register_interface_if, |
| verify_interface, |
| ) |
| |
| |
| def test_register_interface_if_true(): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| pass |
| |
| @register_interface_if(1 == 1, SimpleInterface) |
| class SimpleClass(object): |
| pass |
| |
| assert issubclass(SimpleClass, SimpleInterface) is True |
| |
| |
| def test_register_interface_if_false(): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| pass |
| |
| @register_interface_if(1 == 2, SimpleInterface) |
| class SimpleClass(object): |
| pass |
| |
| assert issubclass(SimpleClass, SimpleInterface) is False |
| |
| |
| class TestVerifyInterface(object): |
| def test_verify_missing_method(self): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| @abc.abstractmethod |
| def method(self): |
| """A simple method""" |
| |
| class NonImplementer(object): |
| pass |
| |
| with pytest.raises(InterfaceNotImplemented): |
| verify_interface(SimpleInterface, NonImplementer) |
| |
| def test_different_arguments(self): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| @abc.abstractmethod |
| def method(self, a): |
| """Method with one argument""" |
| |
| class NonImplementer(object): |
| def method(self): |
| """Method with no arguments""" |
| |
| # Invoke this to ensure the line is covered |
| NonImplementer().method() |
| with pytest.raises(InterfaceNotImplemented): |
| verify_interface(SimpleInterface, NonImplementer) |
| |
| def test_handles_abstract_property(self): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| @abc.abstractproperty |
| def property(self): |
| """An abstract property""" |
| |
| class NonImplementer(object): |
| @property |
| def property(self): |
| """A concrete property""" |
| |
| # Invoke this to ensure the line is covered |
| NonImplementer().property |
| verify_interface(SimpleInterface, NonImplementer) |
| |
| def test_signature_mismatch(self): |
| class SimpleInterface(metaclass=abc.ABCMeta): |
| @abc.abstractmethod |
| def method(self, other: object) -> int: |
| """Method with signature""" |
| |
| class ClassWithoutSignature: |
| def method(self, other): |
| """Method without signature""" |
| |
| class ClassWithSignature: |
| def method(self, other: object) -> int: |
| """Method with signature""" |
| |
| verify_interface(SimpleInterface, ClassWithoutSignature) |
| verify_interface(SimpleInterface, ClassWithSignature) |
| with pytest.raises(InterfaceNotImplemented): |
| verify_interface( |
| SimpleInterface, ClassWithoutSignature, check_annotations=True |
| ) |
| verify_interface( |
| SimpleInterface, ClassWithSignature, check_annotations=True |
| ) |