From 358c5bc6cc94497866a6f1e555b1d1d372a17e01 Mon Sep 17 00:00:00 2001
From: Florian Hussonnois <florian.hussonnois@gmail.com>
Date: Fri, 13 Sep 2024 14:16:37 +0200
Subject: [PATCH] fix(trigger): fix baseUrl when checking run status

---
 .../kestra/plugin/dbt/cloud/TriggerRun.java   | 27 ++++++++++++-------
 1 file changed, 17 insertions(+), 10 deletions(-)

diff --git a/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java b/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java
index 227755b..ccba303 100644
--- a/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java
+++ b/src/main/java/io/kestra/plugin/dbt/cloud/TriggerRun.java
@@ -12,14 +12,20 @@
 import io.micronaut.http.HttpResponse;
 import io.micronaut.http.uri.UriTemplate;
 import io.swagger.v3.oas.annotations.media.Schema;
-import lombok.*;
+import jakarta.validation.constraints.NotNull;
+import lombok.Builder;
+import lombok.EqualsAndHashCode;
+import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.ToString;
 import lombok.experimental.SuperBuilder;
 import org.slf4j.Logger;
 
 import java.net.URI;
 import java.time.Duration;
-import java.util.*;
-import jakarta.validation.constraints.NotNull;
+import java.util.HashMap;
+import java.util.List;
+import java.util.Map;
 
 @SuperBuilder
 @ToString
@@ -226,13 +232,14 @@ public TriggerRun.Output run(RunContext runContext) throws Exception {
         }
 
         CheckStatus checkStatusJob = CheckStatus.builder()
-                                        .runId(runId.toString())
-                                        .token(getToken())
-                                        .accountId(getAccountId())
-                                        .pollFrequency(getPollFrequency())
-                                        .maxDuration(getMaxDuration())
-                                        .parseRunResults(getParseRunResults())
-                                        .build();
+            .runId(runId.toString())
+            .baseUrl(getBaseUrl())
+            .token(getToken())
+            .accountId(getAccountId())
+            .pollFrequency(getPollFrequency())
+            .maxDuration(getMaxDuration())
+            .parseRunResults(getParseRunResults())
+            .build();
 
         CheckStatus.Output runOutput = checkStatusJob.run(runContext);