diff options
author | saturneric <[email protected]> | 2024-01-18 12:58:37 +0000 |
---|---|---|
committer | saturneric <[email protected]> | 2024-01-18 12:58:37 +0000 |
commit | e0d2c4021483e6e1366c1050ccb5a30e53e172bb (patch) | |
tree | 5d85739479622e7c672aea45f84d4e2f577f9332 /src/test/core/GpgCoreTestFileBasicOpera.cpp | |
parent | fix: correct a fault (diff) | |
download | GpgFrontend-e0d2c4021483e6e1366c1050ccb5a30e53e172bb.tar.gz GpgFrontend-e0d2c4021483e6e1366c1050ccb5a30e53e172bb.zip |
fix: add test cases and solve discovered issues
Diffstat (limited to 'src/test/core/GpgCoreTestFileBasicOpera.cpp')
-rw-r--r-- | src/test/core/GpgCoreTestFileBasicOpera.cpp | 386 |
1 files changed, 225 insertions, 161 deletions
diff --git a/src/test/core/GpgCoreTestFileBasicOpera.cpp b/src/test/core/GpgCoreTestFileBasicOpera.cpp index d5b0f52d..db338893 100644 --- a/src/test/core/GpgCoreTestFileBasicOpera.cpp +++ b/src/test/core/GpgCoreTestFileBasicOpera.cpp @@ -28,11 +28,8 @@ #include "GpgCoreTest.h" #include "core/GpgModel.h" -#include "core/function/basic/ChannelObject.h" -#include "core/function/gpg/GpgBasicOperator.h" #include "core/function/gpg/GpgFileOpera.h" #include "core/function/gpg/GpgKeyGetter.h" -#include "core/function/result_analyse/GpgDecryptResultAnalyse.h" #include "core/model/GpgDecryptResult.h" #include "core/model/GpgEncryptResult.h" #include "core/model/GpgSignResult.h" @@ -43,197 +40,264 @@ namespace GpgFrontend::Test { TEST_F(GpgCoreTest, CoreFileEncryptDecrTest) { - std::atomic_bool callback_called_flag{false}; + auto encrypt_key = GpgKeyGetter::GetInstance().GetPubkey( + "E87C6A2D8D95C818DE93B3AE6A2764F8298DEB29"); + + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); + auto output_file = GetTempFilePath(); + + auto [err, data_object] = GpgFileOpera::GetInstance().EncryptFileSync( + {encrypt_key}, input_file, true, output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgEncryptResult>())); + auto result = ExtractParams<GpgEncryptResult>(data_object, 0); + ASSERT_TRUE(result.InvalidRecipients().empty()); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = GpgFileOpera::GetInstance().DecryptFileSync( + output_file, decrpypt_output_file); + auto decr_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_FALSE(decr_result.Recipients().empty()); + ASSERT_EQ(decr_result.Recipients()[0].keyid, "6A2764F8298DEB29"); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); +} + +TEST_F(GpgCoreTest, CoreFileEncryptDecrBinaryTest) { auto encrypt_key = GpgKeyGetter::GetInstance().GetPubkey( "E87C6A2D8D95C818DE93B3AE6A2764F8298DEB29"); - auto input_file = CreateTempFileAndWriteData("Hello GpgFrontend!"); + + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); auto output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().EncryptFile( - {encrypt_key}, input_file, true, output_file, - [output_file, &callback_called_flag](GpgError err, - const DataObjectPtr& data_obj) { - ASSERT_TRUE((data_obj->Check<GpgEncryptResult>())); - - auto result = ExtractParams<GpgEncryptResult>(data_obj, 0); - ASSERT_TRUE(result.InvalidRecipients().empty()); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - - auto decrpypt_output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().DecryptFile( - output_file, decrpypt_output_file, - [decrpypt_output_file, &callback_called_flag]( - GpgError err, const DataObjectPtr& data_obj) { - auto d_result = ExtractParams<GpgDecryptResult>(data_obj, 0); - - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - ASSERT_FALSE(d_result.Recipients().empty()); - ASSERT_EQ(d_result.Recipients()[0].keyid, "6A2764F8298DEB29"); - - const auto [read_success, buffer] = - ReadFileGFBuffer(decrpypt_output_file); - ASSERT_TRUE(read_success); - ASSERT_EQ(buffer, GFBuffer(QString("Hello GpgFrontend!"))); - - // stop waiting - callback_called_flag = true; - }); - }); - - int retry_count = 1000; - while (!callback_called_flag && retry_count-- > 0) { - std::this_thread::sleep_for(std::chrono::milliseconds(10)); - } - - ASSERT_TRUE(callback_called_flag); + auto [err, data_object] = GpgFileOpera::GetInstance().EncryptFileSync( + {encrypt_key}, input_file, false, output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgEncryptResult>())); + auto result = ExtractParams<GpgEncryptResult>(data_object, 0); + ASSERT_TRUE(result.InvalidRecipients().empty()); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = GpgFileOpera::GetInstance().DecryptFileSync( + output_file, decrpypt_output_file); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object_0->Check<GpgDecryptResult>())); + auto decr_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + ASSERT_FALSE(decr_result.Recipients().empty()); + ASSERT_EQ(decr_result.Recipients()[0].keyid, "6A2764F8298DEB29"); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); } TEST_F(GpgCoreTest, CoreFileEncryptSymmetricDecrTest) { - std::atomic_bool callback_called_flag{false}; + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); + auto output_file = GetTempFilePath(); - auto input_file = CreateTempFileAndWriteData("Hello GpgFrontend!"); + auto [err, data_object] = + GpgFileOpera::GetInstance().EncryptFileSymmetricSync(input_file, true, + output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgEncryptResult>())); + auto result = ExtractParams<GpgEncryptResult>(data_object, 0); + ASSERT_TRUE(result.InvalidRecipients().empty()); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = GpgFileOpera::GetInstance().DecryptFileSync( + output_file, decrpypt_output_file); + + ASSERT_TRUE((data_object_0->Check<GpgDecryptResult>())); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + auto decrypt_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + ASSERT_TRUE(decrypt_result.Recipients().empty()); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); +} + +TEST_F(GpgCoreTest, CoreFileEncryptSymmetricDecrBinaryTest) { + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); auto output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().EncryptFileSymmetric( - input_file, true, output_file, - [&callback_called_flag, output_file](GpgError err, - const DataObjectPtr& data_obj) { - ASSERT_TRUE((data_obj->Check<GpgEncryptResult>())); - auto result = ExtractParams<GpgEncryptResult>(data_obj, 0); - ASSERT_TRUE(result.InvalidRecipients().empty()); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - - auto decrpypt_output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().DecryptFile( - output_file, decrpypt_output_file, - [&callback_called_flag, decrpypt_output_file]( - GpgError err, const DataObjectPtr& data_obj) { - ASSERT_TRUE((data_obj->Check<GpgDecryptResult>())); - - auto decrypt_result = - ExtractParams<GpgDecryptResult>(data_obj, 0); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - ASSERT_TRUE(decrypt_result.Recipients().empty()); - - const auto [read_success, buffer] = - ReadFileGFBuffer(decrpypt_output_file); - ASSERT_TRUE(read_success); - ASSERT_EQ(buffer, GFBuffer(QString("Hello GpgFrontend!"))); - - // stop waiting - callback_called_flag = true; - }); - }); - - int retry_count = 1000; - while (!callback_called_flag && retry_count-- > 0) { - std::this_thread::sleep_for(std::chrono::milliseconds(10)); - } - - ASSERT_TRUE(callback_called_flag); + auto [err, data_object] = + GpgFileOpera::GetInstance().EncryptFileSymmetricSync(input_file, false, + output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgEncryptResult>())); + auto result = ExtractParams<GpgEncryptResult>(data_object, 0); + ASSERT_TRUE(result.InvalidRecipients().empty()); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = GpgFileOpera::GetInstance().DecryptFileSync( + output_file, decrpypt_output_file); + + ASSERT_TRUE((data_object_0->Check<GpgDecryptResult>())); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + auto decrypt_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + ASSERT_TRUE(decrypt_result.Recipients().empty()); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); } TEST_F(GpgCoreTest, CoreFileSignVerifyNormalTest) { - std::atomic_bool callback_called_flag{false}; + auto sign_key = GpgKeyGetter::GetInstance().GetPubkey( + "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); + auto input_file = CreateTempFileAndWriteData("Hello GpgFrontend!"); + auto output_file = GetTempFilePath(); + auto [err, data_object] = GpgFileOpera::GetInstance().SignFileSync( + {sign_key}, input_file, true, output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgSignResult>())); + auto result = ExtractParams<GpgSignResult>(data_object, 0); + ASSERT_TRUE(result.InvalidSigners().empty()); + + auto [err_0, data_object_0] = + GpgFileOpera::GetInstance().VerifyFileSync(input_file, output_file); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object_0->Check<GpgVerifyResult>())); + auto verify_result = ExtractParams<GpgVerifyResult>(data_object_0, 0); + ASSERT_FALSE(verify_result.GetSignature().empty()); + ASSERT_EQ(verify_result.GetSignature().at(0).GetFingerprint(), + "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); +} + +TEST_F(GpgCoreTest, CoreFileSignVerifyNormalBinaryTest) { auto sign_key = GpgKeyGetter::GetInstance().GetPubkey( "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); auto input_file = CreateTempFileAndWriteData("Hello GpgFrontend!"); auto output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().SignFile( - {sign_key}, input_file, true, output_file, - [&callback_called_flag, input_file, output_file]( - GpgError err, const DataObjectPtr& data_obj) { - ASSERT_TRUE((data_obj->Check<GpgSignResult>())); - auto result = ExtractParams<GpgSignResult>(data_obj, 0); - ASSERT_TRUE(result.InvalidSigners().empty()); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - - GpgFileOpera::GetInstance().VerifyFile( - input_file, output_file, - [&callback_called_flag](GpgError err, - const DataObjectPtr& data_obj) { - auto d_result = ExtractParams<GpgVerifyResult>(data_obj, 0); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - ASSERT_FALSE(d_result.GetSignature().empty()); - ASSERT_EQ(d_result.GetSignature().at(0).GetFingerprint(), - "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); - - // stop waiting - callback_called_flag = true; - }); - }); - - int retry_count = 1000; - while (!callback_called_flag && retry_count-- > 0) { - std::this_thread::sleep_for(std::chrono::milliseconds(10)); - } - - ASSERT_TRUE(callback_called_flag); + auto [err, data_object] = GpgFileOpera::GetInstance().SignFileSync( + {sign_key}, input_file, false, output_file); + + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object->Check<GpgSignResult>())); + auto result = ExtractParams<GpgSignResult>(data_object, 0); + ASSERT_TRUE(result.InvalidSigners().empty()); + + auto [err_0, data_object_0] = + GpgFileOpera::GetInstance().VerifyFileSync(input_file, output_file); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object_0->Check<GpgVerifyResult>())); + auto verify_result = ExtractParams<GpgVerifyResult>(data_object_0, 0); + ASSERT_FALSE(verify_result.GetSignature().empty()); + ASSERT_EQ(verify_result.GetSignature().at(0).GetFingerprint(), + "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); } TEST_F(GpgCoreTest, CoreFileEncryptSignDecrVerifyTest) { - std::atomic_bool callback_called_flag{false}; + auto encrypt_key = GpgKeyGetter::GetInstance().GetPubkey( + "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); + auto sign_key = GpgKeyGetter::GetInstance().GetKey( + "8933EB283A18995F45D61DAC021D89771B680FFB"); + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); + auto output_file = GetTempFilePath(); + ASSERT_TRUE(sign_key.IsPrivateKey()); + ASSERT_TRUE(sign_key.IsHasActualSigningCapability()); + + auto [err, data_object] = GpgFileOpera::GetInstance().EncryptSignFileSync( + {encrypt_key}, {sign_key}, input_file, true, output_file); + + ASSERT_TRUE((data_object->Check<GpgEncryptResult, GpgSignResult>())); + auto encr_result = ExtractParams<GpgEncryptResult>(data_object, 0); + auto sign_result = ExtractParams<GpgSignResult>(data_object, 1); + ASSERT_TRUE(encr_result.InvalidRecipients().empty()); + ASSERT_TRUE(sign_result.InvalidSigners().empty()); + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = + GpgFileOpera::GetInstance().DecryptVerifyFileSync(output_file, + decrpypt_output_file); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object_0->Check<GpgDecryptResult, GpgVerifyResult>())); + auto decrypt_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + auto verify_reult = ExtractParams<GpgVerifyResult>(data_object_0, 1); + + ASSERT_FALSE(decrypt_result.Recipients().empty()); + ASSERT_EQ(decrypt_result.Recipients()[0].keyid, "F89C95A05088CC93"); + ASSERT_FALSE(verify_reult.GetSignature().empty()); + ASSERT_EQ(verify_reult.GetSignature().at(0).GetFingerprint(), + "8933EB283A18995F45D61DAC021D89771B680FFB"); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); +} + +TEST_F(GpgCoreTest, CoreFileEncryptSignDecrVerifyBinaryTest) { auto encrypt_key = GpgKeyGetter::GetInstance().GetPubkey( "467F14220CE8DCF780CF4BAD8465C55B25C9B7D1"); auto sign_key = GpgKeyGetter::GetInstance().GetKey( "8933EB283A18995F45D61DAC021D89771B680FFB"); - auto input_file = CreateTempFileAndWriteData("Hello GpgFrontend!"); + auto buffer = GFBuffer(QString("Hello GpgFrontend!")); + auto input_file = CreateTempFileAndWriteData(buffer); auto output_file = GetTempFilePath(); ASSERT_TRUE(sign_key.IsPrivateKey()); ASSERT_TRUE(sign_key.IsHasActualSigningCapability()); - GpgFileOpera::GetInstance().EncryptSignFile( - {encrypt_key}, {sign_key}, input_file, true, output_file, - [&callback_called_flag, output_file](GpgError err, - const DataObjectPtr& data_obj) { - ASSERT_TRUE((data_obj->Check<GpgEncryptResult, GpgSignResult>())); - auto encr_result = ExtractParams<GpgEncryptResult>(data_obj, 0); - auto sign_result = ExtractParams<GpgSignResult>(data_obj, 1); - ASSERT_TRUE(encr_result.InvalidRecipients().empty()); - ASSERT_TRUE(sign_result.InvalidSigners().empty()); - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - - auto decrpypt_output_file = GetTempFilePath(); - GpgFileOpera::GetInstance().DecryptVerifyFile( - output_file, decrpypt_output_file, - [&callback_called_flag, decrpypt_output_file]( - GpgError err, const DataObjectPtr& data_obj) { - ASSERT_TRUE( - (data_obj->Check<GpgDecryptResult, GpgVerifyResult>())); - auto decrypt_result = - ExtractParams<GpgDecryptResult>(data_obj, 0); - auto verify_reult = ExtractParams<GpgVerifyResult>(data_obj, 1); - - ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); - - ASSERT_FALSE(decrypt_result.Recipients().empty()); - ASSERT_EQ(decrypt_result.Recipients()[0].keyid, - "F89C95A05088CC93"); - ASSERT_FALSE(verify_reult.GetSignature().empty()); - ASSERT_EQ(verify_reult.GetSignature().at(0).GetFingerprint(), - "8933EB283A18995F45D61DAC021D89771B680FFB"); - - const auto [read_success, buffer] = - ReadFileGFBuffer(decrpypt_output_file); - ASSERT_TRUE(read_success); - ASSERT_EQ(buffer, GFBuffer(QString("Hello GpgFrontend!"))); - - // stop waiting - callback_called_flag = true; - }); - }); - - int retry_count = 1000; - while (!callback_called_flag && retry_count-- > 0) { - std::this_thread::sleep_for(std::chrono::milliseconds(10)); - } - - ASSERT_TRUE(callback_called_flag); + auto [err, data_object] = GpgFileOpera::GetInstance().EncryptSignFileSync( + {encrypt_key}, {sign_key}, input_file, false, output_file); + + ASSERT_TRUE((data_object->Check<GpgEncryptResult, GpgSignResult>())); + auto encr_result = ExtractParams<GpgEncryptResult>(data_object, 0); + auto sign_result = ExtractParams<GpgSignResult>(data_object, 1); + ASSERT_TRUE(encr_result.InvalidRecipients().empty()); + ASSERT_TRUE(sign_result.InvalidSigners().empty()); + ASSERT_EQ(CheckGpgError(err), GPG_ERR_NO_ERROR); + + auto decrpypt_output_file = GetTempFilePath(); + auto [err_0, data_object_0] = + GpgFileOpera::GetInstance().DecryptVerifyFileSync(output_file, + decrpypt_output_file); + + ASSERT_EQ(CheckGpgError(err_0), GPG_ERR_NO_ERROR); + ASSERT_TRUE((data_object_0->Check<GpgDecryptResult, GpgVerifyResult>())); + auto decrypt_result = ExtractParams<GpgDecryptResult>(data_object_0, 0); + auto verify_reult = ExtractParams<GpgVerifyResult>(data_object_0, 1); + + ASSERT_FALSE(decrypt_result.Recipients().empty()); + ASSERT_EQ(decrypt_result.Recipients()[0].keyid, "F89C95A05088CC93"); + ASSERT_FALSE(verify_reult.GetSignature().empty()); + ASSERT_EQ(verify_reult.GetSignature().at(0).GetFingerprint(), + "8933EB283A18995F45D61DAC021D89771B680FFB"); + + const auto [read_success, out_buffer] = + ReadFileGFBuffer(decrpypt_output_file); + ASSERT_TRUE(read_success); + ASSERT_EQ(buffer, out_buffer); } } // namespace GpgFrontend::Test |