Added tests for DFSPrivateKeyServiceImpl and PathEncryptionImpl classes

datasafe-directory/datasafe-directory-impl/src/test/java/de/adorsys/datasafe/directory/impl/profile/keys/DFSPrivateKeyServiceImplTest.java

@@ -0,0 +1,51 @@
+package de.adorsys.datasafe.directory.impl.profile.keys;
+
+
+import de.adorsys.datasafe.directory.api.profile.keys.DocumentKeyStoreOperations;
+import de.adorsys.datasafe.encrypiton.api.keystore.KeyStoreService;
+import de.adorsys.datasafe.encrypiton.api.types.UserID;
+import de.adorsys.datasafe.encrypiton.api.types.UserIDAuth;
+import de.adorsys.datasafe.types.api.shared.BaseMockitoTest;
+import de.adorsys.datasafe.types.api.types.ReadKeyPassword;
+import lombok.SneakyThrows;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.*;
+
+import java.security.*;
+
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+public class DFSPrivateKeyServiceImplTest extends BaseMockitoTest {
+    @Mock
+    private DocumentKeyStoreOperations keyStoreOper;
+    @Mock
+    private KeyStoreService keyStoreService;
+    DFSPrivateKeyServiceImpl privateKeyService;
+    @BeforeEach
+    public void setUp() {
+        privateKeyService = new DFSPrivateKeyServiceImpl(keyStoreOper);
+    }
+
+    @Test
+    @SneakyThrows
+    public void getKeyPair(){
+        ReadKeyPassword readKeyPassword = new ReadKeyPassword("keypass".toCharArray());
+        UserID user = new UserID("user1");
+        UserIDAuth userAuth = new UserIDAuth(user, readKeyPassword);
+
+        KeyPairGenerator KeyGen = KeyPairGenerator.getInstance("RSA");
+        KeyPair keyPair  = KeyGen.generateKeyPair();
+
+        when(keyStoreOper.getKeyPair(any())).thenReturn(keyPair);
+
+        KeyPair keyPair1 = privateKeyService.getKeyPair(userAuth);
+        Assertions.assertEquals(keyPair.getPublic(), keyPair1.getPublic());
+        Assertions.assertEquals(keyPair.getPrivate(), keyPair1.getPrivate());
+    }
+
+}
+

datasafe-encryption/datasafe-encryption-impl/src/test/java/de/adorsys/datasafe/encrypiton/impl/pathencryption/PathEncryptionImplTest.java

@@ -0,0 +1,71 @@
+package de.adorsys.datasafe.encrypiton.impl.pathencryption;
+
+import de.adorsys.datasafe.directory.api.profile.keys.PrivateKeyService;
+import de.adorsys.datasafe.encrypiton.api.keystore.KeyStoreService;
+import de.adorsys.datasafe.encrypiton.api.pathencryption.encryption.SymmetricPathEncryptionService;
+import de.adorsys.datasafe.encrypiton.api.types.UserID;
+import de.adorsys.datasafe.encrypiton.api.types.UserIDAuth;
+import de.adorsys.datasafe.encrypiton.api.types.encryption.EncryptionConfig;
+import de.adorsys.datasafe.encrypiton.api.types.encryption.KeyCreationConfig;
+import de.adorsys.datasafe.encrypiton.api.types.keystore.*;
+import de.adorsys.datasafe.encrypiton.impl.keystore.KeyStoreServiceImpl;
+import de.adorsys.datasafe.types.api.resource.Uri;
+import de.adorsys.datasafe.types.api.shared.BaseMockitoTest;
+import de.adorsys.datasafe.types.api.types.ReadKeyPassword;
+import de.adorsys.datasafe.types.api.types.ReadStorePassword;
+import de.adorsys.keymanagement.juggler.services.DaggerBCJuggler;
+import org.junit.jupiter.api.Assertions;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.*;
+
+import javax.crypto.SecretKey;
+import java.security.KeyStore;
+
+import static org.mockito.ArgumentMatchers.any;
+import static org.mockito.Mockito.when;
+
+public class PathEncryptionImplTest extends BaseMockitoTest {
+    String uriString = "https://192.168.178.0.1:9090/minio/first/folder";
+    private final KeyStoreService keyStoreService = new KeyStoreServiceImpl(
+            EncryptionConfig.builder().build().getKeystore(),
+            DaggerBCJuggler.builder().build()
+    );
+    @Mock
+    private SymmetricPathEncryptionService symmetricPathEncryptionService;
+    @Mock
+    private PrivateKeyService privateKeyService;
+
+    PathEncryptionImpl pathEncryption;
+
+
+
+    @BeforeEach
+    void setUp() {
+        pathEncryption = new PathEncryptionImpl(symmetricPathEncryptionService, privateKeyService);
+    }
+
+    @Test
+    public void testPathEncryption() {
+        ReadStorePassword storePassword = new ReadStorePassword("storepass");
+        ReadKeyPassword readKeyPassword = new ReadKeyPassword("keypass".toCharArray());
+        KeyStoreAuth keyStoreAuth = new KeyStoreAuth(storePassword, readKeyPassword);
+        KeyID keyID = new KeyID("secret");
+
+        KeyCreationConfig config = KeyCreationConfig.builder().signKeyNumber(0).encKeyNumber(1).build();
+        KeyStore keystore = keyStoreService.createKeyStore(keyStoreAuth,config);
+        KeyStoreAccess keyStoreAccess = new KeyStoreAccess(keystore, keyStoreAuth);
+
+        SecretKey secretKey = keyStoreService.getSecretKey(keyStoreAccess, keyID);
+        SecretKeyIDWithKey secretKeyID = new SecretKeyIDWithKey(keyID, secretKey);
+
+        when(symmetricPathEncryptionService.encrypt(any(), any())).thenReturn(new Uri(uriString + ".enc"));
+        when(privateKeyService.pathEncryptionSecretKey(any())).thenReturn(new AuthPathEncryptionSecretKey(secretKeyID,secretKeyID));
+
+        UserID user = new UserID("user1");
+        UserIDAuth userAuth = new UserIDAuth(user, readKeyPassword);
+
+        Uri encryptedPath = pathEncryption.encrypt(userAuth, new Uri(uriString));
+        Assertions.assertEquals(encryptedPath, new Uri(uriString + ".enc"));
+    }
+}