85 lines
3.4 KiB
Python
85 lines
3.4 KiB
Python
# Copyright 2014: Mirantis Inc.
|
|
# All Rights Reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import socket
|
|
import time
|
|
|
|
import mock
|
|
|
|
from opentsdbclient.socket import client
|
|
from opentsdbclient import tests
|
|
|
|
|
|
class SocketClientTest(tests.BaseTestCase):
|
|
def setUp(self):
|
|
super(SocketClientTest, self).setUp()
|
|
self.client = client.SocketOpenTSDBClient(opentsdb_hosts=[(self.host,
|
|
self.port)],
|
|
send_queue_max_size=1)
|
|
self.client.tsd = mock.MagicMock()
|
|
self.client.tsd.sendall = mock.MagicMock()
|
|
self.meters = [{'metric': 'bla1', 'timestamp': 12345,
|
|
'value': 123, 'tags': {'some_tag': 'foo'}},
|
|
{'metric': 'bla2', 'timestamp': 23456,
|
|
'value': 123, 'tags': {'some_tag': 'foo'}}]
|
|
|
|
def test_put_meter(self):
|
|
self.client.maintain_connection = mock.MagicMock()
|
|
|
|
self.client.put_meter(self.meters)
|
|
self.client.tsd.sendall.assert_called_once_with(
|
|
'put bla1 12345 123 some_tag=foo\n'
|
|
'put bla2 23456 123 some_tag=foo\n')
|
|
|
|
def test_send_data(self):
|
|
self.client.send_queue = self.meters
|
|
self.client.send_data()
|
|
self.client.tsd.sendall.assert_called_once_with(
|
|
'put bla1 12345 123 some_tag=foo\n'
|
|
'put bla2 23456 123 some_tag=foo\n')
|
|
|
|
def test_compose_line_from_meter(self):
|
|
res = self.client.compose_line_from_meter(self.meters[0])
|
|
self.assertEqual('bla1 12345 123 some_tag=foo', res)
|
|
|
|
@mock.patch.object(socket, 'getaddrinfo')
|
|
@mock.patch.object(socket, 'socket')
|
|
def test_maintain_connection(self, sock_mock, sock_addr_mock):
|
|
self.client.verify_connection = mock.MagicMock()
|
|
pop_list = [True, False]
|
|
self.client.verify_connection.side_effect = lambda: pop_list.pop()
|
|
sock_addr_mock.side_effect = lambda a, b, c, d, e: [(1, 2, 3, 4, 5)]
|
|
tsd = mock.MagicMock()
|
|
tsd.connect = mock.MagicMock()
|
|
sock_mock.side_effect = tsd
|
|
self.client.maintain_connection()
|
|
sock_addr_mock.assert_called_once_with(self.host, self.port,
|
|
socket.AF_UNSPEC,
|
|
socket.SOCK_STREAM, 0)
|
|
sock_mock.assert_called_once_with(1, 2, 3)
|
|
self.client.tsd.connect.assert_called_once_with(5)
|
|
|
|
def test_verify_connection_non_tsd(self):
|
|
self.client.tsd = None
|
|
self.assertFalse(self.client.verify_connection())
|
|
|
|
def test_verify_connection_tsd(self):
|
|
self.assertTrue(self.client.verify_connection())
|
|
|
|
def test_verify_connection_tsd_reconnect(self):
|
|
self.client.last_verify = time.time() - 3600
|
|
self.client.reconnect_interval = 1
|
|
self.assertFalse(self.client.verify_connection())
|