diff options
Diffstat (limited to 'lang/python/tests')
| -rwxr-xr-x | lang/python/tests/t-idiomatic.py | 19 | 
1 files changed, 14 insertions, 5 deletions
| diff --git a/lang/python/tests/t-idiomatic.py b/lang/python/tests/t-idiomatic.py index 05a377e4..37cfb64a 100755 --- a/lang/python/tests/t-idiomatic.py +++ b/lang/python/tests/t-idiomatic.py @@ -23,7 +23,15 @@ from pyme import core, constants, errors  import support  support.init_gpgme(constants.PROTOCOL_OpenPGP) -c = core.Context() + +# Both Context and Data can be used as context manager: +with core.Context() as c, core.Data() as d: +    c.get_engine_info() +    d.write(b"Halloechen") +    leak_c = c +    leak_d = d +assert leak_c.wrapped == None +assert leak_d.wrapped == None  # Demonstrate automatic wrapping of file-like objects with 'fileno'  # method. @@ -33,10 +41,11 @@ with tempfile.TemporaryFile() as source, \      source.write(b"Hallo Leute\n")      source.seek(0, os.SEEK_SET) -    c.op_sign(source, signed, constants.SIG_MODE_NORMAL) -    signed.seek(0, os.SEEK_SET) -    c.op_verify(signed, None, sink) -    result = c.op_verify_result() +    with core.Context() as c: +        c.op_sign(source, signed, constants.SIG_MODE_NORMAL) +        signed.seek(0, os.SEEK_SET) +        c.op_verify(signed, None, sink) +        result = c.op_verify_result()      assert len(result.signatures) == 1, "Unexpected number of signatures"      sig = result.signatures[0] | 
