Robot Framework + Selenium2Library自动化测试系列(4) - 循环介绍，登录改进：数据集

虽然已经把函数写好，但现在我们还是只有一组用户名和密码。测试需要多组数据，需要一个数据集，那应该把数据写到哪儿呢？对，没错，万能的Excel。

我新建了一个叫TCLOG-1.xls的文件，把tab起名为“login”，创建了两列，一列叫EmpCode，另一列叫Password，然后从第二行开始输入数据。一个简单的数据集就创建好了：

然后把文件放到项目目录下。以我为例，我在项目根目录下创建了个叫Test_Data和Login的文件夹，把文件放进去。路径就是：D:\sourcecode\xiaocesuo.robotframework.selenium2library\Test_Data\Login\TCLOG-1.xls。现在我们需要知道读取文件的关键字就可以了。之前说过，RF是基于Python的，所有的关键字背后都是一个一个python方法，这些方法都存在类库里。为了给大家顺便演示一下，我就借着这个例子自己动手写了一个读取文件的小类库。大家可以直接拿去用，下载源代码里面都有，名称叫DataDriven.py：

这段小程序里有三个方法，分别是read_csv_file，read_xls_file，xls_to_csv。前两个分别用于读格式为.csv和.xls的Excel文件，最后一个是从.xls转换成.csv。这几个方法名称中间用下划线隔开，它们会自动被Robot Framework识别成为新的类库关键字，而且新的关键字会把下划线转换成空格，所以三个新关键字分别为“Read Csv File”，“Read Xls File”，“Xls to Csv”。显然，咱们的例子需要使用第二种。通过程序可知Read Xls File接收filename和worksheet两个参数。前者代表文件名，后者是tab名。返回值是xls_data，代表数据集。在项目下创建一个Lib文件夹把它放里边，然后在MyFirstTestCase中点击Library加载：

你看，它就跟Selenium2Library的作用是一样的，提供了一个类库并且包含了三个方法。回到TCLOG-1中修改如下：

Read Xls File也变成了浅蓝色，跟Selenium2Library关键字显示得一样，证明合法了。所以现在可以总结一下：自己写的类库方法(Read Xls File)会变成浅蓝色，而自己写的一般方法(Login to hysystem)会显示成深蓝色。文件名参数位置写的是文件的路径。tab里写的是login。需要注意的是，“$”在Python中代表字段或变量，而“@”则代表集合。有人可能问为什么返回值是集合呢？因为数据在@{data}里是这样存的：

@{data}={第一行数据，第二行数据，第三行数据}，当然咱们这个例子最简单，只有一行，@{data}={第一行数据}。而每一行数据也是一个数组：@一行数据 ={EmpCode，Password}。所以整个数据集是用一个二维数组来存的。有些人可能理解起来有点困难，不要着急。看完咱们的例子就懂了。

我把程序改成了这个样子：

因为Excel文件里有很多行数据，我用了一个for循环语句来一行一行读。程序会一行一行读取数据，直到最后一行为止。每次读出来一行数据就赋给${row}这个变量，所以每次${row}都在不断更新。因为TCLOG-1.xls里只有一行，所以读完一行程序就停了，刚才说了，@一行数据 ={EmpCode，Password}，所以这里${row}也被看作是一个数组，此时里边就两个值，{“1001”, “123”}。访问数组值的方式是${row[“UserName”]}和${row[“Password”]}，所以第三行的意思就是把”1001”和”123”分别作为参数值传到函数中去。

跑case，测试通过。这篇文章的源代码在[Test3](https://github.com/cslm/cslm.robotframework/tree/master/Test3)中。