生成jmeter可用文件和必要参数化的csv

领导说让用jmeter这个工具参数化测试soap接口，做完参数化后保存的文件是jmx格式，经过同事启发，打开一看这个不是xml格式文件么，应该可以利用python写一个xml格式然后保存成jmx格式的，这样只要维护excel中的测试用例就行了，真正的实现数据驱动参数化。不过在过程中也遇到过编码问题，实在头疼，所以我建议大家还是用英文在线程组或者循环中定义名称吧。在excel中写好接口测试用例，然后运行此文件后生成jmeter可用的jmx和csv文件

#-*-coding:utf8-*-
#__author__=xu.duan
import xlrd
import sys
import os
import os.path
import getopt
import csv
reload(sys)
sys.setdefaultencoding('gb18030')

def getPathFileName(path): #从path中获取线程组名称
    tmp = path.split('/');
    return tmp[len(tmp) -1 ].split('.')[0];

def xlsToxmlPath(path):#把excel表名称直接引用成xml的名称
    return path.split('.')[0] + '.jmx';

def CSVToxmlPath(path):#把excel表名称直接引用成csv的名称
    return path.split('.')[0] + '.csv';

def exportjmx(path):#生成jmx文件过程
    data = xlrd.open_workbook(path,
            formatting_info = True, encoding_override="utf-8")
    table = data.sheets()[0]

    f = open(xlsToxmlPath(path), 'wb')
    f.write(u'<?xml version="1.0" encoding="UTF-8"?>
')
    f.write(u'<jmeterTestPlan version="1.2" properties="2.6" jmeter="2.11 r1554548">
')
    
    f.write(u'''<hashTree>
    <TestPlan guiclass="TestPlanGui" testclass="TestPlan" testname="Test Plan" enabled="true">
      <stringProp name="TestPlan.comments"></stringProp>
      <boolProp name="TestPlan.functional_mode">false</boolProp>
      <boolProp name="TestPlan.serialize_threadgroups">false</boolProp>
      <elementProp name="TestPlan.user_defined_variables" elementType="Arguments" guiclass="ArgumentsPanel" testclass="Arguments" testname="用户定义的变量" enabled="true">
        <collectionProp name="Arguments.arguments"/>
      </elementProp>
      <stringProp name="TestPlan.user_define_classpath"></stringProp>
    </TestPlan>
    <hashTree>
      <ThreadGroup guiclass="ThreadGroupGui" testclass="ThreadGroup" testname="info3" enabled="true">
        <elementProp name="ThreadGroup.main_controller" elementType="LoopController" guiclass="LoopControlPanel" testclass="LoopController" testname="Loop Controller" enabled="true">
          <boolProp name="LoopController.continue_forever">false</boolProp>
          <intProp name="LoopController.loops">-1</intProp>
        </elementProp>
        <stringProp name="ThreadGroup.num_threads">1</stringProp>
        <stringProp name="ThreadGroup.ramp_time">1</stringProp>
        <longProp name="ThreadGroup.start_time">1281132211000</longProp>
        <longProp name="ThreadGroup.end_time">1281132211000</longProp>
        <boolProp name="ThreadGroup.scheduler">false</boolProp>
        <stringProp name="ThreadGroup.on_sample_error">continue</stringProp>
        <stringProp name="ThreadGroup.duration"></stringProp>
        <stringProp name="ThreadGroup.delay"></stringProp>
      </ThreadGroup>
      <hashTree>
        <LoopController guiclass="LoopControlPanel" testclass="LoopController" testname="Step 1" enabled="true">
          <boolProp name="LoopController.continue_forever">false</boolProp>
          <stringProp name="LoopController.loops">2</stringProp>
        </LoopController>
        <hashTree>
          <CSVDataSet guiclass="TestBeanGUI" testclass="CSVDataSet" testname="CSV Data Set Config" enabled="true">
            <stringProp name="delimiter">,</stringProp>
            <stringProp name="fileEncoding">UTF-8</stringProp>
            <stringProp name="filename">E:/T/info3.0/text doc/info3.csv</stringProp>
            <boolProp name="quotedData">true</boolProp>
            <boolProp name="recycle">true</boolProp>
            <stringProp name="shareMode">shareMode.all</stringProp>
            <boolProp name="stopThread">false</boolProp>
            <stringProp name="variableNames">IP,uri,words,language,channel,output,sign</stringProp>
          </CSVDataSet>
          <hashTree/>
          <HTTPSamplerProxy guiclass="HttpTestSampleGui" testclass="HTTPSamplerProxy" testname="testcase1" enabled="true">
            <elementProp name="HTTPsampler.Arguments" elementType="Arguments" guiclass="HTTPArgumentsPanel" testclass="Arguments" testname="User Defined Variables" enabled="true">''')

    f.write(u'<collectionProp name="Arguments.arguments">
')
    
    for i in range(2, table.ncols):
        #s = u'<collectionProp name="Arguments.arguments">
';
        #tmp = [u'%s = "%s"' % (str(table.cell_value(0,j)), str(table.cell_value(i,j))) for j in range(table.ncols)];
        #s +=(u'elementProp name="{{{0}}}" elementType="HTTPArgument"'.format(table.cell_value(0,i)));
        s =(u'''<elementProp name="{0}" elementType="HTTPArgument">
                  <boolProp name="HTTPArgument.always_encode">true</boolProp>
                  <stringProp name="Argument.value">${{{0}}}</stringProp>
                  <stringProp name="Argument.metadata">=</stringProp>
                  <boolProp name="HTTPArgument.use_equals">true</boolProp>
                  <stringProp name="Argument.name">{0}</stringProp>
                </elementProp> 
'''.format(table.cell_value(0,i)));
        #s += ' '.join(map(str,tmp));
        
        f.write(s);
    f.write(u'</collectionProp>
');
    f.write(u'''</elementProp>
            <stringProp name="HTTPSampler.domain">${{{0}}}</stringProp>
            <stringProp name="HTTPSampler.port"></stringProp>
            <stringProp name="HTTPSampler.connect_timeout"></stringProp>
            <stringProp name="HTTPSampler.response_timeout"></stringProp>
            <stringProp name="HTTPSampler.protocol">http</stringProp>
            <stringProp name="HTTPSampler.contentEncoding"></stringProp>
            <stringProp name="HTTPSampler.path">${{{1}}}</stringProp>
            <stringProp name="HTTPSampler.method">GET</stringProp>
            <boolProp name="HTTPSampler.follow_redirects">false</boolProp>
            <boolProp name="HTTPSampler.auto_redirects">true</boolProp>
            <boolProp name="HTTPSampler.use_keepalive">true</boolProp>
            <boolProp name="HTTPSampler.DO_MULTIPART_POST">false</boolProp>
            <stringProp name="HTTPSampler.implementation">Java</stringProp>
            <boolProp name="HTTPSampler.monitor">false</boolProp>
            <stringProp name="HTTPSampler.embedded_url_re"></stringProp>
          </HTTPSamplerProxy>
          <hashTree/>
          <ResponseAssertion guiclass="AssertionGui" testclass="ResponseAssertion" testname="response to assert" enabled="true">
            <collectionProp name="Asserion.test_strings">
              <stringProp name="1259833018">Successful</stringProp>
            </collectionProp>
            <stringProp name="Assertion.test_field">Assertion.response_data</stringProp>
            <boolProp name="Assertion.assume_success">false</boolProp>
            <intProp name="Assertion.test_type">16</intProp>
          </ResponseAssertion>
          <hashTree/>
        </hashTree>
'''.format(table.cell_value(0,0),table.cell_value(0,1)));
    f.write(u'''<ResultCollector guiclass="ViewResultsFullVisualizer" testclass="ResultCollector" testname="Check the result tree" enabled="true">
          <boolProp name="ResultCollector.error_logging">false</boolProp>
          <objProp>
            <value class="SampleSaveConfiguration">
              <time>true</time>
              <latency>true</latency>
              <timestamp>true</timestamp>
              <success>true</success>
              <label>true</label>
              <code>true</code>
              <message>true</message>
              <threadName>true</threadName>
              <dataType>true</dataType>
              <encoding>false</encoding>
              <assertions>true</assertions>
              <subresults>true</subresults>
              <responseData>false</responseData>
              <samplerData>false</samplerData>
              <xml>true</xml>
              <fieldNames>false</fieldNames>
              <responseHeaders>false</responseHeaders>
              <requestHeaders>false</requestHeaders>
              <responseDataOnError>false</responseDataOnError>
              <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
              <assertionsResultsToSave>0</assertionsResultsToSave>
              <bytes>true</bytes>
            </value>
          </objProp>
          <objProp>
            <value class="SampleSaveConfiguration">
              <time>true</time>
              <latency>true</latency>
              <timestamp>true</timestamp>
              <success>true</success>
              <label>true</label>
              <code>true</code>
              <message>true</message>
              <threadName>true</threadName>
              <dataType>true</dataType>
              <encoding>false</encoding>
              <assertions>true</assertions>
              <subresults>true</subresults>
              <responseData>false</responseData>
              <samplerData>false</samplerData>
              <xml>false</xml>
              <fieldNames>false</fieldNames>
              <responseHeaders>false</responseHeaders>
              <requestHeaders>false</requestHeaders>
              <responseDataOnError>false</responseDataOnError>
              <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
              <assertionsResultsToSave>0</assertionsResultsToSave>
              <bytes>true</bytes>
              <threadCounts>true</threadCounts>
            </value>
          </objProp>
          <stringProp name="filename"></stringProp>
        </ResultCollector>
        <hashTree/>
        <ResultCollector guiclass="StatVisualizer" testclass="ResultCollector" testname="Aggregate Report" enabled="true">
          <boolProp name="ResultCollector.error_logging">false</boolProp>
          <objProp>
            <value class="SampleSaveConfiguration">
              <time>true</time>
              <latency>true</latency>
              <timestamp>true</timestamp>
              <success>true</success>
              <label>true</label>
              <code>true</code>
              <message>true</message>
              <threadName>true</threadName>
              <dataType>true</dataType>
              <encoding>false</encoding>
              <assertions>true</assertions>
              <subresults>true</subresults>
              <responseData>false</responseData>
              <samplerData>false</samplerData>
              <xml>true</xml>
              <fieldNames>false</fieldNames>
              <responseHeaders>false</responseHeaders>
              <requestHeaders>false</requestHeaders>
              <responseDataOnError>false</responseDataOnError>
              <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
              <assertionsResultsToSave>0</assertionsResultsToSave>
              <bytes>true</bytes>
            </value>
          </objProp>
          <objProp>
            <value class="SampleSaveConfiguration">
              <time>true</time>
              <latency>true</latency>
              <timestamp>true</timestamp>
              <success>true</success>
              <label>true</label>
              <code>true</code>
              <message>true</message>
              <threadName>true</threadName>
              <dataType>true</dataType>
              <encoding>false</encoding>
              <assertions>true</assertions>
              <subresults>true</subresults>
              <responseData>false</responseData>
              <samplerData>false</samplerData>
              <xml>false</xml>
              <fieldNames>false</fieldNames>
              <responseHeaders>false</responseHeaders>
              <requestHeaders>false</requestHeaders>
              <responseDataOnError>false</responseDataOnError>
              <saveAssertionResultsFailureMessage>false</saveAssertionResultsFailureMessage>
              <assertionsResultsToSave>0</assertionsResultsToSave>
              <bytes>true</bytes>
              <threadCounts>true</threadCounts>
            </value>
          </objProp>
          <stringProp name="filename"></stringProp>
             </ResultCollector>
             <hashTree/>
            </hashTree>
            </hashTree>
            </hashTree>
''')
    f.write(u'</jmeterTestPlan>');
    f.close

def exportcsv(path):#转换csv过程
    data = xlrd.open_workbook(path,
            formatting_info = True, encoding_override="utf-8")
    table = data.sheet_by_name('test1')
    csv_file = open(CSVToxmlPath(path), 'wb')
    wr = csv.writer(csv_file, quoting=csv.QUOTE_ALL)

    for rownum in xrange(table.nrows):
        wr.writerow(table.row_values(rownum))
    csv_file.close()

def usage():#定义argv
    print '1.generate a CSV and jmx  :'
    print sys.argv[0], '--mode genCSVJMX --input_file <file_path> --output_file <output_excel>'

if __name__ == '__main__':
    try:
        opts, args = getopt.getopt(sys.argv[1:], 'm:i:h', 
            ['mode=','input_file=',
             'help'])
    except getopt.GetoptError, e:
        print str(e)
        sys.exit(2) 
    cur_mode = 'genCSVJMX'
    input_file = None 

    for k, v in opts:
        if k in ('-h', '--help'):
            usage()
            sys.exit(1)
        elif k in ('-m', '--mode'):
            cur_mode = v
        elif k in ('--input_file', '-i'):
            input_file = v
        else:
            print 'None option'

    if cur_mode == 'genCSVJMX':
        if os.path.exists(input_file):
            exportjmx('{0}'.format(input_file))
            exportcsv('{0}'.format(input_file))
        else:
            if len(input_file) == 0:
                print 'Error: lack input para --input_file'
                usage()
                sys.exit(1)

            if not os.path.exists(input_file):
                print 'Error: log path does not exist!'
                sys.exit(1)