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