From 3053042752f72cde2fb72c6f018191ea7b9ae6dc Mon Sep 17 00:00:00 2001 From: Benjamin Gaidioz Date: Tue, 10 Sep 2024 12:05:51 +0200 Subject: [PATCH] Added test --- .../builtin/credentials/LocationPackageTest.scala | 14 +++++++++++++- 1 file changed, 13 insertions(+), 1 deletion(-) diff --git a/snapi-compiler/src/test/scala/com/rawlabs/snapi/compiler/tests/builtin/credentials/LocationPackageTest.scala b/snapi-compiler/src/test/scala/com/rawlabs/snapi/compiler/tests/builtin/credentials/LocationPackageTest.scala index fd926d1f2..7c7512387 100644 --- a/snapi-compiler/src/test/scala/com/rawlabs/snapi/compiler/tests/builtin/credentials/LocationPackageTest.scala +++ b/snapi-compiler/src/test/scala/com/rawlabs/snapi/compiler/tests/builtin/credentials/LocationPackageTest.scala @@ -39,7 +39,7 @@ class LocationPackageTest extends SnapiTestContext { | Collection.Count(data) |""".stripMargin)(it => it should evaluateTo("7")) - test(s"""Location.Ll("s3://$UnitTestPublicBucket/publications/").url""") { it => + test(s"""List.Transform(Location.Ls("s3://$UnitTestPublicBucket/publications/"), x -> String.From(x))""") { it => it should evaluateTo("""Collection.Build( | "s3://rawlabs-public-test-data/publications/authors.parquet", | "s3://rawlabs-public-test-data/publications/authors.hjson", @@ -99,4 +99,16 @@ class LocationPackageTest extends SnapiTestContext { test(s"""String.Read(S3.Build("$UnitTestPrivateBucket2", "/file1.csv")) |""".stripMargin)(it => it should evaluateTo(""" "foobar" """)) + test(s"""let dir = S3.Build( + | "$UnitTestPrivateBucket", "/publications/publications-hjson/*.json", + | region = "${UnitTestPrivateBucketCred.getRegion}", + | accessKey = "${UnitTestPrivateBucketCred.getAccessSecretKey.getAccessKey}", + | secretKey = "${UnitTestPrivateBucketCred.getAccessSecretKey.getSecretKey}" + | ), + | files = Location.Ls(dir), + | lines = List.Unnest(files, f -> List.From(String.ReadLines(f))) + |in + | List.Count(lines) + |""".stripMargin)(it => it should evaluateTo("1000")) + }