def setUp(self, gethostname_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) self.logstash = LogstashBackend() self.logstash.gauge = Mock()
def test_gauge(self, gethostname_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) logstash = LogstashBackend() logstash.client = Mock() logstash.gauge("key", 1) dimensions = {'app': 'appname', 'host': 'somehost', 'value': 1.0} logstash.client.count.assert_called_with('key', dimensions=dimensions)
def test_logstash_environs(self, gethostname_mock, measure_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) LogstashBackend() measure_mock.assert_called_with('tsuru', ('localhost', 1984)) envs = { "TSURU_APPNAME": "appname", "LOGSTASH_HOST": "logstashhost", "LOGSTASH_PORT": "123", "LOGSTASH_CLIENT": "clientstash", } os.environ.update(envs) LogstashBackend() measure_mock.assert_called_with('clientstash', ('logstashhost', 123))
class TestLogstash(TestCase): @patch("socket.gethostname") def setUp(self, gethostname_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) self.logstash = LogstashBackend() self.logstash.gauge = Mock() @patch("measures.Measure") @patch("socket.gethostname") def test_logstash_environs(self, gethostname_mock, measure_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) LogstashBackend() measure_mock.assert_called_with('tsuru', ('localhost', 1984)) envs = { "TSURU_APPNAME": "appname", "LOGSTASH_HOST": "logstashhost", "LOGSTASH_PORT": "123", "LOGSTASH_CLIENT": "clientstash", } os.environ.update(envs) LogstashBackend() measure_mock.assert_called_with('clientstash', ('logstashhost', 123)) def test_init(self): self.assertEqual(self.logstash.app_name, "appname") self.assertEqual(self.logstash.host_name, "somehost") def test_mem_max(self): self.logstash.mem_max("name", 123) self.logstash.gauge.assert_called_with('mem_max', 123) def test_cpu_max(self): self.logstash.cpu_max("name", 123) self.logstash.gauge.assert_called_with('cpu_max', 123) def test_mem_sum(self): self.logstash.mem_sum("name", 123) self.logstash.gauge.assert_called_with('mem_sum', 123) def test_cpu_sum(self): self.logstash.cpu_sum("name", 123) self.logstash.gauge.assert_called_with('cpu_sum', 123) def test_mem_pct_sum(self): self.logstash.mem_pct_sum("name", 123) self.logstash.gauge.assert_called_with('mem_pct_sum', 123) def test_mem_pct_max(self): self.logstash.mem_pct_max("name", 123) self.logstash.gauge.assert_called_with('mem_pct_max', 123) def test_disk_usage(self): self.logstash.disk_usage(123) self.logstash.gauge.assert_called_with('disk_usage', 123) def test_net_sent(self): self.logstash.net_sent(123) self.logstash.gauge.assert_called_with('net_sent', 123) def test_net_recv(self): self.logstash.net_recv(123) self.logstash.gauge.assert_called_with('net_recv', 123) def test_net_connections(self): self.logstash.net_connections(123) self.logstash.gauge.assert_called_with('net_connections', 123) @patch("socket.gethostname") def test_gauge(self, gethostname_mock): gethostname_mock.return_value = "somehost" envs = { "TSURU_APPNAME": "appname", } os.environ.update(envs) logstash = LogstashBackend() logstash.client = Mock() logstash.gauge("key", 1) dimensions = {'app': 'appname', 'host': 'somehost', 'value': 1.0} logstash.client.count.assert_called_with('key', dimensions=dimensions)