From b4980f17369182ef6cf2213a08d94b6f4d579c8f Mon Sep 17 00:00:00 2001
From: Min RK <benjaminrk@gmail.com>
Date: Fri, 6 Feb 2015 15:42:50 -0800
Subject: [PATCH] test inconsistent state if Spawner.start fails

---
 jupyterhub/orm.py            |  2 +-
 jupyterhub/tests/test_orm.py | 21 +++++++++++++++++++++
 2 files changed, 22 insertions(+), 1 deletion(-)

diff --git a/jupyterhub/orm.py b/jupyterhub/orm.py
index 959bf052..a43f38c1 100644
--- a/jupyterhub/orm.py
+++ b/jupyterhub/orm.py
@@ -322,9 +322,9 @@ class User(Base):
         spawner.api_token = api_token
         
         self.spawn_pending = True
-        f = spawner.start()
         # wait for spawner.start to return
         try:
+            f = spawner.start()
             yield gen.with_timeout(timedelta(seconds=spawner.start_timeout), f)
         except Exception as e:
             if isinstance(e, gen.TimeoutError):
diff --git a/jupyterhub/tests/test_orm.py b/jupyterhub/tests/test_orm.py
index bb789baa..544a9cf8 100644
--- a/jupyterhub/tests/test_orm.py
+++ b/jupyterhub/tests/test_orm.py
@@ -3,7 +3,11 @@
 # Copyright (c) Jupyter Development Team.
 # Distributed under the terms of the Modified BSD License.
 
+import pytest
+from tornado import gen
+
 from .. import orm
+from .mocking import MockSpawner
 
 
 def test_server(db):
@@ -82,3 +86,20 @@ def test_tokens(db):
     assert found.match(token)
     found = orm.APIToken.find(db, 'something else')
     assert found is None
+
+
+def test_spawn_fails(db, io_loop):
+    user = orm.User(name='aeofel')
+    db.add(user)
+    db.commit()
+    
+    class BadSpawner(MockSpawner):
+        @gen.coroutine
+        def start(self):
+            raise RuntimeError("Split the party")
+    
+    with pytest.raises(Exception) as exc:
+        io_loop.run_sync(lambda : user.spawn(BadSpawner))
+    assert user.server is None
+    assert not user.running
+