diff --git a/app/tests.py b/app/tests.py index d0a79da..265d59e 100644 --- a/app/tests.py +++ b/app/tests.py @@ -76,7 +76,35 @@ def test_annotation(self): self.assertEqual(r.status_code, 200) d = json.loads(r.data) self.assertIn('id', d) + id_resp = d['id'] + r = self.app.get('/view/1/annotations') d = json.loads(r.data) anns = d['data']['2'] self.assertEqual(len(anns), 1) + ann = anns[0] + id_retr = ann['id'] + self.assertEqual(ann['height'], 6) + self.assertEqual(id_retr, id_resp) + + data = { 'doc': 1 + , 'page': 2 + , 'posx': 3 + , 'posy': 4 + , 'width': 5 + , 'height': 60 + , 'value': 'Oh oh' + } + r = self.app.put('/annotation/{}'.format(id_retr), data=data) + r = self.app.get('/view/1/annotations') + d = json.loads(r.data) + anns = d['data']['2'] + self.assertEqual(len(anns), 1) + ann = anns[0] + self.assertEqual(ann['height'], 60) + + r = self.app.delete('/annotation/{}'.format(id_retr)) + self.assertEqual(r.status_code, 200) + r = self.app.get('/view/1/annotations') + d = json.loads(r.data) + self.assertNotIn('2', d['data'])