From cded993dbbff8e8c777b95f4de0597cdf3f75c0d Mon Sep 17 00:00:00 2001 From: Paul Kehrer Date: Thu, 29 Jun 2017 18:43:42 -0500 Subject: dump_privatekey with FILETYPE_TEXT only supports RSA keys (#646) * dump_privatekey with FILETYPE_TEXT only supports RSA keys FILETYPE_TEXT is terrible but everyone hold their nose * also verify it's a pkey --- tests/test_crypto.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) (limited to 'tests') diff --git a/tests/test_crypto.py b/tests/test_crypto.py index 4197f11..916186b 100644 --- a/tests/test_crypto.py +++ b/tests/test_crypto.py @@ -2631,6 +2631,20 @@ class TestFunction(object): with pytest.raises(TypeError): dump_privatekey(FILETYPE_PEM, key, cipher=GOOD_CIPHER) + def test_dump_privatekey_not_rsa_key(self): + """ + `dump_privatekey` raises `TypeError` if called with a key that is + not RSA. + """ + key = PKey() + key.generate_key(TYPE_DSA, 512) + with pytest.raises(TypeError): + dump_privatekey(FILETYPE_TEXT, key) + + def test_dump_privatekey_invalid_pkey(self): + with pytest.raises(TypeError): + dump_privatekey(FILETYPE_TEXT, object()) + def test_dump_privatekey_unknown_cipher(self): """ `dump_privatekey` raises `ValueError` if called with an unrecognized -- cgit v1.2.1