fix the assertion error

Author: imabhichow

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/resource/migration_step_0.py

@@ -83,6 +83,7 @@ def migration_step_0_with_resource(ddb_table_name: str, sort_read_value: int = 0
             or item["partition_key"] == "PlaintextMigrationExample-2"
         )
         assert item["sort_key"] == sort_read_value
-        assert item["attribute1"] == "encrypt and sign me!"
-        assert item["attribute2"] == "sign me!"
-        assert item[":attribute3"] == "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)