auto commit

Author: rishav-karanjit

DynamoDbEncryption/runtimes/rust/examples/main.rs

@@ -14,6 +14,7 @@ pub mod keyring;
 pub mod multi_get_put_example;
 pub mod searchableencryption;
 pub mod test_utils;
+pub mod migration;
 
 use std::convert::From;
 

DynamoDbEncryption/runtimes/rust/examples/migration/plaintext_to_awsdbe/awsdbe/migration_step_1.rs

@@ -9,6 +9,12 @@ use crate::migration::plaintext_to_awsdbe::migration_utils::{
     verify_returned_item, ENCRYPTED_AND_SIGNED_VALUE, SIGN_ONLY_VALUE, DO_NOTHING_VALUE,
 };
 use crate::migration::plaintext_to_awsdbe::awsdbe::common::create_table_configs;
+// We import these packages for testing combination of migration steps  
+use crate::migration::plaintext_to_awsdbe::plaintext::migration_step_0::migration_step_0_example;
+use crate::migration::plaintext_to_awsdbe::awsdbe::migration_step_2::migration_step_2_example;
+use crate::migration::plaintext_to_awsdbe::awsdbe::migration_step_3::migration_step_3_example;
+use crate::test_utils;
+use uuid::Uuid;
 
 /*
 Migration Step 1: This is the first step in the migration process from
@@ -134,3 +140,46 @@ pub async fn migration_step_1_example(
         Err("No item found".into())
     }
 }
+
+#[tokio::test(flavor = "multi_thread")]
+async fn test_migration_step_1() -> Result<(), Box<dyn std::error::Error>> {
+    let kms_key_id = test_utils::TEST_KMS_KEY_ID;
+    let table_name = test_utils::TEST_DDB_TABLE_NAME;
+    let partition_key = Uuid::new_v4().to_string();
+    let sort_keys = ["0", "1", "2", "3"];
+
+    // Given: Step 0 has succeeded
+    let success = migration_step_0_example(table_name, &partition_key, sort_keys[0], sort_keys[0]).await?;
+    assert!(success, "MigrationStep0 should complete successfully");
+
+    // Successfully executes step 1
+    let success = migration_step_1_example(kms_key_id, table_name, &partition_key, sort_keys[1], sort_keys[1]).await?;
+    assert!(success, "MigrationStep1 should complete successfully");
+
+    // When: Execute Step 1 with sortReadValue=0, Then: Success (i.e. can read plaintext values from Step 0)
+    let success = migration_step_1_example(kms_key_id, table_name, &partition_key, sort_keys[1], sort_keys[0]).await?;
+    assert!(success, "MigrationStep1 should be able to read items written by Step 0");
+
+    // Given: Step 2 has succeeded
+    let success = migration_step_2_example(kms_key_id, table_name, &partition_key, sort_keys[2], sort_keys[2]).await?;
+    assert!(success, "MigrationStep2 should complete successfully");
+
+    // When: Execute Step 1 with sortReadValue=2, Then: Success (i.e. can read encrypted values from Step 2)
+    let success = migration_step_1_example(kms_key_id, table_name, &partition_key, sort_keys[1], sort_keys[2]).await?;
+    assert!(success, "MigrationStep1 should be able to read items written by Step 2");
+
+    // Given: Step 3 has succeeded
+    let success = migration_step_3_example(kms_key_id, table_name, &partition_key, sort_keys[3], sort_keys[3]).await?;
+    assert!(success, "MigrationStep3 should complete successfully");
+
+    // When: Execute Step 1 with sortReadValue=3, Then: Success (i.e. can read encrypted values from Step 3)
+    let success = migration_step_1_example(kms_key_id, table_name, &partition_key, sort_keys[1], sort_keys[3]).await?;
+    assert!(success, "MigrationStep1 should be able to read items written by Step 3");
+
+    // Cleanup
+    for sort_key in &sort_keys {
+        test_utils::cleanup_items(table_name, &partition_key, sort_key).await?;
+    }
+
+    Ok(())
+}

DynamoDbEncryption/runtimes/rust/examples/migration/plaintext_to_awsdbe/awsdbe/migration_step_2.rs

@@ -9,6 +9,12 @@ use crate::migration::plaintext_to_awsdbe::migration_utils::{
     verify_returned_item, ENCRYPTED_AND_SIGNED_VALUE, SIGN_ONLY_VALUE, DO_NOTHING_VALUE,
 };
 use crate::migration::plaintext_to_awsdbe::awsdbe::common::create_table_configs;
+// We import these packages for testing combination of migration steps  
+use crate::migration::plaintext_to_awsdbe::plaintext::migration_step_0::migration_step_0_example;
+use crate::migration::plaintext_to_awsdbe::awsdbe::migration_step_1::migration_step_1_example;
+use crate::migration::plaintext_to_awsdbe::awsdbe::migration_step_3::migration_step_3_example;
+use crate::test_utils;
+use uuid::Uuid;
 
 /*
 Migration Step 2: This is the second step in the migration process from
@@ -136,3 +142,46 @@ pub async fn migration_step_2_example(
         Err("No item found".into())
     }
 }
+
+#[tokio::test(flavor = "multi_thread")]
+async fn test_migration_step_2() -> Result<(), Box<dyn std::error::Error>> {
+    let kms_key_id = test_utils::TEST_KMS_KEY_ID;
+    let table_name = test_utils::TEST_DDB_TABLE_NAME;
+    let partition_key = Uuid::new_v4().to_string();
+    let sort_keys = ["0", "1", "2", "3"];
+
+    // Given: Step 0 has succeeded
+    let success = migration_step_0_example(table_name, &partition_key, sort_keys[0], sort_keys[0]).await?;
+    assert!(success, "MigrationStep0 should complete successfully");
+
+    // Given: Step 1 has succeeded
+    let success = migration_step_1_example(kms_key_id, table_name, &partition_key, sort_keys[1], sort_keys[1]).await?;
+    assert!(success, "MigrationStep1 should complete successfully");
+
+    // Successfully executes step 2
+    let success = migration_step_2_example(kms_key_id, table_name, &partition_key, sort_keys[2], sort_keys[2]).await?;
+    assert!(success, "MigrationStep2 should complete successfully");
+
+    // When: Execute Step 2 with sortReadValue=0, Then: Success (i.e. can read plaintext values from Step 0)
+    let success = migration_step_2_example(kms_key_id, table_name, &partition_key, sort_keys[2], sort_keys[0]).await?;
+    assert!(success, "MigrationStep2 should be able to read items written by Step 0");
+
+    // When: Execute Step 2 with sortReadValue=1, Then: Success (i.e. can read plaintext values from Step 1)
+    let success = migration_step_2_example(kms_key_id, table_name, &partition_key, sort_keys[2], sort_keys[1]).await?;
+    assert!(success, "MigrationStep2 should be able to read items written by Step 1");
+
+    // Given: Step 3 has succeeded
+    let success = migration_step_3_example(kms_key_id, table_name, &partition_key, sort_keys[3], sort_keys[3]).await?;
+    assert!(success, "MigrationStep3 should complete successfully");
+
+    // When: Execute Step 2 with sortReadValue=3, Then: Success (i.e. can read encrypted values from Step 3)
+    let success = migration_step_2_example(kms_key_id, table_name, &partition_key, sort_keys[2], sort_keys[3]).await?;
+    assert!(success, "MigrationStep2 should be able to read items written by Step 3");
+
+    // Cleanup
+    for sort_key in &sort_keys {
+        test_utils::cleanup_items(table_name, &partition_key, sort_key).await?;
+    }
+
+    Ok(())
+}

DynamoDbEncryption/runtimes/rust/examples/migration/plaintext_to_awsdbe/migration_tests.rs

@@ -1,9 +1,6 @@
 // Copyright Amazon.com Inc. or its affiliates. All Rights Reserved.
 // SPDX-License-Identifier: Apache-2.0
 
+pub mod awsdbe;
 pub mod migration_utils;
 pub mod plaintext;
-pub mod awsdbe;
-
-#[cfg(test)]
-pub mod migration_tests;