NLeSC / eEcology-Annotation-WS

        self.assertEqual(response, expected)

    def test_upload(self):

        self.request.matchdict['table'] = 'mytable'

        self.request.params['id'] = '355'

        self.request.params['start'] = '2010-06-27T00:00:00+00:00'

        self.request.params['end'] = '2010-06-29T00:00:00+00:00'

        self.config.add_route('annotations.csv', '/uploads/{table}/annotations.csv')

        views = UploadViews(self.request)

        rows = [{

                 'id': 3,

                 'label': 'flying',

                 'color': 'rgb(0,0,255)',

                 }]

        self.cursor.fetchall.return_value = rows

        response = views.upload()

        expected_response = {'annotations_url': '/uploads/mytable/annotations.csv',

                             'classes': '[{"color": "rgb(0,0,255)", "id": 3, "label": "flying"}]',

                             'end': '2010-06-29T00:00:00+00:00',

                             'start': '2010-06-27T00:00:00+00:00',

                             'tracker_id': 355,

                             }

        self.assertEqual(response, expected_response)

    def test_upload_withoutselection(self):

        self.request.matchdict['table'] = 'mytable'

        self.assertIsInstance(response, HTTPFound)

        self.assertEqual(response.location, expected_response.location)

    def test_annotations_as_csv(self):

        self.request.matchdict['table'] = 'mytable'

        self.request.params['id'] = '355'

        self.request.params['start'] = '2010-06-27T00:00:00+00:00'

        self.request.params['end'] = '2010-06-29T00:00:00+00:00'

        views = UploadViews(self.request)

        rows = [{

                 'device_info_serial': 355,

                 'date_time': datetime(2010, 6, 28, 0, 0, 0, 0, UTC),

                 'class_id': 3

                 }]

        self.cursor.fetchall.return_value = rows

        response = views.annotations_as_csv()

        expected_body = 'device_info_serial,date_time,class_id\n355,2010-06-28T00:00:00+00:00Z,3\n'

        self.assertEqual(response.body, expected_body)