使用unittest django测试存储在内存中的csv文件

Question

我在尝试写入csv文件时遇到错误。这是我的两个脚本：

Tasks.py

def get_data():

          customer = Customer.objects.all()
          csvfile = StringIO.StringIO()
          csvwriter = csv.writer(csvfile)

          for i in customers:
              csvwriter.writerow([str(i.phone_number)])

          return csvfile

在tests.py中：

    with open(get_data().getvalue(), 'rb') as myfile:
        data = [row for row in csv.reader(myfile.read().splitlines())]
        print data

这是错误：

IOError: [Errno 2] No such file or directory: '999999999\r\n'

凡＆＃39; 999999999＆＃39;应该写在文件中。

我该如何解决这个问题？

Answer 1

您的get_data()函数不会返回文件，而是返回StringIO个对象。你不需要open()它，你可以这样读它：

myfile = get_data().getvalue()
data = [row for row in csv.reader(myfile.splitlines())]
print data