Revision 468ba00b snf-cyclades-app/synnefo/helpdesk/tests.py

b/snf-cyclades-app/synnefo/helpdesk/tests.py
51 51 

  		





  52
  52
  
    
from synnefo.db import models_factory as mfactory


  		





  53
  53
  
    

  		





  
  54
  
    

  		





  
  55
  
    
class AstakosClientMock():


  		





  
  56
  
    
    def __init__(*args, **kwargs):


  		





  
  57
  
    
        pass


  		





  
  58
  
    

  		





  
  59
  
    
    def get_username(self, token, uuid):


  		





  
  60
  
    
        try:


  		





  
  61
  
    
            return USERS_UUIDS.get(uuid)['displayname']


  		





  
  62
  
    
        except TypeError:


  		





  
  63
  
    
            return None


  		





  
  64
  
    

  		





  
  65
  
    
    def get_uuid(self, token, display_name):


  		





  
  66
  
    
        try:


  		





  
  67
  
    
            return USERS_DISPLAYNAMES.get(display_name)['uuid']


  		





  
  68
  
    
        except TypeError:


  		





  
  69
  
    
            return None


  		





  
  70
  
    

  		





  
  71
  
    

  		





  54
  72
  
    
class AuthClient(Client):


  		





  55
  73
  
    

  		





  56
  74
  
    
    def request(self, **request):


  		





  ......




  79
  97
  
    
                                                           'helpdesk'],


  		





  80
  98
  
    
                                'auth_token': '0001'}


  		





  81
  99
  
    

  		





  82
  
  
    
        def get_uuid_mock(token, displayname, url):


  		





  83
  
  
    
            try:


  		





  84
  
  
    
                return USERS_DISPLAYNAMES.get(displayname)['uuid']


  		





  85
  
  
    
            except TypeError:


  		





  86
  
  
    
                return None


  		





  87
  
  
    

  		





  88
  
  
    
        def get_displayname_mock(token, uuid, url):


  		





  89
  
  
    
            try:


  		





  90
  
  
    
                return USERS_UUIDS.get(uuid)['displayname']


  		





  91
  
  
    
            except TypeError:


  		





  92
  
  
    
                return None


  		





  93
  
  
    

  		





  94
  100
  
    
        # mock the astakos authentication function


  		





  95
  101
  
    
        from snf_django.lib import astakos


  		





  96
  102
  
    
        astakos.get_user = get_user_mock


  		





  97
  
  
    
        astakos.get_displayname = get_displayname_mock


  		





  98
  
  
    
        astakos.get_user_uuid = get_uuid_mock


  		





  
  103
  
    
        import astakosclient


  		





  
  104
  
    
        astakosclient.AstakosClient = AstakosClientMock


  		





  99
  105
  
    

  		





  100
  106
  
    
        settings.SKIP_SSH_VALIDATION = True


  		





  101
  107
  
    
        settings.HELPDESK_ENABLED = True


  		












Also available in:
  Unified diff