|
1
|
|
|
import unittest |
|
2
|
|
|
from mock import MagicMock |
|
3
|
|
|
from kafka_influxdb.writer import influxdb_writer |
|
4
|
|
|
import influxdb |
|
5
|
|
|
|
|
6
|
|
|
|
|
7
|
|
|
class TestInfluxDBWriter(unittest.TestCase): |
|
8
|
|
|
def setUp(self): |
|
9
|
|
|
self.host = "myhost", |
|
10
|
|
|
self.port = 1234 |
|
11
|
|
|
self.user = "test" |
|
12
|
|
|
self.password = "test" |
|
13
|
|
|
self.dbname = "mydb" |
|
14
|
|
|
self.verify_ssl = False |
|
15
|
|
|
|
|
16
|
|
|
def create_writer(self, use_ssl=False, use_udp=False, timeout=None): |
|
17
|
|
|
self.use_ssl = use_ssl |
|
18
|
|
|
self.use_udp = use_udp |
|
19
|
|
|
self.timeout = timeout |
|
20
|
|
|
return influxdb_writer.InfluxDBWriter(self.host, |
|
21
|
|
|
self.port, |
|
22
|
|
|
self.user, |
|
23
|
|
|
self.password, |
|
24
|
|
|
self.dbname, |
|
25
|
|
|
self.use_ssl, |
|
26
|
|
|
self.verify_ssl, |
|
27
|
|
|
self.timeout, |
|
28
|
|
|
self.use_udp, |
|
29
|
|
|
self.port) |
|
30
|
|
|
|
|
31
|
|
|
def test_write(self): |
|
32
|
|
|
writer = self.create_writer() |
|
33
|
|
|
writer.client = MagicMock() |
|
34
|
|
|
writer.write([ |
|
35
|
|
|
"cpu,host=server01,region=uswest value=1.0 1434055562000", |
|
36
|
|
|
"cpu,host=server02,region=uswest value=2.0 1434055562005" |
|
37
|
|
|
]) |
|
38
|
|
|
writer.client.request.assert_called_once_with(url='write', |
|
39
|
|
|
expected_response_code=204, |
|
40
|
|
|
headers={'Content-type': 'application/octet-stream', |
|
41
|
|
|
'Accept': 'text/plain'}, |
|
42
|
|
|
params={ |
|
43
|
|
|
'rp': 1234, 'db': 'mydb'}, |
|
44
|
|
|
data='cpu,host=server01,region=uswest value=1.0 1434055562000\n' |
|
45
|
|
|
'cpu,host=server02,region=uswest value=2.0 1434055562005', |
|
46
|
|
|
method='POST') |
|
47
|
|
|
|
|
48
|
|
|
def test_ssl_connection(self): |
|
49
|
|
|
influxdb.InfluxDBClient = MagicMock() |
|
50
|
|
|
writer = self.create_writer(True) |
|
51
|
|
|
writer.client = MagicMock() |
|
52
|
|
|
influxdb.InfluxDBClient.assert_called_once_with( |
|
53
|
|
|
self.host, self.port, self.user, self.password, self.dbname, |
|
54
|
|
|
self.use_ssl, self.verify_ssl, self.timeout, self.use_udp, self.port |
|
55
|
|
|
) |
|
56
|
|
|
|
mre /
kafka-influxdb
