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_get_ids_empty(self): result = trunk_dao.get_ids() self.assertEqual([], result)