def test_task_queue_name(self): """Task queue name is correct""" k = Kuyruk() def f(): return t = k.task()(f) self.assertEqual(t._queue_for_host(None), 'kuyruk') self.assertEqual(t._queue_for_host('foo'), 'kuyruk.foo')
def test_task_queue_name(self): """Task queue name is correct""" k = Kuyruk() def f(): return t = k.task(f) self.assertEqual(t._queue_for_host(None), 'kuyruk') self.assertEqual(t._queue_for_host('foo'), 'kuyruk.foo')
"ENV variable KUYRUK='%s', assuming you will be sending commands", mytype) return fn else: logger.debug("ENV variable KUYRUK='%s', assuming a worker setup", mytype) return function(fn) return decorate kuyruk = Kuyruk(config) # taskify is our own conditional decorator @taskify(kuyruk.task(socket.gethostname())) def sendfile(filename, contents): """ copy a file """ f = open(filename, "w") result = f.write(contents) f.close() logger.info("> SENDILE: '%s' received", filename) return result # If run from the command line then SEND the task if __name__ == "__main__": filename = sys.argv[1]