Skip to content

Commit e461496

Browse files
wheatear-devwheatear-dev
committed
Add unit tests for CryptoKey
1 parent 7d2ca2d commit e461496

File tree

6 files changed

+139
-0
lines changed

6 files changed

+139
-0
lines changed

‎.gitignore‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ logs/
4343

4444
# Generated by unit tests
4545
tests/data/*.translation
46+
tests/data/crypto/out*
4647

4748
############################
4849
### General build output ###

‎modules/mbedtls/SCsub‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -121,6 +121,9 @@ if env["tests"]:
121121
env_mbed_tls.Append(CPPDEFINES=["TESTS_ENABLED"])
122122
env_mbed_tls.add_source_files(module_obj, "./tests/*.cpp")
123123

124+
if env["disable_exceptions"]:
125+
env_mbed_tls.Append(CPPDEFINES=["DOCTEST_CONFIG_NO_EXCEPTIONS_BUT_WITH_ALL_ASSERTS"])
126+
124127
env.modules_sources += module_obj
125128

126129
# Needed to force rebuilding the module files when the thirdparty library is updated.

‎modules/mbedtls/tests/test_crypto_mbedtls.cpp‎

Lines changed: 39 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -33,6 +33,7 @@
3333
#include "../crypto_mbedtls.h"
3434

3535
#include "tests/test_macros.h"
36+
#include "tests/test_utils.h"
3637

3738
namespace TestCryptoMbedTLS {
3839

@@ -60,4 +61,42 @@ void hmac_context_digest_test(HashingContext::HashType ht, String expected_hex)
6061
String hex = String::hex_encode_buffer(digest.ptr(), digest.size());
6162
CHECK(hex == expected_hex);
6263
}
64+
65+
Ref<CryptoKey> create_crypto_key(const String &p_key_path, bool p_public_only) {
66+
Ref<CryptoKey> crypto_key = Ref<CryptoKey>(CryptoKey::create());
67+
crypto_key->load(p_key_path, p_public_only);
68+
return crypto_key;
69+
}
70+
71+
String read_file_s(const String &p_file_path) {
72+
Ref<FileAccess> file_access = FileAccess::open(p_file_path, FileAccess::READ);
73+
REQUIRE(file_access.is_valid());
74+
return file_access->get_as_utf8_string();
75+
}
76+
77+
bool files_equal(const String &p_in_path, const String &p_out_path) {
78+
const String s_in = read_file_s(p_in_path);
79+
const String s_out = read_file_s(p_out_path);
80+
return s_in == s_out;
81+
}
82+
83+
void crypto_key_public_only_test(const String &p_key_path, bool p_public_only) {
84+
Ref<CryptoKey> crypto_key = create_crypto_key(p_key_path, p_public_only);
85+
bool is_equal = crypto_key->is_public_only() == p_public_only;
86+
CHECK(is_equal);
87+
}
88+
89+
void crypto_key_save_test(const String &p_in_path, const String &p_out_path, bool p_public_only) {
90+
Ref<CryptoKey> crypto_key = create_crypto_key(p_in_path, p_public_only);
91+
crypto_key->save(p_out_path, p_public_only);
92+
bool is_equal = files_equal(p_in_path, p_out_path);
93+
CHECK(is_equal);
94+
}
95+
96+
void crypto_key_save_public_only_test(const String &p_in_priv_path, const String &p_in_pub_path, const String &p_out_path) {
97+
Ref<CryptoKey> crypto_key = create_crypto_key(p_in_priv_path, false);
98+
crypto_key->save(p_out_path, true);
99+
bool is_equal = files_equal(p_in_pub_path, p_out_path);
100+
CHECK(is_equal);
101+
}
63102
} // namespace TestCryptoMbedTLS

‎modules/mbedtls/tests/test_crypto_mbedtls.h‎

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,11 @@
3131
#ifndef TEST_CRYPTO_MBEDTLS_H
3232
#define TEST_CRYPTO_MBEDTLS_H
3333

34+
#include "core/crypto/crypto.h"
3435
#include "core/crypto/hashing_context.h"
3536

3637
#include "tests/test_macros.h"
38+
#include "tests/test_utils.h"
3739

3840
namespace TestCryptoMbedTLS {
3941

@@ -56,6 +58,35 @@ TEST_CASE("[HMACContext] HMAC digest") {
5658
// SHA-1
5759
hmac_context_digest_test(HashingContext::HashType::HASH_SHA1, "a0ac4cd68a2f4812c355983d94e8d025afe7dddf");
5860
}
61+
62+
void crypto_key_public_only_test(const String &p_key_path, bool public_only);
63+
64+
TEST_CASE("[Crypto] CryptoKey is_public_only") {
65+
crypto_key_public_only_test(TestUtils::get_data_path("crypto/in.key"), false);
66+
crypto_key_public_only_test(TestUtils::get_data_path("crypto/in.pub"), true);
67+
}
68+
69+
void crypto_key_save_test(const String &p_in_path, const String &p_out_path, bool public_only);
70+
71+
TEST_CASE("[Crypto] CryptoKey save") {
72+
const String in_priv_path = TestUtils::get_data_path("crypto/in.key");
73+
const String out_priv_path = TestUtils::get_data_path("crypto/out.key");
74+
crypto_key_save_test(in_priv_path, out_priv_path, false);
75+
76+
const String in_pub_path = TestUtils::get_data_path("crypto/in.pub");
77+
const String out_pub_path = TestUtils::get_data_path("crypto/out.pub");
78+
crypto_key_save_test(in_pub_path, out_pub_path, true);
79+
}
80+
81+
void crypto_key_save_public_only_test(const String &p_in_priv_path, const String &p_in_pub_path, const String &p_out_path);
82+
83+
TEST_CASE("[Crypto] CryptoKey save public_only") {
84+
const String in_priv_path = TestUtils::get_data_path("crypto/in.key");
85+
const String in_pub_path = TestUtils::get_data_path("crypto/in.pub");
86+
const String out_path = TestUtils::get_data_path("crypto/out_public_only.pub");
87+
crypto_key_save_public_only_test(in_priv_path, in_pub_path, out_path);
88+
}
89+
5990
} // namespace TestCryptoMbedTLS
6091

6192
#endif // TEST_CRYPTO_MBEDTLS_H

‎tests/data/crypto/in.key‎

Lines changed: 51 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,51 @@
1+
-----BEGIN RSA PRIVATE KEY-----
2+
MIIJKQIBAAKCAgEAp+sWMepqN+dXhmZkS45W0FTLX2f0zRb+kUhXngfiEprgA2Uu
3+
ZP5gGv7/EjOBDniYalGZqYPJum08Px++17P6Hyopr05aPhZ6Ocnt+LAU/WRAFspN
4+
3xXTZglqnmGMsdNcqBl3loVNmC1H/zX53oBeP/L/i4lva3DjCRIpHtGjfy3ufSn7
5+
fTrlBqKIEHYgQZyzos9dxbk1NJcWVxlNmvFRtgPW6DQ/P3g0ahLJeq/hO3ykUFfi
6+
KdhHkS+s9dC9qjZwSxCCEF1o6zkEO6ynxk0sxjOpPTJFw7fQ+2KeJWJQfdZeJ6u+
7+
DpbwK4g9tMUMNxf3QMk1KhAnwwUrKXfZcviS3zweD8Tl5zEj45H1ghoUQfuW7Y3w
8+
gwHMu8lF8iGA87cf/fhqr0V9piCcwWkfVP/athpMoUfyj9Sa3ag0dvDSo9PAet0u
9+
rXmdKTyhMg4lQL6f9BmMuuB/KwWzCuG/5VY9ONxno3OVX6juHTpng5UglbkiDsD3
10+
tivl1gCbvIryoGdt+xI0JmAC5eXfg79Nio/BayDR9Npm1m460p3GeRaawyYysBo/
11+
L5/YZ/S3bYBRoJ7lq6GkTA+22lWAb04IgtS8wxO4Ma8EOtKD+AoR3C+WLivcp9LN
12+
TxbQOMKGL+8imQGBEz3XTR4lrE02QDQy0DIBKy7p7dhlyBdwhTmBX3P2mx0CAwEA
13+
AQKCAgBv7edUjIITE5UnFHeEWbQKmIsb5GqsjshPxV4KDA0pA62Q9dAQJ/Od6x3R
14+
Xx2GrOJD9HKuKRe9ufSvyxRmKiTuwycYIO1Md6UvgierXowPP9TsnBt+Ock5Ocul
15+
GTc0jcQ0lQ0++0p2xrA4MR2GsCCjFfI7a/gmMRBVSpK4ZVtLei1/pw1pM2nYm1yB
16+
RIxJ0A951ioWk1cg4BlXI5m0T2l9H2AQVktWnmSp1C4TJsvG4FWS7JHn/K/v2ky7
17+
alIS9MizcKSSDgHS0aW9tV/8chMHZwZHsYwJYyzddKYgG0G2L7+BSByfEwOysNUY
18+
+0QiMUpyF+zlRfGLMJXNxYLf/UvAhu2xbNi6+A1/xm4FerFF0arMMUzY1Lwwa02t
19+
yBmflhZ+s2ngT4grj3waShC14H6idL2j5HFcyBs/UOA+HkV1I5SoGihNziMP8gfX
20+
IDSb4WBzckPZD6kUAojNFqhx+W7XpWWE5QnWam76b8Mzdg9Xf9pKo6ULt6kwdC8Z
21+
ufbOTRXO08jkb1F64Fmb4F7EAvXLyhFtclY4CuPYSA68Sad8A41ipCsQ5bwvUTMd
22+
o2l7kYplk4f/Bvz2yOhZZVdWGanmKvnGUMehJ+B4zi8HFOIRd21bXkeBwwKjqNni
23+
3kqVairo3O2HWrAJwRvhCZam14HGkr+HQPEGLn48fstquizoQQKCAQEA3slWrItS
24+
wdwciouUbEDbftmWEL+eXAQyVxbIkVuvugLyeZ1lHIk7K5lZ8hecWLNJXPeqyi7P
25+
i2AI8O+7VEoJmePwGmZ19rJHNBB4thgZq114sbkHcQNXXhdFnen8HkdZRdjusViC
26+
BHXCtG8TzxCrpQl/6dLDUkG7D3DiYGaV1IHQ+BNYjJQfh6grkcqRrOq7JH2CnKV0
27+
VCxoSobKgB1zEdc7gKyeRp9SSs5rJe5qTmEMXptrQUeXeElsHfmOV8RUO9b1BBrN
28+
bZhMR3zbE/8Oq3umro7WBaSg7XZMSwCeI0FR8uUAy/FZTrWRPeb86ywmP+pQqcdp
29+
R0OQL0vSWWaLcQKCAQEAwPOzV9p0dQtERy3Z0hh6A3pj+fV5+l5QsA+R4vDoOH/l
30+
GCoEJwSh21qOspSy66fulJR3Jml385/x8I3MC87u8h33aKsdmeVYOdmuiB+lj3sc
31+
DRY/J9w9WbL3mdF3H1rU0ldxfKr1HdfK4xcSdJKBE1FYfO9tB6NFvgdSnvMbmSa1
32+
LVtc8N5hSB1h+d5LWHzh4TC4SG89TivQi0oEacErVJT9OEdGwAtgEU3K8UGKOcTr
33+
OKos0vts281DuKpkfLBstH8l+VOdBD4E+I+MB1Y50oJ/D3h4bl+WDcR0DqlWzay3
34+
3WCSjzZC5T9lEyQ/0TKv7GPgAiH5/41nQnSM6ar8bQKCAQEAvSf9q2pvzaFxqkBw
35+
uKkotD9SJs5LSp1VkJQLnz9VqH2wGooEu4HY91+w+tgJK1auR30RSbENDq1vagJh
36+
72MdW8goqIGuTtN3mUES/KjhwpoOS/dp1g6cM4tW1IlCQwMZTTCvGWyol9jUhBZ7
37+
nyfsVKgILyOAK2sbxDR4QJlZRaEjKD5kxJdPXgLvW02++i4izwyxxQbGCmHZ+s0P
38+
Sk+2z8MLBmmJyTSkzlcMqpwPLpU/x2P2YOrENKFCZwDoVqSfUF9mkSGgohjZSyk7
39+
aXL5pafLEhK8rPXmnTf/9v6DRjPDvJOrZX158lY/B2wD+jj2EPaFnmFthdBbr4yV
40+
AMsMQQKCAQEAn8nxroKRyNAAxjV5Wly8xp6XpsucLTPn/DWYqei5VvjLPxykfa9/
41+
Xsl6vPcZyMA0esUMezoChTXixUSYQvsmtEkOt5ZlmCnuy1GzELWshMr96vSObrMb
42+
92mXVMG7tbKh5mNV71kgTouDUFauCO2+iMHn1ubsUtPqkLk9ubY4F7ePeLVdnXd7
43+
9p2moqdtnCUnZjbTleDRUyhDtuYgC3hWKuCLZwzX0XhaIVpcAzk0gCzMYwvCvSJL
44+
/ybYu1gYiY4NJ9jYGMcelAHMWg9+diD5F5TMJoKssTLlcBdNyUqBQSiUx3cPSBw2
45+
f+TlDloJo3QnbkszmnCKuRBgAA/HFkdsbQKCAQBokUYEzTy3iOwcv9xe2DZoeIeG
46+
Y0B/ri8FxQ+Wt5t/LtGKIwKL5BEpgjXVLL1l/is4jncLkUdGrdqglbjgkJ/fUB/5
47+
/354BjX9a1EBw95XTrUFcdX8uglYkPZkIR9GWY7m87NvLZUdrsmrfl6HA4I4kpAt
48+
1N0dcn/8GIm9vm7COPGDjjPzv+xlMuMjdeTuBxe8tddOwwtXjzkTZmcOdZpianxF
49+
R3zY1LCHk9vPulkAs2o+qCTBGT0qHJp04AamY3KWPW4Cf5GzPousOB4p7Hu8UFyv
50+
FISkNe48bzSYveJ+yZ3myG2fBCGFQmSRJVcauIokPAl6lKF/Vw5DdWp1LYWa
51+
-----END RSA PRIVATE KEY-----

‎tests/data/crypto/in.pub‎

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
-----BEGIN PUBLIC KEY-----
2+
MIICIjANBgkqhkiG9w0BAQEFAAOCAg8AMIICCgKCAgEAp+sWMepqN+dXhmZkS45W
3+
0FTLX2f0zRb+kUhXngfiEprgA2UuZP5gGv7/EjOBDniYalGZqYPJum08Px++17P6
4+
Hyopr05aPhZ6Ocnt+LAU/WRAFspN3xXTZglqnmGMsdNcqBl3loVNmC1H/zX53oBe
5+
P/L/i4lva3DjCRIpHtGjfy3ufSn7fTrlBqKIEHYgQZyzos9dxbk1NJcWVxlNmvFR
6+
tgPW6DQ/P3g0ahLJeq/hO3ykUFfiKdhHkS+s9dC9qjZwSxCCEF1o6zkEO6ynxk0s
7+
xjOpPTJFw7fQ+2KeJWJQfdZeJ6u+DpbwK4g9tMUMNxf3QMk1KhAnwwUrKXfZcviS
8+
3zweD8Tl5zEj45H1ghoUQfuW7Y3wgwHMu8lF8iGA87cf/fhqr0V9piCcwWkfVP/a
9+
thpMoUfyj9Sa3ag0dvDSo9PAet0urXmdKTyhMg4lQL6f9BmMuuB/KwWzCuG/5VY9
10+
ONxno3OVX6juHTpng5UglbkiDsD3tivl1gCbvIryoGdt+xI0JmAC5eXfg79Nio/B
11+
ayDR9Npm1m460p3GeRaawyYysBo/L5/YZ/S3bYBRoJ7lq6GkTA+22lWAb04IgtS8
12+
wxO4Ma8EOtKD+AoR3C+WLivcp9LNTxbQOMKGL+8imQGBEz3XTR4lrE02QDQy0DIB
13+
Ky7p7dhlyBdwhTmBX3P2mx0CAwEAAQ==
14+
-----END PUBLIC KEY-----

0 commit comments

Comments
 (0)