[test][experiment] Test to ensure proper output of experiment run

beat/cmdline/test/test_experiments.py

@@ -30,10 +30,11 @@
 
 import os
 import sys
+import logging
 import nose.tools
 
 from . import platform, disconnected, prefix, tmp_prefix, user, token, temp_cwd
-from .utils import index_experiment_dbs
+from .utils import index_experiment_dbs, MockLoggingHandler
 from ..common import Selector
 from ..scripts.beat import main
 from beat.core.test.utils import slow, cleanup, skipif
@@ -181,6 +182,24 @@ def test_delete_integers_addition_1_cache():
   nose.tools.eq_(call('caches', '--delete', obj, cache=tmp_prefix), 0)
 
 
+@slow
+@nose.tools.with_setup(setup=setup_experiments, teardown=cleanup)
+def test_run_integers_addition_1_twice():
+  log_handler = MockLoggingHandler(level='DEBUG')
+  logging.getLogger().addHandler(log_handler)
+  log_messages = log_handler.messages
+
+  obj = 'user/user/integers_addition/1/integers_addition'
+  nose.tools.eq_(call('run', obj, cache=tmp_prefix), 0)
+  info_len = len(log_messages['info'])
+  nose.tools.eq_(info_len, 5)
+  assert log_messages['info'][info_len - 1].startswith('  Results')
+  nose.tools.eq_(call('run', obj, cache=tmp_prefix), 0)
+  info_len = len(log_messages['info'])
+  nose.tools.eq_(info_len, 8)
+  assert log_messages['info'][info_len - 1].startswith('  Results')
+
+
 @slow
 @nose.tools.with_setup(setup=setup_experiments, teardown=cleanup)
 def test_run_double_triangle_1():