Skip to content

Commit

Permalink
apacheGH-38346: [C++][Parquet] Use new encrypted files for page index…
Browse files Browse the repository at this point in the history
… encryption test (apache#38347)

### Rationale for this change

Encrypted parquet files have been updated to apache/parquet-testing repo. We can use the new files to verify encrypted page index.

### What changes are included in this PR?

Minor refactor read_configurations_test.cc to check page index from all parquet files.

### Are these changes tested?

This is indeed a unit test enhancement.

### Are there any user-facing changes?

NO.
* Closes: apache#38346

Authored-by: Gang Wu <ustcwg@gmail.com>
Signed-off-by: mwish <maplewish117@gmail.com>
  • Loading branch information
wgtmac authored Oct 25, 2023
1 parent 73589dd commit e836061
Show file tree
Hide file tree
Showing 2 changed files with 9 additions and 17 deletions.
24 changes: 8 additions & 16 deletions cpp/src/parquet/encryption/read_configurations_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -203,17 +203,15 @@ class TestDecryptionConfiguration

// Check that the decryption result is as expected.
void CheckResults(const std::string& file_name, unsigned decryption_config_num,
unsigned encryption_config_num, bool file_has_page_index) {
unsigned encryption_config_num) {
// Encryption_configuration number five contains aad_prefix and
// disable_aad_prefix_storage.
// An exception is expected to be thrown if the file is not decrypted with aad_prefix.
if (encryption_config_num == 5) {
if (decryption_config_num == 1 || decryption_config_num == 3) {
EXPECT_THROW(DecryptFile(file_name, decryption_config_num - 1), ParquetException);
if (file_has_page_index) {
EXPECT_THROW(DecryptPageIndex(file_name, decryption_config_num - 1),
ParquetException);
}
EXPECT_THROW(DecryptPageIndex(file_name, decryption_config_num - 1),
ParquetException);
return;
}
}
Expand All @@ -222,10 +220,8 @@ class TestDecryptionConfiguration
if (decryption_config_num == 2) {
if (encryption_config_num != 5 && encryption_config_num != 4) {
EXPECT_THROW(DecryptFile(file_name, decryption_config_num - 1), ParquetException);
if (file_has_page_index) {
EXPECT_THROW(DecryptPageIndex(file_name, decryption_config_num - 1),
ParquetException);
}
EXPECT_THROW(DecryptPageIndex(file_name, decryption_config_num - 1),
ParquetException);
return;
}
}
Expand All @@ -235,9 +231,7 @@ class TestDecryptionConfiguration
return;
}
EXPECT_NO_THROW(DecryptFile(file_name, decryption_config_num - 1));
if (file_has_page_index) {
EXPECT_NO_THROW(DecryptPageIndex(file_name, decryption_config_num - 1));
}
EXPECT_NO_THROW(DecryptPageIndex(file_name, decryption_config_num - 1));
}

// Returns true if file exists. Otherwise returns false.
Expand Down Expand Up @@ -269,8 +263,7 @@ TEST_P(TestDecryptionConfiguration, TestDecryption) {
// parquet file.
for (unsigned index = 0; index < vector_of_decryption_configurations_.size(); ++index) {
unsigned decryption_config_num = index + 1;
CheckResults(file_name, decryption_config_num, encryption_config_num,
/*file_has_page_index=*/true);
CheckResults(file_name, decryption_config_num, encryption_config_num);
}
// Delete temporary test file.
ASSERT_EQ(std::remove(file_name.c_str()), 0);
Expand All @@ -288,8 +281,7 @@ TEST_P(TestDecryptionConfiguration, TestDecryption) {
// parquet file.
for (unsigned index = 0; index < vector_of_decryption_configurations_.size(); ++index) {
unsigned decryption_config_num = index + 1;
CheckResults(file_name, decryption_config_num, encryption_config_num,
/*file_has_page_index=*/false);
CheckResults(file_name, decryption_config_num, encryption_config_num);
}
}

Expand Down

0 comments on commit e836061

Please sign in to comment.