a6419e6cf3
Now it's possible to choose one of two client implementations - either via REST API or via direct socket usage. Socket client currently supports only meter posting. Change-Id: I6005cfa23ffbe4bdba15e62637150e8a29bdefc2
67 lines
2.5 KiB
Python
67 lines
2.5 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 json
|
|
|
|
import mock
|
|
import requests
|
|
|
|
from opentsdbclient.rest import client
|
|
from opentsdbclient import tests
|
|
|
|
|
|
class RESTClientTest(tests.BaseTestCase):
|
|
def setUp(self):
|
|
super(RESTClientTest, self).setUp()
|
|
self.client = client.RESTOpenTSDBClient(opentsdb_hosts=[(self.host,
|
|
self.port)])
|
|
|
|
@mock.patch.object(requests, 'get')
|
|
def test_get_statistics(self, get_mock):
|
|
self.client.get_statistics()
|
|
get_mock.assert_called_once_with('http://127.0.0.1:4242/api/stats')
|
|
|
|
@mock.patch.object(requests, 'post')
|
|
def test_put_meter(self, post_mock):
|
|
put_dict = {'metric': 'bla', 'timestamp': '0',
|
|
'value': 123, 'tags': {'some_tag': 'foo'}}
|
|
self.client.put_meter(put_dict)
|
|
post_mock.assert_called_once_with(
|
|
'http://127.0.0.1:4242/api/put?details', data=json.dumps(put_dict))
|
|
|
|
@mock.patch.object(requests, 'post')
|
|
def test_define_retention(self, post_mock):
|
|
self.client.define_retention('foo', 12)
|
|
post_mock.assert_called_once_with(
|
|
'http://127.0.0.1:4242/api/uid/tsmeta?tsuid=foo',
|
|
data='{"tsuid": "foo", "retention": 12}')
|
|
|
|
@mock.patch.object(requests, 'get')
|
|
def test_get_aggregators(self, get_mock):
|
|
self.client.get_aggregators()
|
|
get_mock.assert_called_once_with(
|
|
'http://127.0.0.1:4242/api/aggregators')
|
|
|
|
@mock.patch.object(requests, 'get')
|
|
def test_get_version(self, get_mock):
|
|
self.client.get_version()
|
|
get_mock.assert_called_once_with('http://127.0.0.1:4242/api/version')
|
|
|
|
@mock.patch.object(requests, 'get')
|
|
def test_get_query(self, get_mock):
|
|
self.client.get_query('start=0&end=12&m=max:2-min:bla')
|
|
get_mock.assert_called_once_with(
|
|
'http://127.0.0.1:4242/api/query?start=0&end=12&m=max:2-min:bla')
|