test: Add queuing for Neptune bulk load tests (#2724)

LeonLuttenberger · web-flow · commit ef3a4191cdf8 · 2024-03-14T09:50:59.000-05:00
diff --git a/tests/unit/test_neptune.py b/tests/unit/test_neptune.py
@@ -236,6 +236,7 @@ def test_gremlin_bulk_load(
         iam_role=neptune_load_iam_role_arn,
         use_threads=use_threads,
         keep_files=keep_files,
+        queue_request="TRUE",
     )
     res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)")
 
@@ -295,6 +296,7 @@ def test_gremlin_bulk_load_from_files(
         client=client,
         path=path,
         iam_role=neptune_load_iam_role_arn,
+        queue_request="TRUE",
         parser_configuration=parser_config,
     )
     res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)")

Original file line number	Diff line number	Diff line change
`@@ -236,6 +236,7 @@ def test_gremlin_bulk_load(`
`236`	`236`	`iam_role=neptune_load_iam_role_arn,`
`237`	`237`	`use_threads=use_threads,`
`238`	`238`	`keep_files=keep_files,`
	`239`	`+ queue_request="TRUE",`
`239`	`240`	`)`
`240`	`241`	`res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)")`
`241`	`242`
`@@ -295,6 +296,7 @@ def test_gremlin_bulk_load_from_files(`
`295`	`296`	`client=client,`
`296`	`297`	`path=path,`
`297`	`298`	`iam_role=neptune_load_iam_role_arn,`
	`299`	`+ queue_request="TRUE",`
`298`	`300`	`parser_configuration=parser_config,`
`299`	`301`	`)`
`300`	`302`	`res_df = wr.neptune.execute_gremlin(client, f"g.V().hasLabel('{label}').valueMap().with(WithOptions.tokens)")`