github.com/jincm/wesharechain@v0.0.0-20210122032815-1537409ce26a/server/block/ut/crypto_test.py (about)

     1  #coding:utf-8
     2  
     3  from util import crypto_rc4
     4  import time
     5  import unittest
     6  
     7  
     8  class CryptoTestCase(unittest.TestCase):
     9      def setUp(self):
    10          self.content = "dsfdsfsdf"
    11          #ts_str = str(time.time() + 3600)
    12          self.token = "b7957cabe762962648"
    13          pass
    14  
    15      def tearDown(self):
    16          pass
    17  
    18      def test_encrypt(self):
    19          """
    20  
    21          :return:
    22          """
    23          token = crypto_rc4.encrypt(self.content, crypto_rc4.SECRET_KEY)
    24          print "token"+token
    25          self.assertTrue(token == self.token)
    26  
    27      def test_decrypt(self):
    28          _ = crypto_rc4.decrypt(self.token, crypto_rc4.SECRET_KEY)
    29          self.assertTrue(_ == self.content)
    30  
    31  
    32  if __name__ == "__main__":
    33      unittest.main()