def test_encryptorfile(tmpdir): # create a plaintext blob bigger than IO_BLOCK_SIZE plaintext1 = b"rvdmfki6iudmx8bb25tx1sozex3f4u0nm7uba4eibscgda0ckledcydz089qw1p1" repeat = int(1.5 * IO_BLOCK_SIZE / len(plaintext1)) plaintext = repeat * plaintext1 fn = tmpdir.join("data").strpath with open(fn, "w+b") as plain_fp: enc_fp = EncryptorFile(plain_fp, CONSTANT_TEST_RSA_PUBLIC_KEY) assert enc_fp.fileno() == plain_fp.fileno() assert enc_fp.readable() is False with pytest.raises(io.UnsupportedOperation): enc_fp.read(1) assert enc_fp.seekable() is False with pytest.raises(io.UnsupportedOperation): enc_fp.seek(1, os.SEEK_CUR) assert enc_fp.writable() is True enc_fp.write(plaintext) enc_fp.write(b"") assert enc_fp.tell() == len(plaintext) assert enc_fp.next_fp.tell() > len(plaintext) enc_fp.close() enc_fp.close() plain_fp.seek(0) dec_fp = DecryptorFile(plain_fp, CONSTANT_TEST_RSA_PRIVATE_KEY) assert dec_fp.fileno() == plain_fp.fileno() assert dec_fp.readable() is True assert dec_fp.seekable() is True assert dec_fp.writable() is False with pytest.raises(io.UnsupportedOperation): dec_fp.write(b"x") dec_fp.flush() result = dec_fp.read() assert plaintext == result