fix test

fly-apps · Jun 25, 2024 · 2d4b622 · 2d4b622
1 parent c3ce2e2
commit 2d4b622
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/flypg/pg_test.go b/internal/flypg/pg_test.go
@@ -117,6 +117,7 @@ func TestPGConfigInitialization(t *testing.T) {
 	t.Run("cloud-archiving", func(t *testing.T) {
 		t.Setenv("CLOUD_ARCHIVING_ENABLED", "true")
 		t.Setenv("AWS_ACCESS_KEY_ID", "my-key")
+		t.Setenv("AWS_ENDPOINT_URL", "https://fly.storage.tigris.dev")
 		t.Setenv("AWS_SECRET_ACCESS_KEY", "my-secret")
 		t.Setenv("AWS_BUCKET_NAME", "my-bucket")
 
@@ -135,7 +136,8 @@ func TestPGConfigInitialization(t *testing.T) {
 			t.Fatalf("expected archive_mode to be on, got %v", cfg["archive_mode"])
 		}
 
-		expected := fmt.Sprintf("'barman-cloud-wal-archive --cloud-provider aws-s3 --endpoint-url https://fly.storage.tigris.dev s3://%s %s %%p'",
+		expected := fmt.Sprintf("'barman-cloud-wal-archive --cloud-provider aws-s3 --endpoint-url %s s3://%s %s %%p'",
+			os.Getenv("AWS_ENDPOINT_URL"),
 			os.Getenv("AWS_BUCKET_NAME"),
 			pgConf.AppName)