@@ -189,7 +189,7 @@ def upload_results_to_azure(file_to_upload, file_name, file_directory):
189189 file_service = FileService (connection_string = args .connection_string )
190190 file_service .create_file_from_path (share_name = args .share_name , file_name = file_name , directory_name = file_directory , local_file_path = file_to_upload , content_settings = ContentSettings (content_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet' ))
191191
192- def download_file_from_azure (file_to_download , file_name , file_directory ):
192+ def download_file_from_azure (out_file_path , file_name , file_directory ):
193193 """
194194 Downloads a file from Azure.
195195
@@ -200,7 +200,7 @@ def download_file_from_azure(file_to_download, file_name, file_directory):
200200 None
201201 """
202202 file_service = FileService (connection_string = args .connection_string )
203- file = file_service .get_file_to_path (share_name = args .share_name , file_name = file_name , directory_name = file_directory , file_path = file_to_download )
203+ file = file_service .get_file_to_path (share_name = args .share_name , file_name = file_name , directory_name = file_directory , file_path = out_file_path )
204204 return file .name
205205
206206def get_git_root ():
@@ -250,7 +250,7 @@ def find_ql_test_paths(directory, extension):
250250 # exclude wfp folder until correct test template is added
251251 ignore_paths = ["wfp" , "QueryTemplate" , "TestTemplate" , ".vs" ]
252252 root = root .replace ("\\ " , "/" )
253- if any (path in ignore_paths for x in root .split ("/" )):
253+ if any (path in ignore_paths for path in root .split ("/" )):
254254 print_conditionally ("Skipping: " + root )
255255 continue
256256 if "TestTemplate" in root :
@@ -265,6 +265,7 @@ def find_ql_test_paths(directory, extension):
265265 ql_obj = ql_test_attributes (use_ntifs = use_ntifs , use_cpp = use_cpp )
266266 ql_obj .set_use_cpp (use_cpp )
267267 ql_files_map [os .path .join (root , file )] = ql_obj
268+ print_conditionally ("Found: " + os .path .join (root , file ))
268269
269270 return ql_files_map
270271
@@ -780,14 +781,14 @@ def run_tests_external_drivers(ql_tests_dict):
780781 result_file = "external_drivers_results.xlsx"
781782 with pd .ExcelWriter (result_file ) as writer :
782783 health_df .to_excel (writer , sheet_name = "Results" )
783- codeql_version_df .to_excel (writer , sheet_name = "CodeQL Version" )
784- codeql_packs_df .to_excel (writer , sheet_name = "CodeQL Packs" )
785- system_info_df .to_excel (writer , sheet_name = "System Info" )
784+ local_codeql_version_df .to_excel (writer , sheet_name = "Local CodeQL Version" )
785+ local_codeql_packs_df .to_excel (writer , sheet_name = "Local CodeQL Packs" )
786+ local_system_info_df .to_excel (writer , sheet_name = "Local System Info" )
786787 with pd .ExcelWriter ("detailed" + result_file ) as writer :
787788 detailed_health_df .to_excel (writer , sheet_name = "Results" )
788- codeql_version_df .to_excel (writer , sheet_name = "CodeQL Version" )
789- codeql_packs_df .to_excel (writer , sheet_name = "CodeQL Packs" )
790- system_info_df .to_excel (writer , sheet_name = "System Info" )
789+ local_codeql_version_df .to_excel (writer , sheet_name = "Local CodeQL Version" )
790+ local_codeql_packs_df .to_excel (writer , sheet_name = "Local CodeQL Packs" )
791+ local_system_info_df .to_excel (writer , sheet_name = "Local System Info" )
791792 if args .compare_results :
792793 compare_health_results ("detailed" + result_file )
793794 compare_health_results (result_file )
@@ -838,9 +839,8 @@ def compare_health_results(curr_results_path):
838839 try :
839840 prev_results = 'azure-' + curr_results_path
840841 print_conditionally ("Downloading previous results from Azure: " + prev_results )
841- temp_file = download_file_from_azure (file_to_download = prev_results ,
842+ _ = download_file_from_azure (out_file_path = prev_results ,
842843 file_name = curr_results_path , file_directory = "" )
843- print_conditionally ("Downloaded previous results: " + temp_file )
844844
845845 except Exception as e :
846846 if "ResourceNotFound" in str (e ):
@@ -853,7 +853,7 @@ def compare_health_results(curr_results_path):
853853 prev_results_df = pd .read_excel (prev_results , index_col = 0 , sheet_name = 0 )
854854 prev_results_codeql_version_df = pd .read_excel (prev_results , index_col = 0 , sheet_name = 1 )
855855 prev_results_codeql_packs_df = pd .read_excel (prev_results , index_col = 0 , sheet_name = 2 )
856- prev_results_system_info_df = pd .read_excel (prev_results , index_col = 0 , sheet_name = 3 )
856+ prev_results_local_system_info_df = pd .read_excel (prev_results , index_col = 0 , sheet_name = 3 )
857857 curr_results_df = pd .read_excel (curr_results_path , index_col = 0 , sheet_name = 0 )
858858 print_conditionally ("Comparing results..." )
859859 print_conditionally ("Previous results: " , prev_results )
@@ -879,16 +879,29 @@ def compare_health_results(curr_results_path):
879879
880880 with pd .ExcelWriter ("diff" + curr_results_path ) as writer :
881881 diff_results .to_excel (writer , sheet_name = "Diff" )
882- codeql_version_df .to_excel (writer , sheet_name = "Current CodeQL Version" )
883- codeql_packs_df .to_excel (writer , sheet_name = "Current CodeQL Packs" )
884- system_info_df .to_excel (writer , sheet_name = "Current System Info" )
885- prev_results_codeql_version_df .to_excel (writer , sheet_name = "Previous CodeQL Version" )
886- prev_results_codeql_packs_df .to_excel (writer , sheet_name = "Previous CodeQL Packs" )
887- prev_results_system_info_df .to_excel (writer , sheet_name = "Previous System Info" )
882+ local_codeql_version_df .to_excel (writer , sheet_name = "Local CodeQL Version" )
883+ local_codeql_packs_df .to_excel (writer , sheet_name = "Local CodeQL Packs" )
884+ local_system_info_df .to_excel (writer , sheet_name = "Local System Info" )
885+ prev_results_codeql_version_df .to_excel (writer , sheet_name = "Last Stored CodeQL Version" )
886+ prev_results_codeql_packs_df .to_excel (writer , sheet_name = "Last Stored CodeQL Packs" )
887+ prev_results_local_system_info_df .to_excel (writer , sheet_name = "Last Stored System Info" )
888888 print_conditionally ("Saved diff results" )
889889
890890 if not args .local_result_storage :
891891 # upload new results to Azure
892+ if args .overwrite_azure_results :
893+ print ("!! Overwriting Azure results !!" )
894+ print ("Type 'yes' to confirm" )
895+ confirm = input ()
896+ if confirm != "yes" :
897+ print ("Exiting" )
898+ exit (1 )
899+ else :
900+ double_confirm = input ("Are you sure?" )
901+ if double_confirm != "yes" :
902+ print ("Exiting" )
903+ exit (1 )
904+
892905 if args .overwrite_azure_results :
893906 print_conditionally ("Uploading results" )
894907 upload_results_to_azure (file_to_upload = curr_results_path ,
@@ -934,14 +947,14 @@ def run_tests(ql_tests_dict):
934947 result_file = "functiontestresults.xlsx"
935948 with pd .ExcelWriter (result_file ) as writer :
936949 health_df .to_excel (writer , sheet_name = "Results" )
937- codeql_version_df .to_excel (writer , sheet_name = "CodeQL Version" )
938- codeql_packs_df .to_excel (writer , sheet_name = "CodeQL Packs" )
939- system_info_df .to_excel (writer , sheet_name = "System Info" )
950+ local_codeql_version_df .to_excel (writer , sheet_name = "Local CodeQL Version" )
951+ local_codeql_packs_df .to_excel (writer , sheet_name = "Local CodeQL Packs" )
952+ local_system_info_df .to_excel (writer , sheet_name = "Local System Info" )
940953 with pd .ExcelWriter ("detailed" + result_file ) as writer :
941954 detailed_health_df .to_excel (writer , sheet_name = "Results" )
942- codeql_version_df .to_excel (writer , sheet_name = "CodeQL Version" )
943- codeql_packs_df .to_excel (writer , sheet_name = "CodeQL Packs" )
944- system_info_df .to_excel (writer , sheet_name = "System Info" )
955+ local_codeql_version_df .to_excel (writer , sheet_name = "Local CodeQL Version" )
956+ local_codeql_packs_df .to_excel (writer , sheet_name = "Local CodeQL Packs" )
957+ local_system_info_df .to_excel (writer , sheet_name = "Local System Info" )
945958 if args .compare_results :
946959 compare_health_results ("detailed" + result_file )
947960 compare_health_results (result_file )
@@ -1008,26 +1021,18 @@ def find_sln_file(path):
10081021 parser .add_argument ('--overwrite_azure_results' , help = 'Overwrite Azure results' ,action = 'store_true' ,required = False ,)
10091022 parser .add_argument ('--build_database_only' , help = 'Build database only' ,action = 'store_true' ,required = False ,)
10101023 args = parser .parse_args ()
1011-
1012- if args .overwrite_azure_results :
1013- print ("Overwriting Azure results" )
1014- print ("Type 'yes' to confirm" )
1015- confirm = input ()
1016- if confirm != "yes" :
1017- print ("Exiting" )
1018- exit (1 )
1019-
1024+
10201025 if args .codeql_path :
10211026 codeql_path = args .codeql_path
10221027 else :
10231028 codeql_path = "codeql"
10241029
10251030 codeql_version = subprocess .run ([codeql_path , "version" ], capture_output = True ) # test codeql is working
1026- codeql_version_df = pd .DataFrame ([x for x in codeql_version .stdout .decode ().split ('\n ' )])
1031+ local_codeql_version_df = pd .DataFrame ([x for x in codeql_version .stdout .decode ().split ('\n ' )])
10271032 codeql_packs = subprocess .run ([codeql_path , "resolve" , "qlpacks" ], capture_output = True )
1028- codeql_packs_df = pd .DataFrame ([x for x in codeql_packs .stdout .decode ().split ('\n ' )])
1033+ local_codeql_packs_df = pd .DataFrame ([x for x in codeql_packs .stdout .decode ().split ('\n ' )])
10291034 system_info = subprocess .run (["systeminfo" ], capture_output = True )
1030- system_info_df = pd .DataFrame ([x for x in system_info .stdout .decode ().split ('\n ' )])
1035+ local_system_info_df = pd .DataFrame ([x for x in system_info .stdout .decode ().split ('\n ' )])
10311036
10321037
10331038 if args .compare_results_no_build :
0 commit comments