Пример #1
0
 def setUp(self):
     self.data = dict(
         authType = 0,
         lastBackupTime = 1461125781055,
         isNew = True,
         isValid = True,
         maxInactive = 7200,
         principalData = dict(
             logined = True,
             verified = True,
             user = '******',
             agreedTerms = True
         ),
         creationTime = 1461125781055,
         principalDataLength = 0,
         version = 2,
         savedPrincipalDataLength = 0,
         savedRequestDataLength = 0,
         idLength = 32,
         thisAccessedTime = 1461125781055,
         sessionFieldsDataLength = 84,
         id = 'a3065b6606ae11e682397831c1c31f4a',
         lastAccessedTime = 1461125781055
     )
     self.transcoder = Transcoder()
Пример #2
0
class TestUnicode(unittest.TestCase):

    def setUp(self):
        self.__base64_data = "AAIAVAAAAVZD345sAAABVkQZzNAAABwgMDEAAAFWRBnM0AAAAVZEGczQACBCREREMzVFN0FGNkMxOUJFRUFBMEE2NDJBNTBGMUZGMAAAAAAAAAAAeyJhZ3JlZWRUZXJtcyI6dHJ1ZSwiZGVmUGljIjoicHMwMC5wbmciLCJ2ZXJpZmllZCI6dHJ1ZSwid3JpdGluZ18xMDV0ZWFjaGVyIjp7ImlkIjoiNTQyZTcxMDYxMjI2OTc1NWJhN2QxMzQ5IiwicHJvZ3JhbXMiOlsid3JpdGluZ18xMDVfSkhTX2hvbWV3b3JrX1BDXzAiLCJ3cml0aW5nXzEwNV9KSFNfaG9tZXdvcmtfUENfMSJdfSwibG9naW5lZCI6dHJ1ZSwid3JpdGluZ18xMDVKSFNfcHJvZ3JhbV9uYW1lIjoiJUU1JTlDJThCJUU0JUI4JUFEJUU1JUFGJUFCJUU0JUJEJTlDJUU5JTgwJTlGJUU2JTg4JTkwIiwid3JpdGluZ18xMDV1c2VyX2lkIjoiNTVjMDY0Y2JlNGIwMDJhYjA1NWQ3ZmQzIiwibmFtZSI6InBldGVy5r2YIiwicmVjYXB0Y2hhIjoiMDc3NSIsIkpIU19wcm9ncmFtX3VybCI6Ii93cml0aW5nXzEwNS9sZWFybmluZy5odG1sJTNGc2Nob29sX3N5c3RlbSUzREpIUyIsIndyaXRpbmdfMTA1dXNlcl9uYW1lIjoicGV0ZXIlRTYlQkQlOTgiLCJ1c2VyIjoiNTVjMDY0Y2JlNGIwMDJhYjA1NWQ3ZmQzIiwiZW1haWwiOiJwZXRlcisyQGVoYW5saW4uY29tLnR3In0="
        self.__transcoder = Transcoder()


    def __isNewVersion(self): return sys.version_info[0] > 2

    def test_unicode(self):
        base64_data = self.__base64_data
        bin_session = base64.b64decode(base64_data)

        if self.__isNewVersion(): bin_session = bin_session.decode('iso-8859-1')

        buffers = list( map( ord, bin_session )) 
        data = self.__transcoder.deserialize( buffers )
        principalData = data.get('principalData')
        name = principalData.get('name')

        if not self.__isNewVersion(): name = principalData.get('name').encode('utf-8')

        self.assertEqual(name, 'peter潘')
Пример #3
0
class MemorySessionAPI( object ):

    def __init__( self, hosts = ["127.0.0.1"] ):

        self.__hosts = hosts
        self.__mc = pylibmc.Client( hosts,  binary=True )
        self.__transcoder = Transcoder()

    def create_session_data( self ):

        creationTime = int ( time.time() * 1000 )

        session_data = dict()
        session_data["version"] = 2
        session_data["sessionFieldsDataLength"] = 84
        session_data["creationTime"] = creationTime
        session_data["lastAccessedTime"] = creationTime
        session_data["maxInactive"] = 7200
        session_data["isNew"] = True
        session_data["isValid"] = True
        session_data["thisAccessedTime"] = creationTime
        session_data["lastBackupTime"] = creationTime
        session_data["id"] = uuid.uuid1().hex
        session_data["idLength"] = len( session_data["id"] )
        session_data["authType"] = 0
        session_data["principalDataLength"] = 0
        session_data["savedRequestDataLength"] = 0
        session_data['principalData'] = {}

        return session_data

    def get_data( self, session_id ):

        b64_session = self.__mc.get( session_id )

        if b64_session:

            bin_session = base64.b64decode( b64_session )

            if sys.version_info[0] > 2: bin_session = bin_session.decode('iso-8859-1')

            buffers = list( map( ord, bin_session ))

            data = self.__transcoder.deserialize( buffers )


        else:

            data = self.create_session_data()

            #bin_data = ''.join( map( chr, self.__transcoder.serialize( data ) ) ) 

            #output = base64.b64encode( bin_data )

        return data

    def get_user_data( self, session_id ):

        data = self.get_data( session_id )

        return data.get( "principalData" )

    def update_user_data( self, session_id, user_data ):

         data = self.get_data( session_id )

         data["principalData"] = user_data

         bin_data = ''.join( map( chr, self.__transcoder.serialize( data ) ) )

         if sys.version_info[0] > 2: bin_data = bin_data.encode('iso-8859-1')

         output = base64.b64encode( bin_data )

         self.__mc.set( session_id , output)

    def has_login( self, session_id ):

        principalData = self.get_user_data( session_id )

        if principalData.get( "logined" ) is True:

            return True

        else:

            return False
Пример #4
0
    def __init__( self, hosts = ["127.0.0.1"] ):

        self.__hosts = hosts
        self.__mc = pylibmc.Client( hosts,  binary=True )
        self.__transcoder = Transcoder()
Пример #5
0
 def setUp(self):
     self.__base64_data = "AAIAVAAAAVZD345sAAABVkQZzNAAABwgMDEAAAFWRBnM0AAAAVZEGczQACBCREREMzVFN0FGNkMxOUJFRUFBMEE2NDJBNTBGMUZGMAAAAAAAAAAAeyJhZ3JlZWRUZXJtcyI6dHJ1ZSwiZGVmUGljIjoicHMwMC5wbmciLCJ2ZXJpZmllZCI6dHJ1ZSwid3JpdGluZ18xMDV0ZWFjaGVyIjp7ImlkIjoiNTQyZTcxMDYxMjI2OTc1NWJhN2QxMzQ5IiwicHJvZ3JhbXMiOlsid3JpdGluZ18xMDVfSkhTX2hvbWV3b3JrX1BDXzAiLCJ3cml0aW5nXzEwNV9KSFNfaG9tZXdvcmtfUENfMSJdfSwibG9naW5lZCI6dHJ1ZSwid3JpdGluZ18xMDVKSFNfcHJvZ3JhbV9uYW1lIjoiJUU1JTlDJThCJUU0JUI4JUFEJUU1JUFGJUFCJUU0JUJEJTlDJUU5JTgwJTlGJUU2JTg4JTkwIiwid3JpdGluZ18xMDV1c2VyX2lkIjoiNTVjMDY0Y2JlNGIwMDJhYjA1NWQ3ZmQzIiwibmFtZSI6InBldGVy5r2YIiwicmVjYXB0Y2hhIjoiMDc3NSIsIkpIU19wcm9ncmFtX3VybCI6Ii93cml0aW5nXzEwNS9sZWFybmluZy5odG1sJTNGc2Nob29sX3N5c3RlbSUzREpIUyIsIndyaXRpbmdfMTA1dXNlcl9uYW1lIjoicGV0ZXIlRTYlQkQlOTgiLCJ1c2VyIjoiNTVjMDY0Y2JlNGIwMDJhYjA1NWQ3ZmQzIiwiZW1haWwiOiJwZXRlcisyQGVoYW5saW4uY29tLnR3In0="
     self.__transcoder = Transcoder()
Пример #6
0
class TestStringMethods(unittest.TestCase):

  def setUp(self):
      self.data = dict(
          authType = 0,
          lastBackupTime = 1461125781055,
          isNew = True,
          isValid = True,
          maxInactive = 7200,
          principalData = dict(
              logined = True,
              verified = True,
              user = '******',
              agreedTerms = True
          ),
          creationTime = 1461125781055,
          principalDataLength = 0,
          version = 2,
          savedPrincipalDataLength = 0,
          savedRequestDataLength = 0,
          idLength = 32,
          thisAccessedTime = 1461125781055,
          sessionFieldsDataLength = 84,
          id = 'a3065b6606ae11e682397831c1c31f4a',
          lastAccessedTime = 1461125781055
      )
      self.transcoder = Transcoder()

  def __assertItemsEqual(self, v1, v2, msg):
      if self.__isNewVersion():
          self.assertCountEqual(v1, v2, msg)
      else:
          self.assertItemsEqual(v1, v2, msg)


  def __assertDictEqual(self, d1, d2, msg=None):
      if self.__isNewVersion():
          items = d1.items()
      else:
          items = d1.iteritems()

      for k,v1 in items:
          self.assertIn(k, d2, msg)
          v2 = d2[k]
          if(isinstance(v1, collections.Iterable)):
              self.__assertItemsEqual(v1, v2, msg)
          else:
              self.assertEqual(v1, v2, msg)
      return True

  def __isNewVersion(self): return sys.version_info[0] > 2

  def __getEncodingBData(self, encoding_data):
      text = ''.join( map( chr, encoding_data ) )
      if self.__isNewVersion():
          return text.encode('iso-8859-1')
      else:
          return text

  def __getDecodingBuffer(self, decoding_data):
      if self.__isNewVersion():
          text = decoding_data.decode('iso-8859-1')
      else:
          text = decoding_data
      return map( ord, text )

  def test_transcoder(self):
      data = self.data
      binary = self.transcoder.serialize(data)
      decode_data = self.transcoder.deserialize(binary)
      self.__assertDictEqual(data, decode_data)

  def test_base64(self):
      data = self.data
      encoding_data = self.transcoder.serialize( data )
      encoding_base64_data = base64.b64encode( self.__getEncodingBData(encoding_data) )
      decoding_base64_data = self.__getDecodingBuffer( base64.b64decode(encoding_base64_data) )
      result = self.transcoder.deserialize( list(decoding_base64_data) )
      self.__assertDictEqual( data, result )