Merge e1d9dd2 into 2b1e756

sacred/observers/sql.py

@@ -3,6 +3,7 @@
 
 import json
 from threading import Lock
+import warnings
 
 from sacred.commandline_options import CommandLineOption
 from sacred.observers.base import RunObserver
@@ -16,17 +17,21 @@
 class SqlObserver(RunObserver):
     @classmethod
     def create(cls, url, echo=False, priority=DEFAULT_SQL_PRIORITY):
+        warnings.warn('SqlObserver.create() is deprecated in favor of'
+                      'SqlObserver().')
+        return cls(url, echo=echo, priority=priority)
+
+    def __init__(self, url=None, echo=False, priority=DEFAULT_SQL_PRIORITY,
+                 mock_args=None):
         from sqlalchemy.orm import sessionmaker, scoped_session
         import sqlalchemy as sa
-        engine = sa.create_engine(url, echo=echo)
-        session_factory = sessionmaker(bind=engine)
         # make session thread-local to avoid problems with sqlite (see #275)
-        session = scoped_session(session_factory)
-        return cls(engine, session, priority)
-
-    def __init__(self, engine, session, priority=DEFAULT_SQL_PRIORITY):
-        self.engine = engine
-        self.session = session
+        if mock_args is None:
+            self.engine = sa.create_engine(url, echo=echo)
+            self.session = scoped_session(sessionmaker(bind=self.engine))
+        else:
+            self.engine = mock_args['engine']
+            self.session = mock_args['session']
         self.priority = priority
         self.run = None
         self.lock = Lock()

tests/test_observers/test_sql_observer.py

@@ -46,7 +46,7 @@ def session(engine):
 
 @pytest.fixture
 def sql_obs(session, engine):
-    return SqlObserver(engine, session)
+    return SqlObserver(mock_args={'engine': engine, 'session': session})
 
 
 @pytest.fixture
@@ -227,7 +227,8 @@ def test_fs_observer_resource_event(sql_obs, sample_run, session, tmpfile):
 
 
 def test_fs_observer_doesnt_duplicate_sources(sql_obs, sample_run, session, tmpfile):
-    sql_obs2 = SqlObserver(sql_obs.engine, session)
+    sql_obs2 = SqlObserver(mock_args={'engine': sql_obs.engine,
+                                      'session': session})
     sample_run['_id'] = None
     sample_run['ex_info']['sources'] = [[tmpfile.name, tmpfile.md5sum]]
 
@@ -239,7 +240,8 @@ def test_fs_observer_doesnt_duplicate_sources(sql_obs, sample_run, session, tmpf
 
 
 def test_fs_observer_doesnt_duplicate_resources(sql_obs, sample_run, session, tmpfile):
-    sql_obs2 = SqlObserver(sql_obs.engine, session)
+    sql_obs2 = SqlObserver(mock_args={'engine': sql_obs.engine,
+                                      'session': session})
     sample_run['_id'] = None
     sample_run['ex_info']['sources'] = [[tmpfile.name, tmpfile.md5sum]]
 
@@ -254,7 +256,8 @@ def test_fs_observer_doesnt_duplicate_resources(sql_obs, sample_run, session, tm
 
 
 def test_sql_observer_equality(sql_obs, engine, session):
-    sql_obs2 = SqlObserver(engine, session)
+    sql_obs2 = SqlObserver(mock_args={'engine': sql_obs.engine,
+                                      'session': session})
     assert sql_obs == sql_obs2
 
     assert not sql_obs != sql_obs2

tests/test_observers/test_sql_observer_not_installed.py

@@ -12,7 +12,7 @@ def ex():
 @pytest.mark.skipif(has_sqlalchemy, reason='We are testing the import error.')
 def test_importerror_sql(ex):
     with pytest.raises(ImportError):
-        ex.observers.append(SqlObserver.create('some_uri'))
+        ex.observers.append(SqlObserver('some_uri'))
 
         @ex.config
         def cfg():