Tests: Add a smoke-check for listall() and attribute_types()

pyldap contributors · encukou · commit ebcdaae35bb1 · 2017-11-24T11:46:40.000+01:00
diff --git a/Tests/t_ldap_schema_subentry.py b/Tests/t_ldap_schema_subentry.py
@@ -10,6 +10,7 @@
 
 import ldif
 import ldap.schema
+from ldap.schema.models import ObjectClass
 
 TEST_SUBSCHEMA_FILES = (
     'Tests/ldif/subschema-ipa.demo1.freeipa.org.ldif',
@@ -30,6 +31,15 @@ def test_subschema_file(self):
             _, subschema_subentry = ldif_parser.all_records[0]
             sub_schema = ldap.schema.SubSchema(subschema_subentry)
 
+            # Smoke-check for listall() and attribute_types()
+            for objclass in sub_schema.listall(ObjectClass):
+                must, may = sub_schema.attribute_types([objclass])
+
+                for oid, attributetype in must.items():
+                    self.assertEqual(attributetype.oid, oid)
+                for oid, attributetype in may.items():
+                    self.assertEqual(attributetype.oid, oid)
+
 
 if __name__ == '__main__':
     unittest.main()
