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)
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_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_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)