예제 #1
0
    def _test_loop(self):
        class ConnectionStub:
            exits = False
            def clean_up(self):
                self.exits = True
        connection_stub = ConnectionStub()
        properties = {DEVICE_GROUP : "test"}
        task_broker = TaskBroker(connection_stub, properties)
        
        #Case where no stop file and keep_looping_true
        class ChannelStub:
            wait_called = False
            def wait(self): 
                self.wait_called = True
                task_broker._keep_looping = False
        channel_stub = ChannelStub()
        task_broker._connection.channel = channel_stub
        task_broker._loop()
        self.assertTrue(channel_stub.wait_called)

        #Case where keep_looping False
        task_broker._keep_looping = False
        task_broker._loop()
        self.assertTrue(connection_stub.exits)

        #Case where stop_file_exists
        connection_stub.exits = False
        task_broker._keep_looping = True
        def _stop_file_exists():
            return True
        task_broker._stop_file_exists = _stop_file_exists
        task_broker._loop()
        self.assertTrue(connection_stub.exits)
예제 #2
0
    def _test_loop(self):
        class ConnectionStub:
            exits = False

            def clean_up(self):
                self.exits = True

        connection_stub = ConnectionStub()
        properties = {DEVICE_GROUP: "test"}
        task_broker = TaskBroker(connection_stub, properties)

        #Case where no stop file and keep_looping_true
        class ChannelStub:
            wait_called = False

            def wait(self):
                self.wait_called = True
                task_broker._keep_looping = False

        channel_stub = ChannelStub()
        task_broker._connection.channel = channel_stub
        task_broker._loop()
        self.assertTrue(channel_stub.wait_called)

        #Case where keep_looping False
        task_broker._keep_looping = False
        task_broker._loop()
        self.assertTrue(connection_stub.exits)

        #Case where stop_file_exists
        connection_stub.exits = False
        task_broker._keep_looping = True

        def _stop_file_exists():
            return True

        task_broker._stop_file_exists = _stop_file_exists
        task_broker._loop()
        self.assertTrue(connection_stub.exits)