test_export_views.py 文件源码

def test_csv_okay(self):
        """
            Export tags as CSV.
        """
        _sign_in(self.admin)

        url = reverse(
            'export:tags', kwargs={"project_id": self.project.pk})
        resp = self.client.post(url)
        self.assertEqual(200, resp.status_code)

        tags = list(read_csv_to_dict(resp.content, encoding='utf-16'))
        self.assertEqual(2, len(tags))

        tag1 = next(t for t in tags if long(t['id']) == self.tag1.pk)
        self.assertEqual(self.tag1.name, tag1['name'])
        self.assertEqual(str(self.tag1.description or ''), tag1['description'])
        self.assertEqual(str(self.tag1.image_url or ''), tag1['image_url'])
        self.assertEqual(
            timezone.make_naive(
                self.tag1.created, timezone.utc
            ).strftime("%Y-%m-%d %H:%M:%S+00:00"),
            datetime.datetime.strptime(tag1['created'], '%Y-%m-%d %H:%M:%S.%f+00:00').strftime("%Y-%m-%d %H:%M:%S+00:00"))
        self.assertEqual(
            timezone.make_naive(
                self.tag1.modified, timezone.utc
            ).strftime("%Y-%m-%d %H:%M:%S+00:00"),
            datetime.datetime.strptime(tag1['modified'], '%Y-%m-%d %H:%M:%S.%f+00:00').strftime("%Y-%m-%d %H:%M:%S+00:00"))
        self.assertEqual('2', tag1['instance_count'])
        self.assertEqual('', tag1['parent_id'])