303 lines
12 KiB
Julia
303 lines
12 KiB
Julia
using Test
|
|
using GeneralUtils
|
|
using GeneralUtils.garageS3
|
|
using AWS
|
|
using Random: randstring
|
|
using UUIDs: uuid4
|
|
|
|
@testset "garageS3.jl" begin
|
|
|
|
# --- SimpleCredentials ---
|
|
|
|
@testset "SimpleCredentials construction" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("AKIAIOSFODNN7EXAMPLE", "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY", "sessiontoken")
|
|
@test creds.access_key_id == "AKIAIOSFODNN7EXAMPLE"
|
|
@test creds.secret_key == "wJalrXUtnFEMI/K7MDENG/bPxRfiCYEXAMPLEKEY"
|
|
@test creds.token == "sessiontoken"
|
|
end
|
|
|
|
@testset "SimpleCredentials AWS extensions" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key1", "secret1", "token1")
|
|
@test AWS.credentials(creds) === creds
|
|
@test AWS.check_credentials(creds) === creds
|
|
@test AWS.refresh!(creds; force=false) === creds
|
|
@test AWS.refresh!(creds) === creds
|
|
end
|
|
|
|
# --- GarageConfig ---
|
|
|
|
@testset "GarageConfig construction" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "us-east-1", creds)
|
|
@test config.endpoint == "https://s3-api.example.com"
|
|
@test config.region == "us-east-1"
|
|
@test config.credentials === creds
|
|
@test config isa AWS.AbstractAWSConfig
|
|
end
|
|
|
|
@testset "GarageConfig AWS extensions" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
|
|
@test AWS.region(config) == "garage"
|
|
@test AWS.credentials(config) === creds
|
|
end
|
|
|
|
# --- AWS.generate_service_url ---
|
|
|
|
@testset "generate_service_url with resource starting with /" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "/mybucket/key")
|
|
@test url == "https://s3-api.example.com/mybucket/key"
|
|
end
|
|
|
|
@testset "generate_service_url with resource NOT starting with /" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "mybucket/key")
|
|
@test url == "https://s3-api.example.com/mybucket/key"
|
|
end
|
|
|
|
@testset "generate_service_url strips trailing slashes from endpoint" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com/", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "/bucket/key")
|
|
@test url == "https://s3-api.example.com/bucket/key"
|
|
end
|
|
|
|
@testset "generate_service_url with multiple trailing slashes" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("https://s3-api.example.com///", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "/bucket/key")
|
|
@test url == "https://s3-api.example.com/bucket/key"
|
|
end
|
|
|
|
@testset "generate_service_url preserves http endpoint" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("http://localhost:3900", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "/bucket/key")
|
|
@test url == "http://localhost:3900/bucket/key"
|
|
end
|
|
|
|
@testset "generate_service_url with http endpoint and trailing slash" begin
|
|
creds = GeneralUtils.garageS3.SimpleCredentials("key", "secret", "")
|
|
config = GeneralUtils.garageS3.GarageConfig("http://localhost:3900/", "garage", creds)
|
|
url = AWS.generate_service_url(config, "s3", "/bucket/key")
|
|
@test url == "http://localhost:3900/bucket/key"
|
|
end
|
|
|
|
# --- GarageStorage ---
|
|
|
|
@testset "GarageStorage default region" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key123", "secret123", "mybucket")
|
|
@test storage.bucket == "mybucket"
|
|
@test storage.config.region == "garage"
|
|
@test storage.config.endpoint == "https://s3-api.example.com"
|
|
end
|
|
|
|
@testset "GarageStorage custom region" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key123", "secret123", "mybucket"; region="custom-region")
|
|
@test storage.config.region == "custom-region"
|
|
end
|
|
|
|
@testset "GarageStorage endpoint without https" begin
|
|
storage = GarageStorage("http://localhost:3900", "key", "secret", "mybucket")
|
|
@test storage.config.endpoint == "http://localhost:3900"
|
|
end
|
|
|
|
@testset "GarageStorage credentials are SimpleCredentials" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "mykey", "mysecret", "mybucket")
|
|
@test storage.config.credentials isa GeneralUtils.garageS3.SimpleCredentials
|
|
@test storage.config.credentials.access_key_id == "mykey"
|
|
@test storage.config.credentials.secret_key == "mysecret"
|
|
@test storage.config.credentials.token == ""
|
|
end
|
|
|
|
@testset "GarageStorage returns correct struct type" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
@test storage isa GarageStorage
|
|
end
|
|
|
|
# --- put_file callable struct (slash detection) ---
|
|
|
|
@testset "put_file callable struct returns nothing for keys with slashes" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
uploader = put_file(storage)
|
|
result = @test_logs (:warn, r"Upload aborted") uploader("a/b.json", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "put_file callable struct type is correct" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
uploader = put_file(storage)
|
|
@test uploader isa put_file
|
|
end
|
|
|
|
@testset "put_file callable struct construction stores storage" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
uploader = put_file(storage)
|
|
@test uploader.storage === storage
|
|
end
|
|
|
|
# --- put_file function (slash detection) ---
|
|
|
|
@testset "put_file function returns nothing for keys with slashes" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
result = @test_logs (:warn, r"Upload aborted") put_file(storage, "a/b.json", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "put_file function returns nothing for keys with multiple slashes" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
result = @test_logs (:warn, r"Upload aborted") put_file(storage, "a/b/c/d.json", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "put_file function returns nothing for key with single slash in middle" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
result = @test_logs (:warn, r"Upload aborted") put_file(storage, "folder/file.json", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "put_file function returns nothing for key with trailing slash pattern" begin
|
|
storage = GarageStorage("https://s3-api.example.com", "key", "secret", "bucket")
|
|
result = @test_logs (:warn, r"Upload aborted") put_file(storage, "a/b", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
# --- Integration tests with real Garage server ---
|
|
|
|
@testset "integration: full CRUD lifecycle" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
|
|
test_key = "testrunner-$(randstring(8)).json"
|
|
test_data = "{\"status\": \"testing\", \"runner\": \"runtests\"}"
|
|
|
|
# Test put_file(storage, key, data) - successful upload
|
|
result = put_file(storage, test_key, test_data)
|
|
@test result !== nothing
|
|
@test haskey(result, :api)
|
|
@test haskey(result, :web)
|
|
@test occursin("testbucket", result.api)
|
|
@test occursin("s3-api", result.api)
|
|
@test occursin("s3-web", result.web)
|
|
|
|
# Test get_file - retrieve the uploaded data
|
|
downloaded = get_file(storage, test_key)
|
|
@test downloaded !== nothing
|
|
@test downloaded isa Vector{UInt8}
|
|
@test String(downloaded) == test_data
|
|
|
|
# Test list_files - verify key appears in listing
|
|
keys = list_files(storage)
|
|
@test test_key in keys
|
|
|
|
# Test put_file callable struct
|
|
test_key2 = "testrunner-callable-$(randstring(8)).json"
|
|
callable_result = put_file(storage)(test_key2, "callable-data")
|
|
@test callable_result !== nothing
|
|
@test callable_result.api != ""
|
|
@test callable_result.web != ""
|
|
|
|
# Verify callable upload
|
|
downloaded2 = get_file(storage, test_key2)
|
|
@test String(downloaded2) == "callable-data"
|
|
|
|
# Test delete_file
|
|
delete_file(storage, test_key)
|
|
deleted_data = get_file(storage, test_key)
|
|
@test deleted_data === nothing
|
|
|
|
# Cleanup callable test object
|
|
delete_file(storage, test_key2)
|
|
deleted_data2 = get_file(storage, test_key2)
|
|
@test deleted_data2 === nothing
|
|
end
|
|
|
|
@testset "integration: put_file rejects slashes even with real server" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
result = @test_logs (:warn, r"Upload aborted") put_file(storage, "folder/test.json", "data")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "integration: get_file returns nothing for missing key" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
result = get_file(storage, "nonexistent-key-$(randstring(8)).json")
|
|
@test result === nothing
|
|
end
|
|
|
|
@testset "integration: binary data upload and download" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
|
|
test_key = "testrunner-binary-$(randstring(8)).bin"
|
|
binary_data = UInt8[0x00, 0x01, 0x02, 0xFF, 0xFE, 0xFD, 0x48, 0x65, 0x6c, 0x6c, 0x6f]
|
|
|
|
result = put_file(storage, test_key, binary_data)
|
|
@test result !== nothing
|
|
|
|
downloaded = get_file(storage, test_key)
|
|
@test downloaded == binary_data
|
|
|
|
delete_file(storage, test_key)
|
|
@test get_file(storage, test_key) === nothing
|
|
end
|
|
|
|
@testset "integration: list_files returns vector of strings" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
|
|
test_key = "testrunner-list-$(randstring(8)).txt"
|
|
put_file(storage, test_key, "list-test-data")
|
|
|
|
keys = list_files(storage)
|
|
@test keys isa Vector{String}
|
|
@test test_key in keys
|
|
|
|
delete_file(storage, test_key)
|
|
end
|
|
|
|
@testset "integration: url format validation" begin
|
|
storage = GarageStorage(
|
|
"https://s3-api.yiem.cc",
|
|
"GKb080154a2e5b19100b1b2c6e",
|
|
"a2c6b1379c2f731d3e6e5a408dd4d6cffca7511717675f55114ff94828febca1",
|
|
"testbucket"
|
|
)
|
|
|
|
test_key = "testrunner-url-$(randstring(8)).json"
|
|
result = put_file(storage, test_key, "url-test")
|
|
|
|
@test startswith(result.api, "https://s3-api.yiem.cc/testbucket/")
|
|
@test occursin("/$test_key", result.api)
|
|
@test startswith(result.web, "https://s3-web.yiem.cc/testbucket/")
|
|
@test occursin("/$test_key", result.web)
|
|
|
|
delete_file(storage, test_key)
|
|
end
|
|
|
|
end
|