def test_requiring_work(self):
# check if return all the streams with pending material statusese.
self.assertEquals(len(Stream.requiring_work()), 1)
self.assertEquals(Stream.requiring_work()[0], self.stream)
[ms.clone_for(self.other_stream) for ms in self.stream.materials.all()]
self.assertEquals(len(Stream.requiring_work()), 2)
self.assertEquals(Stream.requiring_work()[0], self.stream)
def wrap(managers, ids):
self.flag = True
l = Stream.requiring_work().values_list('id', flat=True)
self.assertEquals(len(ids), len(l))
for i in ids:
self.assertIn(i, l)
return original_map(managers, ids)
def step(self):
ids = Stream.requiring_work().values_list('id', flat=True)
# here the map function should not return any value.
self.managers.map(manager_job, ids)
connection.close()
def step(self):
ids = Stream.requiring_work().values_list('id', flat=True)
# here the map function should not return any value.
self.managers.map(manager_job, ids)
connection.close()