Adding test for XML encryption/decryption

Author: behzadmehmood

.github/workflows/test.yml

@@ -497,3 +497,16 @@ jobs:
       run: |
         export PATH="/usr/lib/ccache:/usr/local/opt/ccache/libexec:$PATH"
         ./.github/scripts/build.sh
+
+  test_xml_encryption:
+    name: 'Encryption/Decryption test'
+    strategy:
+      fail-fast: false
+    needs: Build
+    runs-on: ubuntu-22.04
+
+    steps:
+    - name: Test_Encryption_Decryption
+      shell: bash
+      run: |
+        valgrind ./build/libs/libdecrypt/test/test_encrypt_decrypt

libs/libdecrypt/CMakeLists.txt

@@ -44,4 +44,6 @@ target_link_libraries(libdecrypt
                         ${OPENSSL_LIBRARIES}
 )
 install(TARGETS libdecrypt DESTINATION bin)
-install(FILES ${LIB_HEADERS} DESTINATION include/libdecrypt)
+install(FILES ${LIB_HEADERS} DESTINATION include/libdecrypt)
+
+add_subdirectory(test)

libs/libdecrypt/test/CMakeLists.txt

@@ -0,0 +1,19 @@
+cmake_minimum_required(VERSION 3.10)
+project(EncryptionDecryptionTest)
+
+# Find OpenSSL
+find_package(OpenSSL REQUIRED)
+
+add_executable(test_encrypt_decrypt src/test.cpp src/main.cpp)
+
+set(PUBLIC_KEY_FILE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../../libencrypt/public_key.pem")
+target_compile_definitions(test_encrypt_decrypt PRIVATE PUBLIC_KEY_FILE="${PUBLIC_KEY_FILE_PATH}")
+
+include_directories("${CMAKE_CURRENT_SOURCE_DIR}/../../libencrypt/src"
+                        "${CMAKE_CURRENT_SOURCE_DIR}/../libdecrypt/src"
+                        "${CMAKE_CURRENT_SOURCE_DIR}/src")
+
+target_link_libraries(test_encrypt_decrypt
+                        libpugixml
+                        libdecrypt
+                        libencrypt)

libs/libdecrypt/test/src/main.cpp

@@ -0,0 +1,33 @@
+#include "test.h"
+
+int main() {
+    std::string testFilePath = "test.xml";
+    std::string publicKeyFile = PUBLIC_KEY_FILE; // Path to your public key file
+    std::string encryptedFilePath = "test.xmle";
+
+    // Step 1: Create a test XML file
+    createTestXMLFile(testFilePath);
+    
+    std::string originalContent = readFileToString(testFilePath);
+
+    // Step 2: Encrypt XML content
+    Encryption encryption;
+    if (!encryption.encryptFile(publicKeyFile, testFilePath)) {
+        std::cerr << "Encryption failed" << std::endl;
+        return 1;
+    }
+
+    // Step 3: Decrypt XML content
+    Decryption decryption(encryptedFilePath);
+    std::string decryptedContent = decryption.getDecryptedContent();
+
+    // Step 4: Compare original XML content with decrypted XML content
+    if (decryptedContent == originalContent) {
+        std::cout << "Test passed: Decrypted content matches original content" << std::endl;
+    } else {
+        std::cerr << "Test failed: Decrypted content does not match original content" << std::endl;
+        return 1;
+    }
+
+    return 0;
+}

libs/libdecrypt/test/src/test.cpp

@@ -0,0 +1,22 @@
+#include "test.h"
+
+#ifndef PUBLIC_KEY_FILE
+#define PUBLIC_KEY_FILE "public_key.pem" // Default public key file
+#endif
+
+// Function to create a sample XML file for testing
+void createTestXMLFile(const std::string& filePath) {
+    std::ofstream outFile(filePath);
+    outFile << "<Test>\n";
+    outFile << "  <Data>Encryption Data</Data>\n";
+    outFile << "</Test>";
+    outFile.close();
+}
+
+// Function to read the contents of a file into a string
+std::string readFileToString(const std::string& filePath) {
+    std::ifstream inFile(filePath);
+    std::string content((std::istreambuf_iterator<char>(inFile)), std::istreambuf_iterator<char>());
+    inFile.close();
+    return content;
+}

libs/libdecrypt/test/src/test.h

@@ -0,0 +1,14 @@
+#ifndef TEST_H
+#define TEST_H
+
+#include <iostream>
+#include <fstream>
+#include <string>
+#include "encryption.h"
+#include "decryption.h"
+
+void createTestXMLFile(const std::string& filePath);
+std::string readFileToString(const std::string& filePath);
+
+
+#endif // TEST_H