def test_reset(self):
rc4 = crypto.RC4(KEY, True)
assert rc4.crypt(PLAIN) == CIPHER
assert rc4.crypt(PLAIN) == CIPHER
rc4 = crypto.RC4(KEY, False)
assert rc4.crypt(PLAIN) == CIPHER
rc4.reset()
assert rc4.crypt(PLAIN) == CIPHER
def __init__(self, key):
self.rc4enc = crypto.RC4(key)
self.rc4dec = crypto.RC4(key)
def __init__(self, key):
self.key = key
self.rc4 = crypto.RC4(key, True)
def test_encrypt(self):
rc4 = crypto.RC4(KEY)
assert rc4.crypt(PLAIN) == CIPHER
def test_set_key(self):
rc4 = crypto.RC4(b"a" * 16)
assert rc4.crypt(PLAIN) != CIPHER
rc4.set_key(KEY)
assert rc4.crypt(PLAIN) == CIPHER
def test_stream(self):
rc4 = crypto.RC4(KEY)
assert rc4.crypt(PLAIN[:15]) == CIPHER[:15]
assert rc4.crypt(PLAIN[15:]) == CIPHER[15:]
def test_decrypt(self):
rc4 = crypto.RC4(KEY)
assert rc4.crypt(CIPHER) == PLAIN
