Fix integ tests

Author: natoverse

tests/integration/storage/test_blob_pipeline_storage.py

@@ -21,7 +21,6 @@ async def test_find():
             items = list(
                 storage.find(base_dir="input", file_pattern=re.compile(r".*\.txt$"))
             )
-            items = [item[0] for item in items]
             assert items == []
 
             await storage.set(
@@ -30,12 +29,10 @@ async def test_find():
             items = list(
                 storage.find(base_dir="input", file_pattern=re.compile(r".*\.txt$"))
             )
-            items = [item[0] for item in items]
             assert items == ["input/christmas.txt"]
 
             await storage.set("test.txt", "Hello, World!", encoding="utf-8")
             items = list(storage.find(file_pattern=re.compile(r".*\.txt$")))
-            items = [item[0] for item in items]
             assert items == ["input/christmas.txt", "test.txt"]
 
             output = await storage.get("test.txt")
@@ -91,20 +88,17 @@ async def test_child():
             storage = parent.child("input")
             await storage.set("christmas.txt", "Merry Christmas!", encoding="utf-8")
             items = list(storage.find(re.compile(r".*\.txt$")))
-            items = [item[0] for item in items]
             assert items == ["christmas.txt"]
 
             await storage.set("test.txt", "Hello, World!", encoding="utf-8")
             items = list(storage.find(re.compile(r".*\.txt$")))
-            items = [item[0] for item in items]
             print("FOUND", items)
             assert items == ["christmas.txt", "test.txt"]
 
             output = await storage.get("test.txt")
             assert output == "Hello, World!"
 
             items = list(parent.find(re.compile(r".*\.txt$")))
-            items = [item[0] for item in items]
             print("FOUND ITEMS", items)
             assert items == ["input/christmas.txt", "input/test.txt"]
         finally:

tests/integration/storage/test_cosmosdb_storage.py

@@ -30,7 +30,6 @@ async def test_find():
     try:
         try:
             items = list(storage.find(file_pattern=re.compile(r".*\.json$")))
-            items = [item[0] for item in items]
             assert items == []
 
             json_content = {
@@ -40,15 +39,13 @@ async def test_find():
                 "christmas.json", json.dumps(json_content), encoding="utf-8"
             )
             items = list(storage.find(file_pattern=re.compile(r".*\.json$")))
-            items = [item[0] for item in items]
             assert items == ["christmas.json"]
 
             json_content = {
                 "content": "Hello, World!",
             }
             await storage.set("test.json", json.dumps(json_content), encoding="utf-8")
             items = list(storage.find(file_pattern=re.compile(r".*\.json$")))
-            items = [item[0] for item in items]
             assert items == ["christmas.json", "test.json"]
 
             output = await storage.get("test.json")

tests/integration/storage/test_file_pipeline_storage.py

@@ -21,7 +21,7 @@ async def test_find():
             base_dir="tests/fixtures/text/input", file_pattern=re.compile(r".*\.txt$")
         )
     )
-    assert items == [(str(Path("tests/fixtures/text/input/dulce.txt")), {})]
+    assert items == [str(Path("tests/fixtures/text/input/dulce.txt"))]
     output = await storage.get("tests/fixtures/text/input/dulce.txt")
     assert len(output) > 0