git://git.onelab.eu
/
nepi.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Added unit tests for linux application
[nepi.git]
/
test
/
execution
/
ec.py
diff --git
a/test/execution/ec.py
b/test/execution/ec.py
index
fba3cd5
..
9322540
100755
(executable)
--- a/
test/execution/ec.py
+++ b/
test/execution/ec.py
@@
-1,6
+1,6
@@
#!/usr/bin/env python
#!/usr/bin/env python
-from neco.execution.ec import ExperimentController
+from neco.execution.ec import ExperimentController
, ECState
from neco.execution.scheduler import TaskStatus
import datetime
from neco.execution.scheduler import TaskStatus
import datetime
@@
-13,21
+13,19
@@
class ExecuteControllersTestCase(unittest.TestCase):
return 'hola!'
ec = ExperimentController()
return 'hola!'
ec = ExperimentController()
+
+ tid = ec.schedule("0s", myfunc, track=True)
- try:
- tid = ec.schedule("0s", myfunc, track=True)
-
- while True:
- task = ec.get_task(tid)
- if task.status != TaskStatus.NEW:
- break
+ while True:
+ task = ec.get_task(tid)
+ if task.status != TaskStatus.NEW:
+ break
-
time.sleep(1)
+ time.sleep(1)
-
self.assertEquals('hola!', task.result)
+ self.assertEquals('hola!', task.result)
- finally:
- ec.shutdown()
+ ec.shutdown()
def test_schedule_date(self):
def get_time():
def test_schedule_date(self):
def get_time():
@@
-35,25
+33,36
@@
class ExecuteControllersTestCase(unittest.TestCase):
ec = ExperimentController()
ec = ExperimentController()
- try:
- schedule_time = datetime.datetime.now()
-
- tid = ec.schedule("4s", get_time, track=True)
+ schedule_time = datetime.datetime.now()
+
+ tid = ec.schedule("4s", get_time, track=True)
+
+ while True:
+ task = ec.get_task(tid)
+ if task.status != TaskStatus.NEW:
+ break
+
+ time.sleep(1)
- while True:
- task = ec.get_task(tid)
- if task.status != TaskStatus.NEW:
- break
+ execution_time = task.result
+ delta = execution_time - schedule_time
+ self.assertTrue(delta > datetime.timedelta(seconds=4))
+ self.assertTrue(delta < datetime.timedelta(seconds=5))
-
time.sleep(1
)
+
ec.shutdown(
)
- execution_time = task.result
- delta = execution_time - schedule_time
- self.assertTrue(delta > datetime.timedelta(seconds=4))
- self.assertTrue(delta < datetime.timedelta(seconds=5))
+ def test_schedule_exception(self):
+ def raise_error():
+ raise RuntimeError, "the error"
- finally:
- ec.shutdown()
+ ec = ExperimentController()
+ ec.schedule("2s", raise_error)
+
+ while ec.ecstate not in [ECState.FAILED, ECState.TERMINATED]:
+ time.sleep(1)
+
+ self.assertEquals(ec.ecstate, ECState.FAILED)
+ ec.shutdown()
if __name__ == '__main__':
if __name__ == '__main__':