retry readfrommongodb

Author: derrickaw

sdks/python/apache_beam/yaml/tests/mongodb.yaml

@@ -40,36 +40,35 @@ pipelines:
           input: WriteData.my_error_output
           config:
             elements: []
-# TODO: Re-enable ReadFromMongoDB tests once Java MongoDbIO is migrated from legacy BoundedSource.
-#  - pipeline:
-#      type: composite
-#      transforms:
-#        - type: ReadFromMongoDB
-#          name: ReadData
-#          config:
-#            connection_uri: '{mongo_vars[URI]}'
-#            database: '{mongo_vars[DATABASE]}'
-#            collection: '{mongo_vars[COLLECTION]}'
-#            schema: |
-#              {
-#                "type": "object",
-#                "properties": {
-#                  "name": {"type": "string"},
-#                  "age": {"type": "integer"}
-#                },
-#                "required": ["name", "age"]
-#              }
-#            error_handling:
-#              output: my_error_output
-#        - type: AssertEqual
-#          input: ReadData
-#          config:
-#            elements:
-#              - { name: "John", age: 30 }
-#              - { name: "Jane", age: 25 }
-#        - type: AssertEqual
-#          input: ReadData.my_error_output
-#          config:
-#            elements: []
+  - pipeline:
+     type: composite
+     transforms:
+       - type: ReadFromMongoDB
+         name: ReadData
+         config:
+           connection_uri: '{mongo_vars[URI]}'
+           database: '{mongo_vars[DATABASE]}'
+           collection: '{mongo_vars[COLLECTION]}'
+           schema: |
+             {
+               "type": "object",
+               "properties": {
+                 "name": {"type": "string"},
+                 "age": {"type": "integer"}
+               },
+               "required": ["name", "age"]
+             }
+           error_handling:
+             output: my_error_output
+       - type: AssertEqual
+         input: ReadData
+         config:
+           elements:
+             - { name: "John", age: 30 }
+             - { name: "Jane", age: 25 }
+       - type: AssertEqual
+         input: ReadData.my_error_output
+         config:
+           elements: []