CI: Fix prov example test cryptography package version exception handling
This commit is contained in:
parent
acfee765f7
commit
fc067fda94
1 changed files with 8 additions and 2 deletions
|
@ -24,7 +24,7 @@ import proto
|
||||||
from .security import *
|
from .security import *
|
||||||
|
|
||||||
from cryptography.hazmat.backends import default_backend
|
from cryptography.hazmat.backends import default_backend
|
||||||
from cryptography.hazmat.primitives import hashes
|
from cryptography.hazmat.primitives import hashes, serialization
|
||||||
from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey, X25519PublicKey
|
from cryptography.hazmat.primitives.asymmetric.x25519 import X25519PrivateKey, X25519PublicKey
|
||||||
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
|
from cryptography.hazmat.primitives.ciphers import Cipher, algorithms, modes
|
||||||
|
|
||||||
|
@ -79,7 +79,13 @@ class Security1(Security):
|
||||||
def __generate_key(self):
|
def __generate_key(self):
|
||||||
# Generate private and public key pair for client
|
# Generate private and public key pair for client
|
||||||
self.client_private_key = X25519PrivateKey.generate()
|
self.client_private_key = X25519PrivateKey.generate()
|
||||||
self.client_public_key = self.client_private_key.public_key().public_bytes()
|
try:
|
||||||
|
self.client_public_key = self.client_private_key.public_key().public_bytes(
|
||||||
|
encoding=serialization.Encoding.Raw,
|
||||||
|
format=serialization.PublicFormat.Raw)
|
||||||
|
except TypeError:
|
||||||
|
# backward compatible call for older cryptography library
|
||||||
|
self.client_public_key = self.client_private_key.public_key().public_bytes()
|
||||||
|
|
||||||
def _print_verbose(self, data):
|
def _print_verbose(self, data):
|
||||||
if (self.verbose):
|
if (self.verbose):
|
||||||
|
|
Loading…
Reference in a new issue