def setUp(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = log_writer = mock.Mock() gethostname.return_value = "myhost" l_out = "2012-11-06 17:13:55 [12019] [INFO] Starting gunicorn 0.15.0\n" l_err = "2012-11-06 17:13:55 [12019] [ERROR] Error starting gunicorn\n" self.data = {} self.data["stderr"] = {"pid": 12018, "data": l_err, "name": "stderr"} self.data["stdout"] = {"pid": 12018, "data": l_out, "name": "stdout"} self.stream = Stream(watcher_name="mywatcher") TsuruLogWriter.assert_called_with(mock.ANY, self.stream.queue, None, None, 'host2', '514', 'LOCAL0', 'udp', 'appname1') log_writer.start.assert_called_once()
def test_envs_may_be_overriden(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={"TSURU_APPNAME": "mynewappname"}) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = stream._load_envs() self.assertEqual(appname, "mynewappname") self.assertEqual(host, "host1") self.assertEqual(token, "secret123") self.assertEqual(syslog_server, "host2") self.assertEqual(syslog_port, "514") self.assertEqual(syslog_facility, "LOCAL0") self.assertEqual(syslog_socket, "udp")
def test_envs_may_be_overriden(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={"TSURU_APPNAME": "mynewappname"}) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = stream._load_envs() self.assertEqual(appname, "mynewappname") self.assertEqual(host, "host1") self.assertEqual(token, "secret123") self.assertEqual(syslog_server, "host2") self.assertEqual(syslog_port, "514") self.assertEqual(syslog_facility, "LOCAL0") self.assertEqual(syslog_socket, "udp")
def exec_with_envs(commands, with_shell=False, working_dir="/home/application/current", pipe_output=False, envs=None): global running_pipe if not envs: envs = {} app_envs = {} app_envs.update(os.environ) app_envs.update(envs) if not os.path.exists(working_dir): working_dir = "/" for command in commands: popen_output = None if pipe_output: popen_output = subprocess.PIPE pipe = subprocess.Popen(command, shell=with_shell, cwd=working_dir, env=app_envs, stdout=popen_output, stderr=popen_output) running_pipe = pipe if pipe_output: stdout = Stream(echo_output=sys.stdout, default_stream_name='stdout', watcher_name='unit-agent', envs=app_envs) stderr = Stream(echo_output=sys.stderr, default_stream_name='stderr', watcher_name='unit-agent', envs=app_envs) stdout_thread = Thread(target=process_output, args=(pipe.stdout, stdout)) stdout_thread.start() stderr_thread = Thread(target=process_output, args=(pipe.stderr, stderr)) stderr_thread.start() status = pipe.wait() running_pipe = None if pipe_output: stdout_thread.join() stderr_thread.join() if status != 0: sys.exit(status)
def test_should_slience_errors_when_envs_does_not_exist( self, TsuruLogWriter): try: stream = Stream() stream(self.data["stdout"]) except Exception as e: msg = "Should not fail when envs does not exist. " \ "Exception: {}".format(e) self.fail(msg)
def test_envs_with_rate_limit(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={ "LOG_RATE_LIMIT_WINDOW": "60", "LOG_RATE_LIMIT_COUNT": "1000", }) TsuruLogWriter.assert_called_with(mock.ANY, stream.queue, "60", "1000", 'host2', '514', 'LOCAL0', 'udp', 'appname1')
def test_timeout_is_configurable(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", timeout=10) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() url = "{0}/apps/{1}/log?source=watcher&unit=myhost".format( host, appname) expected_msg = "Starting gunicorn 0.15.0\n" entry = stream.queue.get(timeout=2) self.assertEqual(url, entry.url) self.assertEqual(10, entry.timeout) self.assertEqual([expected_msg], entry.messages)
def setUp(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = log_writer = mock.Mock() gethostname.return_value = "myhost" l_out = "2012-11-06 17:13:55 [12019] [INFO] Starting gunicorn 0.15.0\n" l_err = "2012-11-06 17:13:55 [12019] [ERROR] Error starting gunicorn\n" self.data = {} self.data["stderr"] = { "pid": 12018, "data": l_err, "name": "stderr" } self.data["stdout"] = { "pid": 12018, "data": l_out, "name": "stdout" } self.stream = Stream(watcher_name="mywatcher") TsuruLogWriter.assert_called_with(mock.ANY, self.stream.queue, None, None) log_writer.start.assert_called_once()
class StreamTestCase(unittest.TestCase): @mock.patch("os.environ", mocked_environ) @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def setUp(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = log_writer = mock.Mock() gethostname.return_value = "myhost" l_out = "2012-11-06 17:13:55 [12019] [INFO] Starting gunicorn 0.15.0\n" l_err = "2012-11-06 17:13:55 [12019] [ERROR] Error starting gunicorn\n" self.data = {} self.data["stderr"] = { "pid": 12018, "data": l_err, "name": "stderr" } self.data["stdout"] = { "pid": 12018, "data": l_out, "name": "stdout" } self.stream = Stream(watcher_name="mywatcher") TsuruLogWriter.assert_called_with(mock.ANY, self.stream.queue, None, None) log_writer.start.assert_called_once() def tearDown(self): self.stream.writer.stop() def test_should_have_the_close_method(self): self.assertTrue(hasattr(Stream, "close")) def test_should_send_done_message_on_close(self): self.stream.close() entry = self.stream.queue.get() self.assertEqual(entry, QUEUE_DONE_MESSAGE) def test_should_send_log_to_tsuru(self): self.stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() url = "{0}/apps/{1}/log?source=mywatcher&unit=myhost".format(host, appname) expected_msg = "Starting gunicorn 0.15.0\n" entry = self.stream.queue.get() self.assertEqual(url, entry.url) self.assertEqual(2, entry.timeout) self.assertEqual([expected_msg], entry.messages) @mock.patch("logging.getLogger") @mock.patch("logging.handlers.SysLogHandler") def test_should_send_log_to_syslog_as_info(self, s_handler, logger): self.stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() my_logger = logger(appname) log_handler = s_handler(address=(syslog_server, syslog_port), facility=syslog_facility, socktype=syslog_socket) expected_msg = "Starting gunicorn 0.15.0\n" my_logger.addHandler(log_handler) my_logger.info.assert_called_with(expected_msg) @mock.patch("logging.getLogger") @mock.patch("tsuru_unit_agent.syslog.SysLogHandler") def test_should_send_log_to_syslog_as_error(self, s_handler, logger): s_handler.return_value = syslog = mock.Mock() self.stream(self.data["stderr"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() my_logger = logger(appname) s_handler.assert_called_with(address=(syslog_server, int(syslog_port)), facility=syslog_facility, socktype=socket.SOCK_DGRAM) my_logger.addHandler.assert_called_with(syslog) my_logger.error.assert_called_with("Error starting gunicorn\n") def test_should_send_log_to_syslog_and_use_one_handler(self): self.stream(self.data["stderr"]) self.stream(self.data["stderr"]) self.stream(self.data["stderr"]) self.stream(self.data["stderr"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() my_logger = logging.getLogger(appname) self.assertEqual(len(my_logger.handlers), 1) @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_timeout_is_configurable(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", timeout=10) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() url = "{0}/apps/{1}/log?source=watcher&unit=myhost".format(host, appname) expected_msg = "Starting gunicorn 0.15.0\n" entry = stream.queue.get(timeout=2) self.assertEqual(url, entry.url) self.assertEqual(10, entry.timeout) self.assertEqual([expected_msg], entry.messages) @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_envs_may_be_overriden(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={"TSURU_APPNAME": "mynewappname"}) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = stream._load_envs() self.assertEqual(appname, "mynewappname") self.assertEqual(host, "host1") self.assertEqual(token, "secret123") self.assertEqual(syslog_server, "host2") self.assertEqual(syslog_port, "514") self.assertEqual(syslog_facility, "LOCAL0") self.assertEqual(syslog_socket, "udp") @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_envs_with_rate_limit(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={ "LOG_RATE_LIMIT_WINDOW": "60", "LOG_RATE_LIMIT_COUNT": "1000", }) TsuruLogWriter.assert_called_with(mock.ANY, stream.queue, "60", "1000") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") @mock.patch("os.environ", {}) def test_should_slience_errors_when_envs_does_not_exist(self, TsuruLogWriter): try: stream = Stream() stream(self.data["stdout"]) except Exception as e: msg = "Should not fail when envs does not exist. " \ "Exception: {}".format(e) self.fail(msg) def test_get_messagess_no_buffering(self): msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " \ "[INFO] Using worker: sync\n2012-11-06 18:30:10 [13890] " \ "[INFO] Booting worker with pid: 13890\n2012-11-06 18:30:10 " \ "[13890] [ERROR] Exception in worker process:\nTraceback " \ "(most recent call last):\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", ] messages = self.stream._get_messages(msg) self.assertEqual("", self.stream._buffer) self.assertEqual(expected, messages) def test_get_messagess_buffering(self): msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " \ "[INFO] Using worker: sync\n2012-11-06 18:30:10 [13890] " \ "[INFO] Booting worker with pid: 13890\n2012-11-06 18:30:10 " \ "[13890] [ERROR] Exception in worker process:\nTraceback " \ "(most recent call last):" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", ] messages = self.stream._get_messages(msg) self.assertEqual("Traceback (most recent call last):", self.stream._buffer) self.assertEqual(expected, messages) def test_get_messagess_buffered(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " msg2 = "[INFO] Using worker: sync\n2012-11-06 18:30:10 " msg3 = "[13890] [INFO] Booting worker with pid: 13890\n2012-11-06 " \ "18:30:10 [13890] [ERROR] Exception in worker process:\n" \ "Traceback (most recent call last):\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", ] self.assertEqual(expected[:1], self.stream._get_messages(msg1)) self.assertEqual(expected[1:2], self.stream._get_messages(msg2)) self.assertEqual(expected[2:], self.stream._get_messages(msg3)) def test_get_messagess_full_buffer(self): self.stream._max_buffer_size = 20 self.stream._buffer = 13 * "a" msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)" expected = [self.stream._buffer + "Listening at: http://127.0.0.1:8000 (13887)"] self.assertEqual(expected, self.stream._get_messages(msg)) def test_get_messages_buffered_multiple_times(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " msg2 = "[INFO] Using worker: sync\n2012-11-06 18:30:10 " msg3 = "[13890] [INFO] Booting worker with pid: 13890\n2012-11-06 " \ "18:30:10 [13890] [ERROR] Exception in worker process:\n" \ "Traceback (most recent call last):" msg4 = "\n2012-11-06 18:30:10 [13887] [INFO] Booting another worker "\ "with pid: 13891\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", "Booting another worker with pid: 13891\n", ] self.assertEqual(expected[:1], self.stream._get_messages(msg1)) self.assertEqual(expected[1:2], self.stream._get_messages(msg2)) self.assertEqual(expected[2:4], self.stream._get_messages(msg3)) self.assertEqual(expected[4:], self.stream._get_messages(msg4)) def test_get_messages_sequential_bufferings(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " msg2 = "http://127.0.0.1:8000 " msg3 = "(13887)\n" self.assertEqual([], self.stream._get_messages(msg1)) self.assertEqual([], self.stream._get_messages(msg2)) self.assertEqual(["Listening at: http://127.0.0.1:8000 (13887)\n"], self.stream._get_messages(msg3)) def test_default_max_buffer_size(self): self.assertEqual(10240, self.stream._max_buffer_size) @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_max_buffer_size_is_configurable(self, TsuruLogWriter): stream = Stream(max_buffer_size=500) self.assertEqual(500, stream._max_buffer_size)
def test_max_buffer_size_is_configurable(self, TsuruLogWriter): stream = Stream(max_buffer_size=500) self.assertEqual(500, stream._max_buffer_size)
class StreamTestCase(unittest.TestCase): @mock.patch("os.environ", mocked_environ) @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def setUp(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = log_writer = mock.Mock() gethostname.return_value = "myhost" l_out = "2012-11-06 17:13:55 [12019] [INFO] Starting gunicorn 0.15.0\n" l_err = "2012-11-06 17:13:55 [12019] [ERROR] Error starting gunicorn\n" self.data = {} self.data["stderr"] = {"pid": 12018, "data": l_err, "name": "stderr"} self.data["stdout"] = {"pid": 12018, "data": l_out, "name": "stdout"} self.stream = Stream(watcher_name="mywatcher") TsuruLogWriter.assert_called_with(mock.ANY, self.stream.queue, None, None, 'host2', '514', 'LOCAL0', 'udp', 'appname1') log_writer.start.assert_called_once() def tearDown(self): self.stream.writer.stop() def test_should_have_the_close_method(self): self.assertTrue(hasattr(Stream, "close")) def test_should_send_done_message_on_close(self): self.stream.close() entry = self.stream.queue.get() self.assertEqual(entry, QUEUE_DONE_MESSAGE) def test_should_send_log_to_tsuru(self): self.stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() url = "{0}/apps/{1}/log?source=mywatcher&unit=myhost".format( host, appname) expected_msg = "Starting gunicorn 0.15.0\n" entry = self.stream.queue.get() self.assertEqual(url, entry.url) self.assertEqual(2, entry.timeout) self.assertEqual([expected_msg], entry.messages) def test_should_send_log_to_syslog_as_info(self): self.stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() expected_msg = "Starting gunicorn 0.15.0\n" entry = self.stream.queue.get() self.assertEqual(entry.messages, [expected_msg]) self.assertEqual(entry.stream_name, "stdout") def test_should_send_log_to_syslog_as_error(self): self.stream(self.data["stderr"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() expected_msg = "Error starting gunicorn\n" entry = self.stream.queue.get() self.assertEqual(entry.messages, [expected_msg]) self.assertEqual(entry.stream_name, "stderr") @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_timeout_is_configurable(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", timeout=10) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = self.stream._load_envs() url = "{0}/apps/{1}/log?source=watcher&unit=myhost".format( host, appname) expected_msg = "Starting gunicorn 0.15.0\n" entry = stream.queue.get(timeout=2) self.assertEqual(url, entry.url) self.assertEqual(10, entry.timeout) self.assertEqual([expected_msg], entry.messages) @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_envs_may_be_overriden(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={"TSURU_APPNAME": "mynewappname"}) stream(self.data["stdout"]) (appname, host, token, syslog_server, syslog_port, syslog_facility, syslog_socket) = stream._load_envs() self.assertEqual(appname, "mynewappname") self.assertEqual(host, "host1") self.assertEqual(token, "secret123") self.assertEqual(syslog_server, "host2") self.assertEqual(syslog_port, "514") self.assertEqual(syslog_facility, "LOCAL0") self.assertEqual(syslog_socket, "udp") @mock.patch("tsuru_unit_agent.stream.gethostname") @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_envs_with_rate_limit(self, TsuruLogWriter, gethostname): TsuruLogWriter.return_value = mock.Mock() gethostname.return_value = "myhost" stream = Stream(watcher_name="watcher", envs={ "LOG_RATE_LIMIT_WINDOW": "60", "LOG_RATE_LIMIT_COUNT": "1000", }) TsuruLogWriter.assert_called_with(mock.ANY, stream.queue, "60", "1000", 'host2', '514', 'LOCAL0', 'udp', 'appname1') @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") @mock.patch("os.environ", {}) def test_should_slience_errors_when_envs_does_not_exist( self, TsuruLogWriter): try: stream = Stream() stream(self.data["stdout"]) except Exception as e: msg = "Should not fail when envs does not exist. " \ "Exception: {}".format(e) self.fail(msg) def test_get_messagess_no_buffering(self): msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " \ "[INFO] Using worker: sync\n2012-11-06 18:30:10 [13890] " \ "[INFO] Booting worker with pid: 13890\n2012-11-06 18:30:10 " \ "[13890] [ERROR] Exception in worker process:\nTraceback " \ "(most recent call last):\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", ] messages = self.stream._get_messages(msg) self.assertEqual("", self.stream._buffer) self.assertEqual(expected, messages) def test_get_messagess_buffering(self): msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " \ "[INFO] Using worker: sync\n2012-11-06 18:30:10 [13890] " \ "[INFO] Booting worker with pid: 13890\n2012-11-06 18:30:10 " \ "[13890] [ERROR] Exception in worker process:\nTraceback " \ "(most recent call last):" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", ] messages = self.stream._get_messages(msg) self.assertEqual("Traceback (most recent call last):", self.stream._buffer) self.assertEqual(expected, messages) def test_get_messagess_buffered(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " msg2 = "[INFO] Using worker: sync\n2012-11-06 18:30:10 " msg3 = "[13890] [INFO] Booting worker with pid: 13890\n2012-11-06 " \ "18:30:10 [13890] [ERROR] Exception in worker process:\n" \ "Traceback (most recent call last):\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", ] self.assertEqual(expected[:1], self.stream._get_messages(msg1)) self.assertEqual(expected[1:2], self.stream._get_messages(msg2)) self.assertEqual(expected[2:], self.stream._get_messages(msg3)) def test_get_messagess_full_buffer(self): self.stream._max_buffer_size = 20 self.stream._buffer = 13 * "a" msg = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)" expected = [ self.stream._buffer + "Listening at: http://127.0.0.1:8000 (13887)" ] self.assertEqual(expected, self.stream._get_messages(msg)) def test_get_messages_buffered_multiple_times(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " \ "http://127.0.0.1:8000 (13887)\n2012-11-06 18:30:10 [13887] " msg2 = "[INFO] Using worker: sync\n2012-11-06 18:30:10 " msg3 = "[13890] [INFO] Booting worker with pid: 13890\n2012-11-06 " \ "18:30:10 [13890] [ERROR] Exception in worker process:\n" \ "Traceback (most recent call last):" msg4 = "\n2012-11-06 18:30:10 [13887] [INFO] Booting another worker "\ "with pid: 13891\n" expected = [ "Listening at: http://127.0.0.1:8000 (13887)\n", "Using worker: sync\n", "Booting worker with pid: 13890\n", "Exception in worker process:\n", "Traceback (most recent call last):\n", "Booting another worker with pid: 13891\n", ] self.assertEqual(expected[:1], self.stream._get_messages(msg1)) self.assertEqual(expected[1:2], self.stream._get_messages(msg2)) self.assertEqual(expected[2:4], self.stream._get_messages(msg3)) self.assertEqual(expected[4:], self.stream._get_messages(msg4)) def test_get_messages_sequential_bufferings(self): msg1 = "2012-11-06 18:30:10 [13887] [INFO] Listening at: " msg2 = "http://127.0.0.1:8000 " msg3 = "(13887)\n" self.assertEqual([], self.stream._get_messages(msg1)) self.assertEqual([], self.stream._get_messages(msg2)) self.assertEqual(["Listening at: http://127.0.0.1:8000 (13887)\n"], self.stream._get_messages(msg3)) def test_default_max_buffer_size(self): self.assertEqual(10240, self.stream._max_buffer_size) @mock.patch("tsuru_unit_agent.stream.TsuruLogWriter") def test_max_buffer_size_is_configurable(self, TsuruLogWriter): stream = Stream(max_buffer_size=500) self.assertEqual(500, stream._max_buffer_size)