@@ -251,12 +251,6 @@ def get_sdk_folder_with_package_name(self):
251251 folder_info = glob (f'sdk/*/azure-mgmt-{ self .package_name } ' )[0 ]
252252 self .sdk_folder = Path (folder_info ).parts [1 ]
253253
254- def prepare_branch (self ):
255- if self .spec_readme :
256- self .prepare_branch_with_readme ()
257- # else:
258- # self.prepare_branch_with_base_branch()
259-
260254 def check_sdk_readme (self ):
261255 sdk_readme = str (Path (f'sdk/{ self .sdk_folder } /azure-mgmt-{ self .package_name } /README.md' ))
262256
@@ -470,12 +464,13 @@ def prepare_test_env(self):
470464 def run_test_proc (self ):
471465 # run test
472466 os .chdir (self .sdk_code_path ())
473- succeeded_result = 'Live test success'
474- failed_result = 'Live test fail, detailed info is in pipeline log(search keyword FAILED)!!!'
467+ test_mode = "Live test" if os .getenv ("AZURE_TEST_RUN_LIVE" ) else "Recording test"
468+ succeeded_result = f'{ test_mode } success'
469+ failed_result = f'{ test_mode } fail, detailed info is in pipeline log(search keyword FAILED)!!!'
475470 try :
476471 print_check (f'pytest --collect-only' )
477472 except :
478- log ('live test run done, do not find any test !!!' )
473+ log (f' { test_mode } run done, do not find any test !!!' )
479474 self .test_result = succeeded_result
480475 return
481476
@@ -485,7 +480,7 @@ def run_test_proc(self):
485480 log ('some test failed, please fix it locally' )
486481 self .test_result = failed_result
487482 else :
488- log ('live test run done, do not find failure !!!' )
483+ log (f' { test_mode } run done, do not find failure !!!' )
489484 self .test_result = succeeded_result
490485
491486 def run_test (self ):
@@ -597,10 +592,16 @@ def create_pr(self):
597592 self .issue_comment ()
598593
599594 def run (self ):
600- self .prepare_branch ()
601- self .check_file ()
602- self .run_test ()
603- self .create_pr ()
595+ if "https:" in self .spec_readme :
596+ self .prepare_branch_with_readme ()
597+ self .check_file ()
598+ self .run_test ()
599+ self .create_pr ()
600+ else :
601+ self .sdk_folder = self .spec_readme .split ('/' )[0 ]
602+ self .package_name = self .spec_readme .split ('/' )[- 1 ].split ('-' )[- 1 ]
603+ self .checkout_branch ("DEBUG_SDK_BRANCH" , "azure-sdk-for-python" )
604+ self .run_test ()
604605
605606
606607if __name__ == '__main__' :
0 commit comments