1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
# Copyright (c) 2016 Uber Technologies, Inc.
#
# 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.
from unittest import mock
import unittest
from jaeger_client import utils
from jaeger_client.sampler import Sampler
class ConfigTests(unittest.TestCase):
def check_boolean(self, string, default, correct):
assert utils.get_boolean(string, default) == correct
def test_get_false_boolean(self):
self.check_boolean('false', 'asdf', False)
def test_get_0_boolean(self):
self.check_boolean('0', 'asdf', False)
def test_get_true_boolean(self):
self.check_boolean('true', 'qwer', True)
def test_get_1_boolean(self):
self.check_boolean('1', 'qwer', True)
def test_get_unknown_boolean(self):
self.check_boolean('zxcv', 'qwer', 'qwer')
def test_get_None_boolean(self):
self.check_boolean(None, 'qwer', False)
# def test_error_reporter_doesnt_send_metrics_if_not_configured(self):
# er = utils.ErrorReporter(False)
# er.error('foo', 1)
# assert not mock_metrics.count.called
def test_error_reporter_sends_metrics_if_configured(self):
mock_metrics = mock.MagicMock()
er = utils.ErrorReporter(mock_metrics)
er.error('foo', 1)
assert mock_metrics.count.called_with('foo', 1)
def test_error_reporter_doesnt_send_log_messages_if_before_deadline(self):
mock_logger = mock.MagicMock()
er = utils.ErrorReporter(None, logger=mock_logger, log_interval_minutes=1000)
er.error('foo', 1)
assert not mock_logger.error.called
def test_error_reporter_sends_log_messages_if_after_deadline(self):
mock_logger = mock.MagicMock()
# 0 log interval means we're always after the deadline, so always log
er = utils.ErrorReporter(None, logger=mock_logger, log_interval_minutes=0)
er._last_error_reported_at = 0
er.error('foo', 1, 'error args')
assert mock_logger.error.call_args == (('foo', 1, 'error args',),)
def test_local_ip_does_not_blow_up():
import socket
import jaeger_client.utils
socket.gethostname()
with mock.patch('socket.gethostbyname',
side_effect=[IOError(), '127.0.0.1']):
jaeger_client.utils.local_ip()
def test_get_local_ip_by_socket_does_not_blow_up():
import jaeger_client.utils
jaeger_client.utils.get_local_ip_by_socket()
class MockSampler(Sampler):
def __init__(self):
pass
|
