From 271b0da1c28fea17b84ae13e00606e18d1fbb2a4 Mon Sep 17 00:00:00 2001
From: helmutm <helmutm@fd906abe-77d9-0310-91a1-e0d9ade77398>
Date: Sat, 9 Jun 2007 16:47:26 +0000
Subject: [PATCH] Work in progress: set up basic agent functionality (with unit
 tests...)

git-svn-id: svn://svn.cy55.de/Zope3/src/loops/trunk@1784 fd906abe-77d9-0310-91a1-e0d9ade77398
---
 agent/tests.py | 24 +++++++++++++++++-------
 1 file changed, 17 insertions(+), 7 deletions(-)

diff --git a/agent/tests.py b/agent/tests.py
index a95bc42..fa43e7b 100755
--- a/agent/tests.py
+++ b/agent/tests.py
@@ -1,8 +1,11 @@
 # $Id$
 
-import unittest, doctest
+import unittest as standard_unittest
+import doctest
 import time
 from twisted.internet import reactor
+from twisted.internet.defer import Deferred
+from twisted.trial import unittest
 
 from loops.agent.core import Agent
 from loops.agent.schedule import Job
@@ -10,9 +13,10 @@ from loops.agent.schedule import Job
 
 class TestJob(Job):
 
-    def execute(self, **kw):
+    def execute(self, deferred, **kw):
         d = super(TestJob, self).execute(**kw)
-        print 'executing'
+        #print 'executing'
+        deferred.callback('Done')
         return d
 
 
@@ -22,15 +26,21 @@ class Test(unittest.TestCase):
     def setUp(self):
         self.agent = Agent()
 
+    def tearDown(self):
+        pass
+
     def testScheduling(self):
-        d = self.agent.scheduler.schedule(TestJob(), int(time.time())+1)
-        time.sleep(1)
+        d = Deferred()
+        job = TestJob()
+        job.params['deferred'] = d
+        w = self.agent.scheduler.schedule(job, int(time.time())+1)
+        return d
 
 
 def test_suite():
     flags = doctest.NORMALIZE_WHITESPACE | doctest.ELLIPSIS
-    return unittest.TestSuite((
-                unittest.makeSuite(Test),
+    return standard_unittest.TestSuite((
+                standard_unittest.makeSuite(Test),
                 doctest.DocFileSuite('README.txt', optionflags=flags),
             ))