Show More
@@ -126,7 +126,8 b' class TestClient(ClusterTestCase):' | |||||
126 | self.assertEquals(sorted(qs.keys()), ['completed', 'queue', 'tasks']) |
|
126 | self.assertEquals(sorted(qs.keys()), ['completed', 'queue', 'tasks']) | |
127 | allqs = self.client.queue_status() |
|
127 | allqs = self.client.queue_status() | |
128 | self.assertTrue(isinstance(allqs, dict)) |
|
128 | self.assertTrue(isinstance(allqs, dict)) | |
129 | self.assertEquals(sorted(allqs.keys()), self.client.ids) |
|
129 | self.assertEquals(sorted(allqs.keys()), sorted(self.client.ids + ['unassigned'])) | |
|
130 | unassigned = allqs.pop('unassigned') | |||
130 | for eid,qs in allqs.items(): |
|
131 | for eid,qs in allqs.items(): | |
131 | self.assertTrue(isinstance(qs, dict)) |
|
132 | self.assertTrue(isinstance(qs, dict)) | |
132 | self.assertEquals(sorted(qs.keys()), ['completed', 'queue', 'tasks']) |
|
133 | self.assertEquals(sorted(qs.keys()), ['completed', 'queue', 'tasks']) |
@@ -74,7 +74,7 b' class TestView(ClusterTestCase):' | |||||
74 | r = ar.get() |
|
74 | r = ar.get() | |
75 | self.assertEquals(r, nengines*[data]) |
|
75 | self.assertEquals(r, nengines*[data]) | |
76 | self.client[:].push(dict(a=10,b=20)) |
|
76 | self.client[:].push(dict(a=10,b=20)) | |
77 | r = self.client[:].pull(('a','b')) |
|
77 | r = self.client[:].pull(('a','b'), block=True) | |
78 | self.assertEquals(r, nengines*[[10,20]]) |
|
78 | self.assertEquals(r, nengines*[[10,20]]) | |
79 |
|
79 | |||
80 | def test_push_pull_function(self): |
|
80 | def test_push_pull_function(self): | |
@@ -83,10 +83,11 b' class TestView(ClusterTestCase):' | |||||
83 | return 2.0*x |
|
83 | return 2.0*x | |
84 |
|
84 | |||
85 | t = self.client.ids[-1] |
|
85 | t = self.client.ids[-1] | |
86 |
self.client[t] |
|
86 | v = self.client[t] | |
87 | push = self.client[t].push |
|
87 | v.block=True | |
88 | pull = self.client[t].pull |
|
88 | push = v.push | |
89 | execute = self.client[t].execute |
|
89 | pull = v.pull | |
|
90 | execute = v.execute | |||
90 | push({'testf':testf}) |
|
91 | push({'testf':testf}) | |
91 | r = pull('testf') |
|
92 | r = pull('testf') | |
92 | self.assertEqual(r(1.0), testf(1.0)) |
|
93 | self.assertEqual(r(1.0), testf(1.0)) |
General Comments 0
You need to be logged in to leave comments.
Login now