fix the assertion error

Author: imabhichow

.github/workflows/ci_examples_python.yml

@@ -97,4 +97,4 @@ jobs:
           # Run legacy migration examples
           tox -e legacymigration
           # Run plaintext migration examples
-          tox -e plaintextmigration
+          tox -e plaintextmigration

Examples/runtimes/python/Migration/plaintext_to_awsdbe/src/plaintext/client/migration_step_0.py

@@ -66,6 +66,6 @@ def migration_step_0_with_client(ddb_table_name: str, sort_read_value: int = 0):
     # Demonstrate we get the expected item back
     assert plaintext_item["partition_key"]["S"] == "PlaintextMigrationExample"
     assert plaintext_item["sort_key"]["N"] == str(sort_read_value)
-    assert plaintext_item["attribute1"]["S"] == "encrypt and sign me!"
-    assert plaintext_item["attribute2"]["S"] == "sign me!"
-    assert plaintext_item[":attribute3"]["S"] == "ignore me!"
+    assert "S" in plaintext_item["attribute1"] and plaintext_item["attribute1"]["S"] == "encrypt and sign me!"
+    assert "S" in plaintext_item["attribute2"] and plaintext_item["attribute2"]["S"] == "sign me!"
+    assert "S" in plaintext_item[":attribute3"] and plaintext_item[":attribute3"]["S"] == "ignore me!"

Examples/runtimes/python/Migration/plaintext_to_awsdbe/src/plaintext/paginator/migration_step_0.py

@@ -68,6 +68,6 @@ def migration_step_0_with_paginator(ddb_table_name: str, sort_read_value: int =
     item = next((i for i in items if i["sort_key"]["N"] == str(sort_read_value)), None)
     assert item is not None
     assert item["partition_key"]["S"] == "PlaintextMigrationExample"
-    assert item["attribute1"]["S"] == "encrypt and sign me!"
-    assert item["attribute2"]["S"] == "sign me!"
-    assert item[":attribute3"]["S"] == "ignore me!"
+    assert "S" in item["attribute1"] and item["attribute1"]["S"] == "encrypt and sign me!"
+    assert "S" in item["attribute2"] and item["attribute2"]["S"] == "sign me!"
+    assert "S" in item[":attribute3"] and item[":attribute3"]["S"] == "ignore me!"

Examples/runtimes/python/Migration/plaintext_to_awsdbe/test/plaintext/client/test_migration_step_0.py

@@ -37,17 +37,17 @@ def test_migration_step_0_with_client():
         kms_key_id=TEST_KMS_KEY_ID, ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=2
     )
     # When: Execute Step 0 with sort_read_value=2
-    # Then: throws KeyError (i.e. cannot read encrypted values)
-    # KeyError: attribute1 now has type "B" (binary) instead of "S" (string)
-    with pytest.raises(KeyError):
+    # Then: throws AssertionError (i.e. cannot read encrypted values)
+    # AssertionError is raised when attributes have type "B" (binary) instead of "S" (string)
+    with pytest.raises(AssertionError):
         migration_step_0.migration_step_0_with_client(ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=2)
 
     # Given: Step 3 has succeeded
     migration_step_3.migration_step_3_with_client(
         kms_key_id=TEST_KMS_KEY_ID, ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=3
     )
     # When: Execute Step 0 with sort_read_value=3
-    # Then: throws KeyError (i.e. cannot read encrypted values)
-    # KeyError: attribute1 now has type "B" (binary) instead of "S" (string)
-    with pytest.raises(KeyError):
+    # Then: throws AssertionError (i.e. cannot read encrypted values)
+    # AssertionError is raised when attributes have type "B" (binary) instead of "S" (string)
+    with pytest.raises(AssertionError):
         migration_step_0.migration_step_0_with_client(ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=3)

Examples/runtimes/python/Migration/plaintext_to_awsdbe/test/plaintext/paginator/test_migration_step_0.py

@@ -37,17 +37,17 @@ def test_migration_step_0_with_paginator():
         kms_key_id=TEST_KMS_KEY_ID, ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=2
     )
     # When: Execute Step 0 with sort_read_value=2
-    # Then: throws KeyError (i.e. cannot read encrypted values)
-    # KeyError: attribute1 now has type "B" (binary) instead of "S" (string)
-    with pytest.raises(KeyError):
+    # Then: throws AssertionError (i.e. cannot read encrypted values)
+    # AssertionError is raised when attributes have type "B" (binary) instead of "S" (string)
+    with pytest.raises(AssertionError):
         migration_step_0.migration_step_0_with_paginator(ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=2)
 
     # Given: Step 3 has succeeded
     migration_step_3.migration_step_3_with_paginator(
         kms_key_id=TEST_KMS_KEY_ID, ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=3
     )
     # When: Execute Step 0 with sort_read_value=3
-    # Then: throws KeyError (i.e. cannot read encrypted values)
-    # KeyError: attribute1 now has type "B" (binary) instead of "S" (string)
-    with pytest.raises(KeyError):
+    # Then: throws AssertionError (i.e. cannot read encrypted values)
+    # AssertionError is raised when attributes have type "B" (binary) instead of "S" (string)
+    with pytest.raises(AssertionError):
         migration_step_0.migration_step_0_with_paginator(ddb_table_name=TEST_DDB_TABLE_NAME, sort_read_value=3)