[#6414] improve(CLI): Add catalog command context CLI

fix test bugs.
apache · Feb 11, 2025 · 2135aab · 2135aab
1 parent a65b394
commit 2135aab
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/clients/cli/src/test/java/org/apache/gravitino/cli/TestCatalogCommands.java b/clients/cli/src/test/java/org/apache/gravitino/cli/TestCatalogCommands.java
@@ -70,7 +70,6 @@ void setUp() {
     mockOptions = mock(Options.class);
     System.setOut(new PrintStream(outContent));
     System.setErr(new PrintStream(errContent));
-    Main.useExit = false;
   }
 
   @AfterEach
@@ -179,6 +178,7 @@ void testCreateCatalogCommand() {
 
   @Test
   void testCreateCatalogCommandWithoutProvider() {
+    Main.useExit = false;
     CommandContext mockContext = mock(CommandContext.class);
     when(mockContext.url()).thenReturn("http://localhost:8080");
     CreateCatalog mockCreateCatalog =
@@ -258,6 +258,7 @@ void testSetCatalogPropertyCommand() {
 
   @Test
   void testSetCatalogPropertyCommandWithoutPropertyAndValue() {
+    Main.useExit = false;
     CommandContext mockContext = mock(CommandContext.class);
     when(mockContext.url()).thenReturn("http://localhost:8080");
     SetCatalogProperty mockSetProperty =
@@ -270,6 +271,7 @@ void testSetCatalogPropertyCommandWithoutPropertyAndValue() {
 
   @Test
   void testSetCatalogPropertyCommandWithoutProperty() {
+    Main.useExit = false;
     CommandContext mockContext = mock(CommandContext.class);
     when(mockContext.url()).thenReturn("http://localhost:8080");
     SetCatalogProperty mockSetProperty =