Exemple #1
0
    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')
Exemple #2
0
    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')
Exemple #3
0
                "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]