Add kadmin functionality for string attributes

git-svn-id: svn://anonsvn.mit.edu/krb5/trunk@25215 dc483132-0cff-0310-8789-dd5450dbe970

2 files changed, 57 insertions, 0 deletions

diff --git a/src/tests/Makefile.in b/src/tests/Makefile.in
index 0d6f0b795..6578d3309 100644
--- a/src/tests/Makefile.in
+++ b/src/tests/Makefile.in
@@ -69,6 +69,7 @@ check-pytests::
 	$(RUNPYTEST) $(srcdir)/t_renew.py $(PYTESTFLAGS)
 	$(RUNPYTEST) $(srcdir)/t_renprinc.py $(PYTESTFLAGS)
 	$(RUNPYTEST) $(srcdir)/t_cccol.py $(PYTESTFLAGS)
+	$(RUNPYTEST) $(srcdir)/t_stringattr.py $(PYTESTFLAGS)
 
 clean::
 	$(RM) kdc.conf
diff --git a/src/tests/t_stringattr.py b/src/tests/t_stringattr.py
new file mode 100644
index 000000000..392ab6c98
--- /dev/null
+++ b/src/tests/t_stringattr.py
@@ -0,0 +1,56 @@
+# Copyright (C) 2011 by the Massachusetts Institute of Technology.
+# All rights reserved.
+
+# Export of this software from the United States of America may
+#   require a specific license from the United States Government.
+#   It is the responsibility of any person or organization contemplating
+#   export to obtain such a license before exporting.
+#
+# WITHIN THAT CONSTRAINT, permission to use, copy, modify, and
+# distribute this software and its documentation for any purpose and
+# without fee is hereby granted, provided that the above copyright
+# notice appear in all copies and that both that copyright notice and
+# this permission notice appear in supporting documentation, and that
+# the name of M.I.T. not be used in advertising or publicity pertaining
+# to distribution of the software without specific, written prior
+# permission.  Furthermore if you modify this software you must label
+# your software as modified software and not distribute it in such a
+# fashion that it might be confused with the original M.I.T. software.
+# M.I.T. makes no representations about the suitability of
+# this software for any purpose.  It is provided "as is" without express
+# or implied warranty.
+
+#!/usr/bin/python
+from k5test import *
+
+def run_kadmin(query):
+    global realm
+    return realm.run_as_master([kadmin, '-c', realm.ccache, '-q', query])
+
+realm = K5Realm(create_host=False, get_creds=False)
+
+realm.kinit(realm.admin_princ, password('admin'), flags=['-S', 'kadmin/admin'])
+
+output = run_kadmin('getstrs user')
+if '(No string attributes.)' not in output:
+    fail('Empty attribute query')
+
+output = run_kadmin('setstr user attr1 value1')
+if 'Attribute set for principal' not in output:
+    fail('Setting attr1')
+output = run_kadmin('setstr user attr2 value2')
+if 'Attribute set for principal' not in output:
+    fail('Setting attr2')
+output = run_kadmin('delstr user attr1')
+if 'Attribute removed from principal' not in output:
+    fail('Deleting attr1')
+output = run_kadmin('setstr user attr3 value3')
+if 'Attribute set for principal' not in output:
+    fail('Setting attr3')
+
+output = run_kadmin('getstrs user')
+if 'attr2: value2' not in output or 'attr3: value3' not in output or \
+        'attr1:' in output:
+    fail('Final attribute query')
+
+success('KDB string attributes.')