| Yesudeep Mangalapilly | b5bab22 | 2011-08-24 16:53:39 +0530 | [diff] [blame] | 1 | '''Tests string operations.''' |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 2 | |
| Yesudeep Mangalapilly | 5802431 | 2011-08-11 01:48:25 +0530 | [diff] [blame] | 3 | from __future__ import absolute_import |
| 4 | |
| Yesudeep Mangalapilly | aed6e86 | 2011-08-11 01:59:50 +0530 | [diff] [blame] | 5 | import unittest2 |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 6 | |
| 7 | import rsa |
| 8 | |
| Sybren A. Stüvel | 15e257a | 2011-11-06 10:11:51 +0100 | [diff] [blame] | 9 | from constants import unicode_string |
| Yesudeep Mangalapilly | 5802431 | 2011-08-11 01:48:25 +0530 | [diff] [blame] | 10 | |
| Yesudeep Mangalapilly | aed6e86 | 2011-08-11 01:59:50 +0530 | [diff] [blame] | 11 | class StringTest(unittest2.TestCase): |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 12 | |
| 13 | def setUp(self): |
| Sybren A. Stüvel | 8c857f4 | 2011-07-30 20:39:02 +0200 | [diff] [blame] | 14 | (self.pub, self.priv) = rsa.newkeys(384) |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 15 | |
| 16 | def test_enc_dec(self): |
| 17 | |
| Yesudeep Mangalapilly | 5802431 | 2011-08-11 01:48:25 +0530 | [diff] [blame] | 18 | message = unicode_string.encode('utf-8') |
| 19 | print("\tMessage: %s" % message) |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 20 | |
| 21 | encrypted = rsa.encrypt(message, self.pub) |
| Yesudeep Mangalapilly | 5802431 | 2011-08-11 01:48:25 +0530 | [diff] [blame] | 22 | print("\tEncrypted: %s" % encrypted) |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 23 | |
| 24 | decrypted = rsa.decrypt(encrypted, self.priv) |
| Yesudeep Mangalapilly | 5802431 | 2011-08-11 01:48:25 +0530 | [diff] [blame] | 25 | print("\tDecrypted: %s" % decrypted) |
| Sybren A. Stüvel | ff3a1d0 | 2011-06-20 00:13:53 +0200 | [diff] [blame] | 26 | |
| 27 | self.assertEqual(message, decrypted) |
| 28 | |