simple ut

apache · Mar 7, 2024 · 8054753 · 8054753
1 parent 3c4432a
commit 8054753
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/...it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerAppRegisterSuite.scala b/...it/src/test/scala/org/apache/celeborn/tests/client/LifecycleManagerAppRegisterSuite.scala
@@ -19,20 +19,22 @@ package org.apache.celeborn.tests.client
 
 import java.util
 
-import org.apache.celeborn.client.{LifecycleManager, WithShuffleClientSuite}
+import org.apache.celeborn.CelebornFunSuite
+import org.apache.celeborn.client.LifecycleManager
 import org.apache.celeborn.common.CelebornConf
 import org.apache.celeborn.common.identity.UserIdentifier
 import org.apache.celeborn.service.deploy.MiniClusterFeature
 
-class LifecycleManagerAppRegisterSuite extends WithShuffleClientSuite with MiniClusterFeature {
+class LifecycleManagerAppRegisterSuite extends CelebornFunSuite with MiniClusterFeature {
 
-  celebornConf.set(CelebornConf.APP_REGISTER_ENABLED, true)
+  private val celebornConf: CelebornConf = new CelebornConf()
+    .set(CelebornConf.APP_REGISTER_ENABLED, true)
     .set(CelebornConf.INTERNAL_PORT_ENABLED, true)
 
-  override val userIdentifier = UserIdentifier("test", "celeborn")
+  private val APP = "app-" + System.currentTimeMillis()
+  private val userIdentifier = UserIdentifier("test", "celeborn")
 
   override def beforeAll(): Unit = {
-    super.beforeAll()
     System.setProperty(CelebornConf.QUOTA_USER_SPECIFIC_TENANT.key, userIdentifier.tenantId)
     System.setProperty(CelebornConf.QUOTA_USER_SPECIFIC_USERNAME.key, userIdentifier.name)
     val (master, _) = setupMiniClusterWithRandomPorts(