def test_find_by_proto_name_iax(self):
trunk_name = 'my_trunk'
trunk = TrunkFeatures()
trunk.protocolid = 5454
trunk.protocol = 'iax'
useriax = UserIAX()
useriax.id = trunk.protocolid
useriax.name = trunk_name
useriax.type = 'peer'
useriax.category = 'user'
self.session.begin()
map(self.session.add, [trunk, useriax])
self.session.commit()
result = trunk_dao.find_by_proto_name('iax', trunk_name)
self.assertEqual(result, trunk.id)
def test_find_by_proto_name_dahdi_upper(self):
dahdi_interface = 'dahdi/g1'
trunk = TrunkFeatures()
trunk.protocolid = 7878
trunk.protocol = 'custom'
usercustom = UserCustom()
usercustom.name = 'dahdi_test'
usercustom.id = trunk.protocolid
usercustom.interface = dahdi_interface
usercustom.category = 'user'
self.session.begin()
map(self.session.add, [trunk, usercustom])
self.session.commit()
result = trunk_dao.find_by_proto_name('custom', 'DAHDI/g1')
self.assertEqual(result, trunk.id)
def test_find_by_proto_name_sip(self):
trunk_name = 'my_trunk'
trunk = TrunkFeatures()
trunk.protocolid = 5436
trunk.protocol = 'sip'
usersip = UserSIP()
usersip.id = trunk.protocolid
usersip.name = trunk_name
usersip.type = 'peer'
usersip.category = 'user'
self.session.begin()
map(self.session.add, [trunk, usersip])
self.session.commit()
result = trunk_dao.find_by_proto_name('sip', trunk_name)
self.assertEqual(result, trunk.id)
def ztrunks(self, protocol, name):
try:
return trunk_dao.find_by_proto_name(protocol, name)
except (LookupError, ValueError):
return None
