Skip to content

Commit a3eae49

Browse files
committed
First round of review
1 parent 20e8c49 commit a3eae49

9 files changed

Lines changed: 40 additions & 23 deletions

File tree

tests/dh/dh-test.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,8 @@ def test_dhparam_noout(self):
6767
self.addCleanup(lambda: os.remove(params_file)
6868
if os.path.exists(params_file) else None)
6969

70-
run_wolfssl("dhparam", "1024", "-out", params_file)
70+
r = run_wolfssl("dhparam", "1024", "-out", params_file)
71+
self.assertEqual(r.returncode, 0, r.stderr)
7172

7273
r = run_wolfssl("dhparam", "-in", params_file, "-noout")
7374
self.assertEqual(r.returncode, 0, r.stderr)
@@ -78,7 +79,8 @@ def test_dhparam_genkey(self):
7879
self.addCleanup(lambda: os.remove(params_file)
7980
if os.path.exists(params_file) else None)
8081

81-
run_wolfssl("dhparam", "1024", "-out", params_file)
82+
r = run_wolfssl("dhparam", "1024", "-out", params_file)
83+
self.assertEqual(r.returncode, 0, r.stderr)
8284

8385
r = run_wolfssl("dhparam", "-in", params_file, "-genkey")
8486
self.assertEqual(r.returncode, 0, r.stderr)
@@ -90,7 +92,8 @@ def test_dhparam_genkey_noout(self):
9092
self.addCleanup(lambda: os.remove(params_file)
9193
if os.path.exists(params_file) else None)
9294

93-
run_wolfssl("dhparam", "1024", "-out", params_file)
95+
r = run_wolfssl("dhparam", "1024", "-out", params_file)
96+
self.assertEqual(r.returncode, 0, r.stderr)
9497

9598
r = run_wolfssl("dhparam", "-in", params_file, "-genkey", "-noout")
9699
self.assertEqual(r.returncode, 0, r.stderr)

tests/dsa/dsa-test.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,8 @@ def test_dsaparam_noout(self):
5454
self.addCleanup(lambda: os.remove(params_file)
5555
if os.path.exists(params_file) else None)
5656

57-
run_wolfssl("dsaparam", "-out", params_file, "1024")
57+
r = run_wolfssl("dsaparam", "-out", params_file, "1024")
58+
self.assertEqual(r.returncode, 0, r.stderr)
5859

5960
r = run_wolfssl("dsaparam", "-in", params_file, "-noout")
6061
self.assertEqual(r.returncode, 0, r.stderr)
@@ -65,7 +66,8 @@ def test_dsaparam_genkey(self):
6566
self.addCleanup(lambda: os.remove(params_file)
6667
if os.path.exists(params_file) else None)
6768

68-
run_wolfssl("dsaparam", "-out", params_file, "1024")
69+
r = run_wolfssl("dsaparam", "-out", params_file, "1024")
70+
self.assertEqual(r.returncode, 0, r.stderr)
6971

7072
r = run_wolfssl("dsaparam", "-in", params_file, "-genkey")
7173
self.assertEqual(r.returncode, 0, r.stderr)
@@ -77,7 +79,8 @@ def test_dsaparam_genkey_noout(self):
7779
self.addCleanup(lambda: os.remove(params_file)
7880
if os.path.exists(params_file) else None)
7981

80-
run_wolfssl("dsaparam", "-out", params_file, "1024")
82+
r = run_wolfssl("dsaparam", "-out", params_file, "1024")
83+
self.assertEqual(r.returncode, 0, r.stderr)
8184

8285
r = run_wolfssl("dsaparam", "-in", params_file, "-genkey", "-noout")
8386
self.assertEqual(r.returncode, 0, r.stderr)

tests/encrypt/enc-test.py

Lines changed: 17 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -152,9 +152,10 @@ def test_openssl_enc_wolfssl_dec(self):
152152
orig = os.path.join(CERTS_DIR, "crl.der")
153153
self._cleanup(enc, dec)
154154

155-
subprocess.run(["openssl", "enc", "-base64", "-aes-256-cbc",
156-
"-k", "test password", "-in", orig, "-out", enc],
157-
capture_output=True)
155+
ossl = subprocess.run(["openssl", "enc", "-base64", "-aes-256-cbc",
156+
"-k", "test password", "-in", orig, "-out", enc],
157+
capture_output=True)
158+
self.assertEqual(ossl.returncode, 0, ossl.stderr)
158159

159160
r = run_enc("enc", "-base64", "-d", "-aes-256-cbc",
160161
"-in", enc, "-out", dec, password="test password")
@@ -171,9 +172,10 @@ def test_wolfssl_enc_openssl_dec(self):
171172
"-in", orig, "-out", enc, password="test password")
172173
self.assertEqual(r.returncode, 0, r.stderr)
173174

174-
subprocess.run(["openssl", "enc", "-base64", "-d", "-aes-256-cbc",
175-
"-k", "test password", "-in", enc, "-out", dec],
176-
capture_output=True)
175+
ossl = subprocess.run(["openssl", "enc", "-base64", "-d", "-aes-256-cbc",
176+
"-k", "test password", "-in", enc, "-out", dec],
177+
capture_output=True)
178+
self.assertEqual(ossl.returncode, 0, ossl.stderr)
177179
self.assertTrue(filecmp.cmp(orig, dec, shallow=False))
178180

179181
def test_pbkdf2_openssl_enc_wolfssl_dec(self):
@@ -182,9 +184,10 @@ def test_pbkdf2_openssl_enc_wolfssl_dec(self):
182184
orig = os.path.join(CERTS_DIR, "crl.der")
183185
self._cleanup(enc, dec)
184186

185-
subprocess.run(["openssl", "enc", "-base64", "-pbkdf2", "-aes-256-cbc",
186-
"-k", "long test password", "-in", orig, "-out", enc],
187-
capture_output=True)
187+
ossl = subprocess.run(["openssl", "enc", "-base64", "-pbkdf2", "-aes-256-cbc",
188+
"-k", "long test password", "-in", orig, "-out", enc],
189+
capture_output=True)
190+
self.assertEqual(ossl.returncode, 0, ossl.stderr)
188191

189192
r = run_enc("enc", "-base64", "-d", "-pbkdf2", "-aes-256-cbc",
190193
"-in", enc, "-out", dec, password="long test password")
@@ -201,10 +204,11 @@ def test_pbkdf2_wolfssl_enc_openssl_dec(self):
201204
"-in", orig, "-out", enc, password="long test password")
202205
self.assertEqual(r.returncode, 0, r.stderr)
203206

204-
subprocess.run(["openssl", "enc", "-base64", "-d", "-pbkdf2",
205-
"-aes-256-cbc", "-k", "long test password",
206-
"-in", enc, "-out", dec],
207-
capture_output=True)
207+
ossl = subprocess.run(["openssl", "enc", "-base64", "-d", "-pbkdf2",
208+
"-aes-256-cbc", "-k", "long test password",
209+
"-in", enc, "-out", dec],
210+
capture_output=True)
211+
self.assertEqual(ossl.returncode, 0, ossl.stderr)
208212
self.assertTrue(filecmp.cmp(orig, dec, shallow=False))
209213

210214
def test_pbkdf2_wolfssl_pass_flag(self):

tests/wolfclu_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -32,13 +32,15 @@ def _find_wolfssl_bin():
3232
CERTS_DIR = os.path.join(".", "certs")
3333

3434

35-
def run_wolfssl(*args, stdin_data=None):
35+
def run_wolfssl(*args, stdin_data=None, timeout=60):
3636
"""Run the wolfssl binary with the given arguments.
3737
3838
Returns a CompletedProcess instance.
39+
A default timeout of 60 seconds prevents indefinite hangs in CI.
40+
Network-facing tests (s_client, ocsp) manage their own timeouts.
3941
"""
4042
cmd = [WOLFSSL_BIN] + list(args)
41-
kwargs = dict(capture_output=True, text=True)
43+
kwargs = dict(capture_output=True, text=True, timeout=timeout)
4244
if stdin_data is not None:
4345
kwargs["input"] = stdin_data
4446
else:

tests/x509/CRL-verify-test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
1+
#!/usr/bin/env python3
12
"""Tests for wolfssl crl (converted from CRL-verify-test.sh)."""
23

34
import os
45
import sys
56
import unittest
67

78
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
8-
from wolfclu_test import WOLFSSL_BIN, CERTS_DIR, run_wolfssl
9+
from wolfclu_test import CERTS_DIR, run_wolfssl
910

1011

1112
def _has_crl():

tests/x509/x509-ca-test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#!/usr/bin/env python3
12
"""Tests for wolfssl ca (converted from x509-ca-test.sh)."""
23

34
import os

tests/x509/x509-process-test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#!/usr/bin/env python3
12
"""Tests for wolfssl x509 processing (converted from x509-process-test.sh)."""
23

34
import os

tests/x509/x509-req-test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,4 @@
1+
#!/usr/bin/env python3
12
"""Tests for wolfssl req and x509 -req (converted from x509-req-test.sh)."""
23

34
import os

tests/x509/x509-verify-test.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,12 @@
1+
#!/usr/bin/env python3
12
"""Tests for wolfssl verify (converted from x509-verify-test.sh)."""
23

34
import os
45
import sys
56
import unittest
67

78
sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
8-
from wolfclu_test import WOLFSSL_BIN, CERTS_DIR, run_wolfssl
9+
from wolfclu_test import CERTS_DIR, run_wolfssl
910

1011

1112
def _has_crl():

0 commit comments

Comments
 (0)