chore: Add RetryOnExceptionsExtension for test execution exception ha…

…ndling
dRAGon-Okinawa · Jul 26, 2024 · 996355a · 996355a
1 parent 556d996
commit 996355a
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/backend/src/test/java/ai/dragon/controller/api/raag/OpenAiCompatibleV1ApiControllerTest.java b/backend/src/test/java/ai/dragon/controller/api/raag/OpenAiCompatibleV1ApiControllerTest.java
@@ -30,11 +30,11 @@
 import ai.dragon.enumeration.IngestorLoaderType;
 import ai.dragon.enumeration.LanguageModelType;
 import ai.dragon.enumeration.VectorStoreType;
+import ai.dragon.junit.AbstractTest;
+import ai.dragon.junit.extension.retry.RetryOnExceptions;
 import ai.dragon.repository.FarmRepository;
 import ai.dragon.repository.SiloRepository;
 import ai.dragon.service.IngestorService;
-import ai.dragon.test.AbstractTest;
-import ai.dragon.test.junit.extension.retry.RetryOnExceptions;
 import dev.ai4j.openai4j.OpenAiClient;
 import dev.ai4j.openai4j.OpenAiHttpException;
 import dev.ai4j.openai4j.chat.ChatCompletionRequest;

diff --git a/...est/java/ai/dragon/test/AbstractTest.java → ...st/java/ai/dragon/junit/AbstractTest.java b/...est/java/ai/dragon/test/AbstractTest.java → ...st/java/ai/dragon/junit/AbstractTest.java
@@ -1,4 +1,4 @@
-package ai.dragon.test;
+package ai.dragon.junit;
 
 public abstract class AbstractTest {
     protected boolean canRunOpenAiRelatedTests() {

diff --git a/...it/extension/retry/RetryOnExceptions.java → ...it/extension/retry/RetryOnExceptions.java b/...it/extension/retry/RetryOnExceptions.java → ...it/extension/retry/RetryOnExceptions.java
@@ -1,4 +1,4 @@
-package ai.dragon.test.junit.extension.retry;
+package ai.dragon.junit.extension.retry;
 
 import org.junit.jupiter.api.extension.ExtendWith;
 
@@ -13,4 +13,4 @@
 public @interface RetryOnExceptions {
     int value() default 3; // Default retry count
     Class<? extends Throwable>[] onExceptions() default { Throwable.class }; // Exceptions to retry on
-}
+}
diff --git a/...ion/retry/RetryOnExceptionsExtension.java → ...ion/retry/RetryOnExceptionsExtension.java b/...ion/retry/RetryOnExceptionsExtension.java → ...ion/retry/RetryOnExceptionsExtension.java
@@ -1,4 +1,4 @@
-package ai.dragon.test.junit.extension.retry;
+package ai.dragon.junit.extension.retry;
 
 import java.lang.reflect.Method;
 
@@ -48,4 +48,4 @@ public void beforeEach(ExtensionContext context) throws Exception {
             store.put("retries", 0); // Reset retry count before each test
         }
     }
-}
+}