tests/testrest.py

   1 #!/usr/bin/python
   2 #
   3 # Copyright (C) 2015 Ipsilon project Contributors, for license see COPYING
   4
   5 from helpers.common import IpsilonTestBase  # pylint: disable=relative-import
   6 from helpers.http import HttpSessions  # pylint: disable=relative-import
   7 import os
   8 import pwd
   9 import sys
  10 from string import Template
  11
  12
  13 idp_g = {'TEMPLATES': '${TESTDIR}/templates/install',
  14          'CONFDIR': '${TESTDIR}/etc',
  15          'DATADIR': '${TESTDIR}/lib',
  16          'HTTPDCONFD': '${TESTDIR}/${NAME}/conf.d',
  17          'STATICDIR': '${ROOTDIR}',
  18          'BINDIR': '${ROOTDIR}/ipsilon',
  19          'WSGI_SOCKET_PREFIX': '${TESTDIR}/${NAME}/logs/wsgi'}
  20
  21
  22 idp_a = {'hostname': '${ADDRESS}:${PORT}',
  23          'admin_user': '${TEST_USER}',
  24          'system_user': '${TEST_USER}',
  25          'instance': '${NAME}',
  26          'secure': 'no',
  27          'testauth': 'yes',
  28          'pam': 'no',
  29          'gssapi': 'no',
  30          'ipa': 'no',
  31          'server_debugging': 'True'}
  32
  33
  34 sp_g = {'HTTPDCONFD': '${TESTDIR}/${NAME}/conf.d',
  35         'SAML2_TEMPLATE': '${TESTDIR}/templates/install/saml2/sp.conf',
  36         'SAML2_CONFFILE': '${TESTDIR}/${NAME}/conf.d/ipsilon-saml.conf',
  37         'SAML2_HTTPDIR': '${TESTDIR}/${NAME}/saml2'}
  38
  39
  40 sp_a = {'hostname': '${ADDRESS}:${PORT}',
  41         'saml_idp_metadata': 'http://127.0.0.10:45080/idp1/saml2/metadata',
  42         'saml_secure_setup': 'False',
  43         'saml_auth': '/sp',
  44         'httpd_user': '${TEST_USER}'}
  45
  46
  47 sp2_g = {'HTTPDCONFD': '${TESTDIR}/${NAME}/conf.d',
  48          'SAML2_TEMPLATE': '${TESTDIR}/templates/install/saml2/sp.conf',
  49          'SAML2_CONFFILE': '${TESTDIR}/${NAME}/conf.d/ipsilon-saml.conf',
  50          'SAML2_HTTPDIR': '${TESTDIR}/${NAME}/saml2'}
  51
  52
  53 sp2_a = {'hostname': '${ADDRESS}:${PORT}',
  54          'saml_idp_metadata': 'http://127.0.0.10:45080/idp1/saml2/metadata',
  55          'saml_secure_setup': 'False',
  56          'saml_auth': '/sp',
  57          'httpd_user': '${TEST_USER}'}
  58
  59 sp3_g = {'HTTPDCONFD': '${TESTDIR}/${NAME}/conf.d',
  60          'SAML2_TEMPLATE': '${TESTDIR}/templates/install/saml2/sp.conf',
  61          'SAML2_CONFFILE': '${TESTDIR}/${NAME}/conf.d/ipsilon-saml.conf',
  62          'SAML2_HTTPDIR': '${TESTDIR}/${NAME}/saml2'}
  63
  64
  65 sp3_a = {'hostname': '${ADDRESS}:${PORT}',
  66          'saml_idp_metadata': 'http://127.0.0.10:45080/idp1/saml2/metadata',
  67          'saml_secure_setup': 'False',
  68          'saml_auth': '/sp',
  69          'httpd_user': '${TEST_USER}'}
  70
  71
  72 def fixup_sp_httpd(httpdir, alias):
  73     location = """
  74
  75 Alias /${ALIAS} ${HTTPDIR}/sp
  76
  77 <Directory ${HTTPDIR}/${ALIAS}>
  78     Require all granted
  79 </Directory>
  80 """
  81     index = """WORKS!"""
  82
  83     t = Template(location)
  84     text = t.substitute({'HTTPDIR': httpdir, 'ALIAS': alias})
  85     with open(httpdir + '/conf.d/ipsilon-saml.conf', 'a') as f:
  86         f.write(text)
  87
  88     os.mkdir(httpdir + '/sp')
  89     with open(httpdir + '/sp/index.html', 'w') as f:
  90         f.write(index)
  91
  92
  93 class IpsilonTest(IpsilonTestBase):
  94
  95     def __init__(self):
  96         super(IpsilonTest, self).__init__('testrest', __file__)
  97
  98     def setup_servers(self, env=None):
  99         print "Installing IDP server"
 100         name = 'idp1'
 101         addr = '127.0.0.10'
 102         port = '45080'
 103         idp = self.generate_profile(idp_g, idp_a, name, addr, port)
 104         conf = self.setup_idp_server(idp, name, addr, port, env)
 105
 106         print "Starting IDP's httpd server"
 107         self.start_http_server(conf, env)
 108
 109         print "Installing SP server"
 110         name = 'sp1'
 111         addr = '127.0.0.11'
 112         port = '45081'
 113         sp = self.generate_profile(sp_g, sp_a, name, addr, port)
 114         conf = self.setup_sp_server(sp, name, addr, port, env)
 115         fixup_sp_httpd(os.path.dirname(conf), name)
 116
 117         print "Starting SP's httpd server"
 118         self.start_http_server(conf, env)
 119
 120         print "Installing second SP server"
 121         name = 'sp2'
 122         addr = '127.0.0.10'
 123         port = '45082'
 124         sp2 = self.generate_profile(sp2_g, sp2_a, name, addr, port)
 125         conf = self.setup_sp_server(sp2, name, addr, port, env)
 126         fixup_sp_httpd(os.path.dirname(conf), name)
 127
 128         print "Starting SP's httpd server"
 129         self.start_http_server(conf, env)
 130
 131         print "Installing third SP server"
 132         name = 'sp3.invalid'
 133         addr = '127.0.0.10'
 134         port = '45083'
 135         sp3 = self.generate_profile(sp3_g, sp3_a, name, addr, port)
 136         conf = self.setup_sp_server(sp3, name, addr, port, env)
 137         fixup_sp_httpd(os.path.dirname(conf), name)
 138
 139         print "Starting SP's httpd server"
 140         self.start_http_server(conf, env)
 141
 142
 143 if __name__ == '__main__':
 144
 145     idpname = 'idp1'
 146     spname = 'sp1'
 147     sp2name = 'sp2'
 148     sp3name = 'sp3.invalid'
 149     user = pwd.getpwuid(os.getuid())[0]
 150
 151     sess = HttpSessions()
 152     sess.add_server(idpname, 'http://127.0.0.10:45080', user, 'ipsilon')
 153     sess.add_server(spname, 'http://127.0.0.11:45081')
 154     sess.add_server(sp2name, 'http://127.0.0.10:45082')
 155     sess.add_server(sp3name, 'http://127.0.0.10:45083')
 156
 157     print "testrest: Authenticate to IDP ...",
 158     try:
 159         sess.auth_to_idp(idpname)
 160     except Exception, e:  # pylint: disable=broad-except
 161         print >> sys.stderr, " ERROR: %s" % repr(e)
 162         sys.exit(1)
 163     print " SUCCESS"
 164
 165     print "testrest: List initial Service Providers via REST ...",
 166     try:
 167         result = sess.get_rest_sp(idpname)
 168         if len(result['result']) != 0:
 169             raise ValueError(
 170                 'Expected no SP and got %d' % len(result['result'])
 171             )
 172     except ValueError, e:
 173         print >> sys.stderr, " ERROR: %s" % repr(e)
 174         sys.exit(1)
 175     print " SUCCESS"
 176
 177     print "testrest: Add SP Metadata to IDP via admin ...",
 178     try:
 179         sess.add_sp_metadata(idpname, spname)
 180     except Exception, e:  # pylint: disable=broad-except
 181         print >> sys.stderr, " ERROR: %s" % repr(e)
 182         sys.exit(1)
 183     print " SUCCESS"
 184
 185     print "testrest: List Service Providers via REST ...",
 186     try:
 187         result = sess.get_rest_sp(idpname)
 188         if len(result['result']) != 1:
 189             raise ValueError(
 190                 'Expected 1 SP and got %d' % len(result['result'])
 191             )
 192         if result['result'][0].get('provider') != spname:
 193             raise ValueError(
 194                 'Expected %s and got %s' %
 195                 (spname, result['result'][0].get('provider'))
 196             )
 197     except ValueError, e:
 198         print >> sys.stderr, " ERROR: %s" % repr(e)
 199         sys.exit(1)
 200     print " SUCCESS"
 201
 202     print "testrest: Add Service Provider via REST ...",
 203     try:
 204         sess.add_sp_metadata(idpname, sp2name, rest=True)
 205     except ValueError, e:
 206         print >> sys.stderr, " ERROR: %s" % repr(e)
 207         sys.exit(1)
 208     print " SUCCESS"
 209
 210     print "testrest: List Service Providers via REST ...",
 211     try:
 212         result = sess.get_rest_sp(idpname)
 213         if len(result['result']) != 2:
 214             raise ValueError(
 215                 'Expected 2 SPs and got %d' % len(result['result'])
 216             )
 217     except ValueError, e:
 218         print >> sys.stderr, " ERROR: %s" % repr(e)
 219         sys.exit(1)
 220     print " SUCCESS"
 221
 222     print "testrest: List Specific Service Providers via REST ...",
 223     try:
 224         result = sess.get_rest_sp(idpname, spname)
 225         if len(result['result']) != 1:
 226             raise ValueError(
 227                 'Expected 1 SPs and got %d' % len(result['result'])
 228             )
 229         if result['result'][0].get('provider') != spname:
 230             raise ValueError(
 231                 'Expected %s and got %s' %
 232                 (spname, result['result'][0].get('provider'))
 233             )
 234     except ValueError, e:
 235         print >> sys.stderr, " ERROR: %s" % repr(e)
 236         sys.exit(1)
 237     print " SUCCESS"
 238
 239     # Now for some negative testing
 240
 241     print "testrest: Add illegally named Service Provider via REST ...",
 242     try:
 243         sess.add_sp_metadata(idpname, sp3name, rest=True)
 244     except ValueError, e:
 245         print " SUCCESS"
 246     else:
 247         print >> sys.stderr, "ERROR: " \
 248             "Adding SP with invalid name should have failed and it didn't"
 249         sys.exit(1)
 250
 251     print "testrest: Fetch non-existent REST endpoint ...",
 252     try:
 253         result = sess.fetch_rest_page(
 254             idpname,
 255             '/%s/rest/providers/saml2/notfound' % idpname
 256         )
 257     except ValueError, e:
 258         if '(501)' not in e.message:
 259             print >> sys.stderr, " ERROR: %s" % repr(e)
 260             sys.exit(1)
 261         else:
 262             print " SUCCESS"
 263     else:
 264         print >> sys.stderr, "ERROR: should have returned a 404"
 265         sys.exit(1)
 266
 267     print "testrest: Fetch non-existent SP via REST ...",
 268     try:
 269         result = sess.get_rest_sp(idpname, 'foo')
 270     except ValueError, e:
 271         if '(404)' not in e.message:
 272             print >> sys.stderr, " ERROR: %s" % repr(e)
 273             sys.exit(1)
 274         else:
 275             print " SUCCESS"
 276     else:
 277         print >> sys.stderr, "ERROR: should have returned a 404"
 278         sys.exit(1)
 279
 280     print "testrest: Re-add Service Provider via REST ...",
 281     try:
 282         sess.add_sp_metadata(idpname, sp2name, rest=True)
 283     except ValueError, e:
 284         print " SUCCESS"
 285     else:
 286         print >> sys.stderr, "ERROR: " \
 287             "Adding duplicate SP should have failed and it didn't"
 288         sys.exit(1)