def test_create_mapping(self, *_):
mapping = '10.0.0.10', 40112, 'desc', True, 3600
mapper = IGDPortMapper()
mapper._mapping_exists = Mock(return_value=False)
mapper.upnp.addanyportmapping.side_effect = Exception
mapper.upnp.addportmapping.return_value = 41102
mapper.upnp.getspecificportmapping = lambda x, *_: \
None if x == 41102 else mapping
assert mapper.create_mapping(40102, 40102) == 41102
assert mapper.create_mapping(40102, 40112) == 41102
assert mapper.create_mapping(40102) == 41102
mapper.upnp.addanyportmapping.side_effect = lambda *_: 45555
assert mapper.create_mapping(40102, 40102) == 45555
assert mapper.create_mapping(40102, 40112) == 45555
assert mapper.create_mapping(40102) == 45555
def test_create_mapping_exists(self, *_):
mapper = IGDPortMapper()
mapper._mapping_exists = Mock(return_value=True)
assert mapper.create_mapping(40102, 40102) == 40102
assert mapper.create_mapping(40102, 40112) == 40112