added the logic to some other flaky test cases according to the sprea… · aws/amazon-s3-encryption-client-java@26620bb