lint

BalaBalaYi · Jun 17, 2024 · 11f401e · 11f401e
1 parent 1df73a6
commit 11f401e
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/dlrover/python/tests/test_pod_scaler.py b/dlrover/python/tests/test_pod_scaler.py
@@ -57,14 +57,26 @@ def test_init_pod_template(self):
  ],
  )
 
- def test_create_pod(self):
+ def test_check_master_service_avaliable(self):
  scaler = PodScaler("elasticjob-sample", "default")
  _dlrover_ctx.config_master_port()
  passed = scaler._check_master_service_avaliable(
  "elasticjob-test-master", 2222, 2
  )
  self.assertFalse(passed)
 
+ passed = scaler._check_master_service_avaliable("localhost", 2222, 2)
+ self.assertFalse(passed)
+
+ passed = scaler._check_master_service_avaliable(
+ "localhost", _dlrover_ctx.master_port, 2
+ )
+ self.assertFalse(passed)
+
+ def test_create_pod(self):
+ scaler = PodScaler("elasticjob-sample", "default")
+ _dlrover_ctx.config_master_port()
+
  scaler.start()
  scaler._init_pod_config_by_job()
  scaler._distribution_strategy = DistributionStrategy.PS