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 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 add_trunk(self, **kwargs):
kwargs.setdefault('tenant_uuid', self.default_tenant.uuid)
trunk = TrunkFeatures(**kwargs)
self.add_me(trunk)
return trunk
def test_get_ids(self):
trunk1 = TrunkFeatures()
trunk1.protocolid = '1234'
trunk1.protocol = 'sip'
trunk2 = TrunkFeatures()
trunk2.protocolid = '4321'
trunk2.protocol = 'iax'
trunk3 = TrunkFeatures()
trunk3.protocolid = '5678'
trunk3.protocol = 'sip'
self.session.begin()
map(self.session.add, [trunk1, trunk2, trunk3])
self.session.commit()
expected = sorted([trunk1.id, trunk2.id, trunk3.id])
result = sorted(trunk_dao.get_ids())
self.assertEqual(expected, result)
