X-Git-Url: https://git.decadent.org.uk/gitweb/?a=blobdiff_plain;f=tests%2Fdb_test.py;h=fc9ce893d61d4c25aef3f1adc269cb6c21c7dc43;hb=a6b1715c117b4f306a7d301ecc4492c67c6ef98b;hp=4f01f6d5d1583a89128a4b12c1fd41c9546fd88e;hpb=a6e908f0f30d02794e24f66a6d5f4bedc929456a;p=dak.git diff --git a/tests/db_test.py b/tests/db_test.py index 4f01f6d5..fc9ce893 100644 --- a/tests/db_test.py +++ b/tests/db_test.py @@ -3,7 +3,7 @@ from base_test import DakTestCase, fixture from daklib.config import Config from daklib.dbconn import DBConn -from sqlalchemy import create_engine, __version__ +from sqlalchemy import create_engine, func, __version__ from sqlalchemy.exc import SADeprecationWarning from sqlalchemy.schema import DDL @@ -72,6 +72,15 @@ class DBDakTestCase(DakTestCase): self.initialize() self.session = DBConn().session() + def now(self): + "returns the current time at the db server" + + # we fetch a fresh session each time to avoid caching + local_session = DBConn().session() + current_time = local_session.query(func.now()).scalar() + local_session.close() + return current_time + def classes_to_clean(self): """ The function classes_to_clean() returns a list of classes. All objects