chore(test/tls)!: Improve stability of PQC-related tests

Author: c0r0n3r

test/tls/test_all.py

@@ -184,13 +184,12 @@ def test_real(self):
             TlsCipherSuite.TLS_CHACHA20_POLY1305_SHA256,
         ]))
         self.assertEqual(result.dhparams, None)
-        self.assertEqual(set(result.curves.curves), set([
-            TlsNamedCurve.X25519_KYBER_512_DRAFT00,
-            TlsNamedCurve.X25519_KYBER_768_DRAFT00,
-            TlsNamedCurve.X25519,
-            TlsNamedCurve.SECP256R1,
-            TlsNamedCurve.SECP384R1,
-        ]))
+        curves = result.curves.curves
+        # different instances run with different configuration, the following is the common subset
+        self.assertIn(TlsNamedCurve.X25519_KYBER_512_DRAFT00, curves)
+        self.assertIn(TlsNamedCurve.X25519_KYBER_768_DRAFT00, curves)
+        self.assertIn(TlsNamedCurve.X25519, curves)
+        self.assertIn(TlsNamedCurve.SECP256R1, curves)
 
     def test_markdown(self):
         result = self.get_result('rc4-md5.badssl.com', 443)

test/tls/test_curves.py

@@ -140,16 +140,13 @@ def test_tls_1_3(self):
         )
 
     def test_pqc(self):
-        self.assertEqual(
-            self.get_result('pq.cloudflareresearch.com', 443, TlsProtocolVersion(TlsVersion.TLS1_3)).curves,
-            [
-                TlsNamedCurve.X25519_KYBER_512_DRAFT00,
-                TlsNamedCurve.X25519_KYBER_768_DRAFT00,
-                TlsNamedCurve.X25519,
-                TlsNamedCurve.SECP256R1,
-                TlsNamedCurve.SECP384R1,
-            ]
-        )
+        curves = self.get_result('pq.cloudflareresearch.com', 443, TlsProtocolVersion(TlsVersion.TLS1_3)).curves
+
+        # different instances run with different configuration, the following is the common subset
+        self.assertIn(TlsNamedCurve.X25519_KYBER_512_DRAFT00, curves)
+        self.assertIn(TlsNamedCurve.X25519_KYBER_768_DRAFT00, curves)
+        self.assertIn(TlsNamedCurve.X25519, curves)
+        self.assertIn(TlsNamedCurve.SECP256R1, curves)
 
     def test_plain_text_response(self):
         threaded_server = L7ServerTlsTest(