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