impl default

litentry · Jun 15, 2024 · 72ad32d · 72ad32d
1 parent 2d383fe
commit 72ad32d
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git a/tee-worker/core-primitives/sgx/crypto/src/mocks.rs b/tee-worker/core-primitives/sgx/crypto/src/mocks.rs
@@ -27,14 +27,14 @@ use sgx_crypto::rsa::Rsa3072KeyPair;
 #[derive(Default)]
 pub struct KeyRepositoryMock<KeyType>
 where
-	KeyType: Clone,
+	KeyType: Clone + Default,
 {
 	key: RwLock<KeyType>,
 }
 
 impl<KeyType> KeyRepositoryMock<KeyType>
 where
-	KeyType: Clone,
+	KeyType: Clone + Default,
 {
 	pub fn new(key: KeyType) -> Self {
 		KeyRepositoryMock { key: RwLock::new(key) }
@@ -43,7 +43,7 @@ where
 
 impl<KeyType> AccessKey for KeyRepositoryMock<KeyType>
 where
-	KeyType: Clone,
+	KeyType: Clone + Default,
 {
 	type KeyType = KeyType;
 
@@ -54,7 +54,7 @@ where
 
 impl<KeyType> MutateKey<KeyType> for KeyRepositoryMock<KeyType>
 where
-	KeyType: Clone,
+	KeyType: Clone + Default,
 {
 	fn update_key(&self, key: KeyType) -> Result<()> {
 		let mut lock = self.key.write().unwrap();

diff --git a/tee-worker/core-primitives/test/src/mock/shielding_crypto_mock.rs b/tee-worker/core-primitives/test/src/mock/shielding_crypto_mock.rs
@@ -38,6 +38,12 @@ pub struct ShieldingCryptoMock {
 	use_dummy_enc_dec: bool,
 }
 
+impl Default for ShieldingCryptoMock {
+	fn default() -> Self {
+		Self::new(false)
+	}
+}
+
 impl ShieldingCryptoMock {
 	pub fn new(use_dummy_enc_dec: bool) -> Self {
 		ShieldingCryptoMock {