test annotations a bit more

cloverfeed · Oct 20, 2014 · 7e45f47 · 7e45f47
1 parent bef5f14
commit 7e45f47
Showing 1 changed file with 28 additions and 0 deletions.
diff --git 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'])