goa/test/unittest_metadata.py
changeset 0 b97547f5f1fa
child 1398 5fe84a5f7035
--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/goa/test/unittest_metadata.py	Wed Nov 05 15:52:50 2008 +0100
@@ -0,0 +1,106 @@
+from cubicweb.goa.testlib import *
+
+import time
+from mx.DateTime import DateTimeType
+from datetime import datetime
+from cubicweb.goa import db
+
+from google.appengine.api import datastore
+
+class Article(db.Model):        
+    content = db.TextProperty()
+    synopsis = db.StringProperty(default='hello')
+
+class Blog(db.Model):
+    diem = db.DateProperty(required=True, auto_now_add=True)
+    title = db.StringProperty(required=True)
+    content = db.TextProperty()
+    talks_about = db.ReferenceProperty(Article) 
+    cites = db.SelfReferenceProperty() 
+
+  
+class MetaDataTC(GAEBasedTC):
+    MODEL_CLASSES = (Article, Blog)
+    
+    def setUp(self):
+        GAEBasedTC.setUp(self)
+        self.req = self.request()
+        self.a = self.add_entity('Article')
+        self.p = self.add_entity('EProperty', pkey=u'ui.language', value=u'en')
+        self.session.commit()
+        
+    def _test_timestamp(self, entity, attr, sleep=0.1):
+        timestamp = getattr(entity, attr)
+        self.failUnless(timestamp)
+        self.assertIsInstance(timestamp, DateTimeType)
+        self.assertIsInstance(entity.to_gae_model()['s_'+attr], datetime)
+        time.sleep(sleep)
+        if entity.id == 'Article':
+            entity.set_attributes(content=u'zou')
+        else:
+            entity.set_attributes(value=u'en')
+        self.session.commit()
+        return timestamp
+    
+    def test_creation_date_dbmodel(self):
+        cdate = self._test_timestamp(self.a, 'creation_date')
+        self.assertEquals(cdate, self.a.creation_date)
+        
+    def test_creation_date_yams(self):
+        cdate = self._test_timestamp(self.p, 'creation_date')
+        self.assertEquals(cdate, self.p.creation_date)
+        
+    def test_modification_date_dbmodel(self):
+        mdate = self._test_timestamp(self.a, 'modification_date', sleep=1)
+        a = self.execute('Any X WHERE X eid %(x)s', {'x': self.a.eid}, 'x').get_entity(0, 0)
+        self.failUnless(mdate < a.modification_date, (mdate, a.modification_date))
+        
+    def test_modification_date_yams(self):
+        mdate = self._test_timestamp(self.p, 'modification_date', sleep=1)
+        p = self.execute('Any X WHERE X eid %(x)s', {'x': self.p.eid}, 'x').get_entity(0, 0)
+        self.failUnless(mdate < p.modification_date, (mdate, p.modification_date))
+
+    def _test_owned_by(self, entity):
+        self.assertEquals(len(entity.owned_by), 1)
+        owner = entity.owned_by[0]
+        self.assertIsInstance(owner, db.Model)
+        dbmodel = entity.to_gae_model()
+        self.assertEquals(len(dbmodel['s_owned_by']), 1)
+        self.assertIsInstance(dbmodel['s_owned_by'][0], datastore.Key)
+        
+    def test_owned_by_dbmodel(self):
+        self._test_owned_by(self.a)
+        
+    def test_owned_by_yams(self):
+        self._test_owned_by(self.p)
+
+    def _test_created_by(self, entity):
+        self.assertEquals(len(entity.created_by), 1)
+        creator = entity.created_by[0]
+        self.assertIsInstance(creator, db.Model)
+        self.assertIsInstance(entity.to_gae_model()['s_created_by'], datastore.Key)
+        
+    def test_created_by_dbmodel(self):
+        self._test_created_by(self.a)
+        
+    def test_created_by_dbmodel(self):
+        self._test_created_by(self.p)
+        
+    def test_user_owns_dbmodel(self):
+        self.failUnless(self.req.user.owns(self.a.eid))
+        
+    def test_user_owns_yams(self):
+        self.failUnless(self.req.user.owns(self.p.eid))
+
+    def test_is_relation(self):
+        en = self.execute('Any EN WHERE E name EN, X is E, X eid %(x)s', {'x': self.a.eid}, 'x')[0][0]
+        self.assertEquals(en, 'Article')
+        en = self.execute('Any EN WHERE E name EN, X is E, X eid %(x)s', {'x': self.p.eid}, 'x')[0][0]
+        self.assertEquals(en, 'EProperty') 
+        en = self.execute('Any EN WHERE E name EN, X is E, X eid %(x)s', {'x': self.req.user.eid}, 'x')[0][0]
+        self.assertEquals(en, 'EUser')
+
+        
+if __name__ == '__main__':
+    from logilab.common.testlib import unittest_main
+    unittest_main()