From a34ff7f6c1afda516d3f5a6b995b98cc84b52902 Mon Sep 17 00:00:00 2001
From: Tomas Babej <tbabej@redhat.com>
Date: Mon, 26 Jan 2015 16:29:29 +0100
Subject: [PATCH] ipatests: Add coverage for adding and removing sshpubkeys in
 ID overrides

Adds xmlrpc tests for:
  - Adding a user ID override with sshpubkey
  - Modifying a user ID override to contain sshpubkey
  - Removing a sshpubkey value from a user ID override

https://fedorahosted.org/freeipa/ticket/4868

Reviewed-By: Martin Kosek <mkosek@redhat.com>
---
 ipatests/test_xmlrpc/test_idviews_plugin.py | 61 +++++++++++++++++++++
 1 file changed, 61 insertions(+)

diff --git a/ipatests/test_xmlrpc/test_idviews_plugin.py b/ipatests/test_xmlrpc/test_idviews_plugin.py
index e4e7e553b..2823082b2 100644
--- a/ipatests/test_xmlrpc/test_idviews_plugin.py
+++ b/ipatests/test_xmlrpc/test_idviews_plugin.py
@@ -55,6 +55,16 @@ host2 = u'testhost2'
 host3 = u'testhost3'
 host4 = u'testhost4'
 
+sshpubkey = (u'ssh-rsa AAAAB3NzaC1yc2EAAAADAQABAAABAQDGAX3xAeLeaJggwTqMjxNwa6X'
+              'HBUAikXPGMzEpVrlLDCZtv00djsFTBi38PkgxBJVkgRWMrcBsr/35lq7P6w8KGI'
+              'wA8GI48Z0qBS2NBMJ2u9WQ2hjLN6GdMlo77O0uJY3251p12pCVIS/bHRSq8kHO2'
+              'No8g7KA9fGGcagPfQH+ee3t7HUkpbQkFTmbPPN++r3V8oVUk5LxbryB3UIIVzNm'
+              'cSIn3JrXynlvui4MixvrtX6zx+O/bBo68o8/eZD26QrahVbA09fivrn/4h3TM01'
+              '9Eu/c2jOdckfU3cHUV/3Tno5d6JicibyaoDDK7S/yjdn5jhaz8MSEayQvFkZkiF'
+              '0L public key test')
+sshpubkeyfp = (u'13:67:6B:BF:4E:A2:05:8E:AE:25:8B:A1:31:DE:6F:1B '
+                'public key test (ssh-rsa)')
+
 
 # Test helpers
 def get_idview_dn(name):
@@ -522,6 +532,54 @@ class test_idviews(Declarative):
             ),
         ),
 
+        dict(
+            desc='Modify User ID override "%s" to override '
+                 'sshpubkey' % idoverrideuser1,
+            command=(
+                'idoverrideuser_mod',
+                [idview1, idoverrideuser1],
+                dict(ipasshpubkey=sshpubkey, all=True)
+            ),
+            expected=dict(
+                value=idoverrideuser1,
+                summary=u'Modified an User ID override "%s"' % idoverrideuser1,
+                result=dict(
+                    dn=get_override_dn(idview1, idoverrideuser1),
+                    objectclass=objectclasses.idoverrideuser,
+                    ipaanchoruuid=[idoverrideuser1],
+                    ipaoriginaluid=[idoverrideuser1],
+                    description=[u'description'],
+                    homedirectory=[u'/home/newhome'],
+                    uid=[u'newlogin'],
+                    ipasshpubkey=[sshpubkey],
+                    sshpubkeyfp=[sshpubkeyfp],
+                )
+            ),
+        ),
+
+        dict(
+            desc='Modify User ID override "%s" to not override '
+                 'sshpubkey' % idoverrideuser1,
+            command=(
+                'idoverrideuser_mod',
+                [idview1, idoverrideuser1],
+                dict(ipasshpubkey=None, all=True)
+            ),
+            expected=dict(
+                value=idoverrideuser1,
+                summary=u'Modified an User ID override "%s"' % idoverrideuser1,
+                result=dict(
+                    dn=get_override_dn(idview1, idoverrideuser1),
+                    objectclass=objectclasses.idoverrideuser,
+                    ipaanchoruuid=[idoverrideuser1],
+                    ipaoriginaluid=[idoverrideuser1],
+                    description=[u'description'],
+                    homedirectory=[u'/home/newhome'],
+                    uid=[u'newlogin'],
+                )
+            ),
+        ),
+
         dict(
             desc='Remove User ID override "%s"' % idoverrideuser1,
             command=('idoverrideuser_del', [idview1, idoverrideuser1], {}),
@@ -541,6 +599,7 @@ class test_idviews(Declarative):
                      homedirectory=u'/home/newhome',
                      uid=u'newlogin',
                      uidnumber=12345,
+                     ipasshpubkey=sshpubkey,
                 )
             ),
             expected=dict(
@@ -555,6 +614,8 @@ class test_idviews(Declarative):
                     homedirectory=[u'/home/newhome'],
                     uidnumber=[u'12345'],
                     uid=[u'newlogin'],
+                    ipasshpubkey=[sshpubkey],
+                    sshpubkeyfp=[sshpubkeyfp],
                 )
             ),
         ),