@@ -60,35 +60,35 @@ def n_close(conns):
6060def debug_output (qs ,qs_len ,pid ,query ,expected ):
6161something_happened = False
6262if (qs_len and len (qs )!= qs_len ):
63- print ("len(qs): " ,len (qs ),", expected: " ,qs_len )
63+ print ("len(qs): " ,len (qs ),", expected: " ,qs_len )
6464something_happened = True
6565if (pid and qs [0 ][0 ]!= pid ):
66- print ("qs[0][0]: " ,qs [0 ][0 ]," = " ,pid )
66+ print ("qs[0][0]: " ,qs [0 ][0 ]," = " ,pid )
6767something_happened = True
6868if (qs [0 ][1 ]!= 0 ):
69- print ("qs[0][1]: " ,qs [0 ][1 ],", expected: 0" )
69+ print ("qs[0][1]: " ,qs [0 ][1 ],", expected: 0" )
7070something_happened = True
7171if (qs [0 ][2 ]!= query ):
72- print ("qs[0][2]:\n " ,qs [0 ][2 ])
73- print ("Expected:\n " ,query )
72+ print ("qs[0][2]:\n " ,qs [0 ][2 ])
73+ print ("Expected:\n " ,query )
7474something_happened = True
7575if (not (re .match (expected ,qs [0 ][3 ]))):
76- print ("qs[0][3]:\n " ,qs [0 ][3 ])
77- print ("Expected:\n " ,expected )
76+ print ("qs[0][3]:\n " ,qs [0 ][3 ])
77+ print ("Expected:\n " ,expected )
7878something_happened = True
7979if (qs [0 ][4 ]!= None ):
80- print ("qs[0][4]: " ,qs [0 ][4 ],"Expected: None" )
80+ print ("qs[0][4]: " ,qs [0 ][4 ],"Expected: None" )
8181something_happened = True
8282if (qs_len and len (qs )> qs_len ):
8383for i in range (qs_len ,len (qs )):
84- print ("qs[" ,i ,"][0]: " ,qs [i ][0 ])
85- print ("qs[" ,i ,"][1]: " ,qs [i ][1 ])
86- print ("qs[" ,i ,"][2]: " ,qs [i ][2 ])
87- print ("qs[" ,i ,"][3]: " ,qs [i ][3 ])
88- print ("qs[" ,i ,"][4]: " ,qs [i ][4 ])
84+ print ("qs[" ,i ,"][0]: " ,qs [i ][0 ])
85+ print ("qs[" ,i ,"][1]: " ,qs [i ][1 ])
86+ print ("qs[" ,i ,"][2]: " ,qs [i ][2 ])
87+ print ("qs[" ,i ,"][3]: " ,qs [i ][3 ])
88+ print ("qs[" ,i ,"][4]: " ,qs [i ][4 ])
8989something_happened = True
9090if (something_happened ):
91- print ("If test have not crashed, then it's OK" )
91+ print ("If test have not crashed, then it's OK" )
9292
9393def notices_warning ():
9494if (len (notices )> 0 ):
@@ -546,7 +546,7 @@ def load_tpcds_data(config):
546546table_name = os .path .splitext (os .path .basename (table_datafile ))[0 ]
547547copy_cmd = "COPY %s FROM '/pg/testdir/tmp_stress/tpcds-kit/tools/tables/%s' CSV DELIMITER '|'" % (table_name ,table_datafile )
548548
549- print (" Loading table" ,table_name )
549+ print (' Loading table' ,table_name )
550550# cur.execute("TRUNCATE %s" % table_name)
551551cur .execute (copy_cmd )
552552
@@ -559,24 +559,28 @@ def load_tpcds_data(config):
559559
560560print ('done!' )
561561
562- def stress_test (config ):
562+ def test_tpc_ds (config ):
563563"""TPC-DS stress test"""
564564global stress_in_progress
565565
566566stress_in_progress = True
567567load_tpcds_data (config )
568568
569569print ('Preparing TPC-DS queries...' )
570- # Execute query in separate thread
571- # with open('tmp_stress/tpcds-kit/tools/query_0.sql', 'r') as f:
572- with open ('tests/query_tpcds.sql' ,'r' )as f :
573- sql = f .read ()
574-
575- queries = sql .split (';' )
576- for i ,query in enumerate (queries ):
577- queries [i ]= query .replace ('%' ,'%%' )
578- if (len (query .strip ())== 0 ):
579- del queries [i ]
570+ # # Execute query in separate thread
571+ # # with open('tmp_stress/tpcds-kit/tools/query_0.sql', 'r') as f:
572+ # with open('tests/query_tpcds.sql', 'r') as f:
573+ # sql = f.read()
574+
575+ # queries = sql.split(';')
576+ # for i, query in enumerate(queries):
577+ # queries[i] = query.replace('%','%%')
578+ # if (len(query.strip()) == 0):
579+ # del queries[i]
580+ queries = []
581+ for query_file in sorted (os .listdir ('tmp_stress/tpcds-result-reproduction/query_qualification/' )):
582+ with open ('tmp_stress/tpcds-result-reproduction/query_qualification/%s' % query_file ,'r' )as f :
583+ queries .append (f .read ())
580584
581585acon ,= n_async_connect (config )
582586
@@ -598,6 +602,6 @@ def stress_test(config):
598602n_close ((acon ,))
599603
600604if len (timeout_list )> 0 :
601- print ('There were pg_query_state timeouts (%s s) on queries:' % TPC_DS_STATEMENT_TIMEOUT ,timeout_list )
605+ print ('\n There were pg_query_state timeouts (%s s) on queries:' % TPC_DS_STATEMENT_TIMEOUT ,timeout_list )
602606
603607stress_in_progress = False