Revision ed355f9a snf-astakos-app/astakos/im/tests/auth.py

b/snf-astakos-app/astakos/im/tests/auth.py
76 76 
        client.set_tokens(mail="kpap@synnefo.org", eppn="kpapeppn",
77 77 
                          cn="Kostas Papadimitriou",
78 78 
                          ep_affiliation="Test Affiliation")
79 
        r = client.get(ui_url('login/shibboleth?'), follow=True)
79 
        r = client.get(ui_url('login/shibboleth?'), follow=True,
80 
                       **{'HTTP_SHIB_CUSTOM_IDP_KEY': 'test'})
80 81 
        token = PendingThirdPartyUser.objects.get().token
81 82 
        self.assertRedirects(r, ui_url('signup?third_party_token=%s' % token))
82 83 
        self.assertEqual(r.status_code, 200)
......
128 129 
        self.assertEqual(AstakosUserAuthProvider.objects.count(), 1)
129 130 
        self.assertEqual(PendingThirdPartyUser.objects.count(), 0)
130 131 

  		





  
  132
  
    
        user = AstakosUser.objects.get()


  		





  
  133
  
    
        provider = user.get_auth_provider("shibboleth")


  		





  
  134
  
    
        headers = provider.provider_details['info']['headers']


  		





  
  135
  
    
        self.assertEqual(headers.get('SHIB_CUSTOM_IDP_KEY'), 'test')


  		





  
  136
  
    

  		





  131
  137
  
    
        # provider info stored


  		





  132
  138
  
    
        provider = AstakosUserAuthProvider.objects.get(module="shibboleth")


  		





  133
  139
  
    
        self.assertEqual(provider.affiliation, 'Test Affiliation')


  		





  ......




  238
  244
  
    
        self.assertTrue(r.context['request'].user.email == "kpap@synnefo.org")


  		





  239
  245
  
    
        self.assertRedirects(r, ui_url('landing'))


  		





  240
  246
  
    
        self.assertEqual(r.status_code, 200)


  		





  
  247
  
    

  		





  
  248
  
    
        user = r.context['request'].user


  		





  241
  249
  
    
        client.logout()


  		





  242
  250
  
    
        client.reset_tokens()


  		





  243
  251
  
    

  		












Also available in:
  Unified diff