From b35e9773198ef2d8b37c4ca223f08147db47de0b Mon Sep 17 00:00:00 2001
From: François Schmidts <francois.schmidts@gmail.com>
Date: Sat, 12 Dec 2015 21:14:28 +0100
Subject: moving the root of source code from / to /src/

---
 src/tests/base.py | 41 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)
 create mode 100644 src/tests/base.py

(limited to 'src/tests/base.py')

diff --git a/src/tests/base.py b/src/tests/base.py
new file mode 100644
index 00000000..d6f62583
--- /dev/null
+++ b/src/tests/base.py
@@ -0,0 +1,41 @@
+import os
+os.environ['PYAGG_TESTING'] = 'true'
+
+import unittest
+from bootstrap import db
+import runserver
+from tests.fixtures import populate_db, reset_db
+from werkzeug.exceptions import NotFound
+
+
+class BasePyaggTest(unittest.TestCase):
+    _contr_cls = None
+
+    def _get_from_contr(self, obj_id, user_id=None):
+        return self._contr_cls(user_id).get(id=obj_id).dump()
+
+    def _test_controller_rights(self, obj, user_id):
+        obj_id = obj['id']
+        self.assertEquals(obj, self._get_from_contr(obj_id))
+        self.assertEquals(obj, self._get_from_contr(obj_id, user_id))
+        # fetching non existent object
+        self.assertRaises(NotFound, self._get_from_contr, 99, user_id)
+        # fetching object with inexistent user
+        self.assertRaises(NotFound, self._get_from_contr, obj_id, 99)
+        # fetching object with wrong user
+        self.assertRaises(NotFound, self._get_from_contr, obj_id, user_id + 1)
+        self.assertRaises(NotFound, self._contr_cls().delete, 99)
+        self.assertRaises(NotFound, self._contr_cls(user_id).delete, 99)
+        self.assertEquals(obj['id'],
+                self._contr_cls(user_id).delete(obj_id).id)
+        self.assertRaises(NotFound, self._contr_cls(user_id).delete, obj_id)
+
+    def setUp(self):
+        populate_db(db)
+
+    def tearDown(self):
+        reset_db(db)
+
+
+if __name__ == '__main__':
+    unittest.main()
-- 
cgit