tests: Add a test for the unknown status

Author: rkdarst

batchspawner/tests/test_spawners.py

@@ -28,6 +28,7 @@ class BatchDummy(BatchSpawnerRegexStates):
     state_pending_re = Unicode('PEND')
     state_running_re = Unicode('RUN')
     state_exechost_re = Unicode('RUN (.*)$')
+    state_unknown_re = Unicode('UNKNOWN')
 
     cmd_expectlist = None
     out_expectlist = None
@@ -160,6 +161,20 @@ def test_poll_fails(db, io_loop):
     assert spawner.job_id == ''
     assert spawner.job_status == ''
 
+def test_unknown_status(db, io_loop):
+    """Polling returns an unknown status"""
+    spawner = new_spawner(db=db)
+    assert spawner.get_state() == {}
+    # The start is successful:
+    io_loop.run_sync(spawner.start, timeout=30)
+    spawner.batch_query_cmd = 'echo UNKNOWN'
+    # This poll should not fail:
+    io_loop.run_sync(spawner.poll, timeout=30)
+    status = io_loop.run_sync(spawner.query_job_status, timeout=30)
+    assert status == JobStatus.UNKNOWN
+    assert spawner.job_id == '12345'
+    assert spawner.job_status != ''
+
 
 def test_templates(db, io_loop):
     """Test templates in the run_command commands"""