aesxx/utils: add tests for *_bmp utils

4 files changed, 62 insertions, 0 deletions

diff --git a/aesxx/utils/CMakeLists.txt b/aesxx/utils/CMakeLists.txt
index 114efa5..44c4acc 100644
--- a/aesxx/utils/CMakeLists.txt
+++ b/aesxx/utils/CMakeLists.txt
@@ -44,3 +44,4 @@ find_package(Python3 REQUIRED COMPONENTS Interpreter)
 
 include(test/basic.cmake)
 include(test/block.cmake)
+include(test/bmp.cmake)
diff --git a/aesxx/utils/bmp/cipherfly_cbc.bmp b/aesxx/utils/bmp/cipherfly_cbc.bmp
index 664b557..e087685 100644
--- a/aesxx/utils/bmp/cipherfly_cbc.bmp
+++ b/aesxx/utils/bmp/cipherfly_cbc.bmp
diff --git a/aesxx/utils/bmp/cipherfly_ecb.bmp b/aesxx/utils/bmp/cipherfly_ecb.bmp
index 78de9a8..4c2660e 100644
--- a/aesxx/utils/bmp/cipherfly_ecb.bmp
+++ b/aesxx/utils/bmp/cipherfly_ecb.bmp
diff --git a/aesxx/utils/test/bmp.cmake b/aesxx/utils/test/bmp.cmake
new file mode 100644
index 0000000..868f89c
--- /dev/null
+++ b/aesxx/utils/test/bmp.cmake
@@ -0,0 +1,61 @@
+set(orig_dir "${CMAKE_CURRENT_SOURCE_DIR}/bmp")
+set(test_dir "${CMAKE_CURRENT_SOURCE_DIR}/out")
+
+add_test(
+    NAME create_test_dir
+    COMMAND ${CMAKE_COMMAND} -E make_directory "${test_dir}"
+)
+set_tests_properties(create_test_dir PROPERTIES FIXTURES_SETUP test_dir)
+add_test(
+    NAME remove_test_dir
+    COMMAND ${CMAKE_COMMAND} -E rm -r -- "${test_dir}"
+)
+set_tests_properties(remove_test_dir PROPERTIES FIXTURES_CLEANUP test_dir)
+
+set(args
+    -a aes128
+    -k 000102030405060708090a0b0c0d0e0f
+    -v 11111111111111112222222222222222
+)
+
+foreach(mode ecb cbc)
+    add_test(
+        NAME "encrypt_bmp_${mode}"
+        COMMAND util_encrypt_bmp ${args} -m "${mode}"
+            -i "${orig_dir}/butterfly.bmp"
+            -o "${test_dir}/cipherfly_${mode}.bmp"
+    )
+    set_tests_properties("encrypt_bmp_${mode}" PROPERTIES
+        FIXTURES_REQUIRED test_dir
+        FIXTURES_SETUP "encrypt_bmp_${mode}"
+    )
+    add_test(
+        NAME "encrypt_bmp_${mode}_compare"
+        COMMAND ${CMAKE_COMMAND} -E compare_files
+            "${orig_dir}/cipherfly_${mode}.bmp"
+            "${test_dir}/cipherfly_${mode}.bmp"
+    )
+    set_tests_properties("encrypt_bmp_${mode}_compare" PROPERTIES
+        FIXTURES_REQUIRED "test_dir;encrypt_bmp_${mode}"
+    )
+
+    add_test(
+        NAME "decrypt_bmp_${mode}"
+        COMMAND util_decrypt_bmp ${args} -m "${mode}"
+            -i "${test_dir}/cipherfly_${mode}.bmp"
+            -o "${test_dir}/butterfly_${mode}.bmp"
+    )
+    set_tests_properties("decrypt_bmp_${mode}" PROPERTIES
+        FIXTURES_REQUIRED test_dir
+        FIXTURES_SETUP "decrypt_bmp_${mode}"
+    )
+    add_test(
+        NAME "decrypt_bmp_${mode}_compare"
+        COMMAND ${CMAKE_COMMAND} -E compare_files
+            "${orig_dir}/butterfly.bmp"
+            "${test_dir}/butterfly_${mode}.bmp"
+    )
+    set_tests_properties("decrypt_bmp_${mode}_compare" PROPERTIES
+        FIXTURES_REQUIRED "test_dir;decrypt_bmp_${mode}"
+    )
+endforeach()