def test_add_mixed(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 44)
gl.add("name2", 55.66, 2222)
assert gl._data == {"name": (44, 1111), "name2": (55.66, 2222)}
def test_add_mixed(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 44)
gl.add("name2", 55.66, 2222)
assert gl._data == {"name": (44, 1111), "name2": (55.66, 2222)}
def test_sendall_exception(m_create_connection, setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add("name", 44)
m_create_connection.side_effect = Exception("oops")
# No exception should be raised
gl.sendall()
assert gl._data == {}
def test_sendall_exception(m_create_connection, setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add("name", 44)
m_create_connection.side_effect = Exception("oops")
# No exception should be raised
gl.sendall()
assert gl._data == {}
def test_generate_data(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 33)
gl.add("name2", 66, 2222)
expected_data = "prefix.mine.name 33 1111\n" \
"prefix.mine.name2 66 2222\n"
assert gl.generate_data("prefix.mine") == expected_data
def test_generate_data(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 33)
gl.add("name2", 66, 2222)
expected_data = "prefix.mine.name 33 1111\n" \
"prefix.mine.name2 66 2222\n"
assert gl.generate_data("prefix.mine") == expected_data
def test_single_server(setup):
gl = get_graphite_logger()
gl.add("name", 44)
gl.add_destination("host1", 1111, "prefix1")
with mock.patch("socket.create_connection") as conn:
gl.sendall()
expected_calls = [
mock.call(("host1", 1111), timeout=10),
]
conn.assert_has_calls(expected_calls)
def test_single_server(setup):
gl = get_graphite_logger()
gl.add("name", 44)
gl.add_destination("host1", 1111, "prefix1")
with mock.patch("socket.create_connection") as conn:
gl.sendall()
expected_calls = [
mock.call(("host1", 1111), timeout=10),
]
conn.assert_has_calls(expected_calls)
def test_multiple_servers(setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add_destination("host2", 2222, "prefix2")
gl.add("name", 44)
with mock.patch("socket.create_connection") as conn:
gl.sendall()
expected_calls = [
mock.call(("host1", 1111), timeout=10),
mock.call(("host2", 2222), timeout=10),
]
conn.assert_has_calls(expected_calls, any_order=True)
def test_multiple_servers(setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add_destination("host2", 2222, "prefix2")
gl.add("name", 44)
with mock.patch("socket.create_connection") as conn:
gl.sendall()
expected_calls = [
mock.call(("host1", 1111), timeout=10),
mock.call(("host2", 2222), timeout=10),
]
conn.assert_has_calls(expected_calls, any_order=True)
def test_multiple_servers_sendall(setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add_destination("host2", 2222, "prefix2")
with mock.patch("time.time") as m_time:
m_time.return_value = 9999
gl.add("name", 44)
with mock.patch("socket.create_connection") as conn:
sock = mock.MagicMock()
conn.return_value = sock
gl.sendall()
expected_calls = [
mock.call("prefix1.name 44 9999\n"),
mock.call("prefix2.name 44 9999\n"),
]
sock.sendall.assert_has_calls(expected_calls)
def test_multiple_servers_sendall(setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.add_destination("host2", 2222, "prefix2")
with mock.patch("time.time") as m_time:
m_time.return_value = 9999
gl.add("name", 44)
with mock.patch("socket.create_connection") as conn:
sock = mock.MagicMock()
conn.return_value = sock
gl.sendall()
expected_calls = [
mock.call("prefix1.name 44 9999\n"),
mock.call("prefix2.name 44 9999\n"),
]
sock.sendall.assert_has_calls(expected_calls)
def test_add_no_timestamp(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 55)
assert gl._data == {"name": (55, 1111)}
def test_collect(setup):
gl = get_graphite_logger()
gl.add("name", 44, collect=True)
gl.add("name", 55.66, 2222, collect=True)
assert gl._data == {"name": (99.66, 2222)}
def test_add_destination(setup):
gl = get_graphite_logger()
gl.add_destination("host0", 888, "prefix0")
gl.add_destination("host9", 999, "prefix9")
assert gl._servers == [("host0", 888, "prefix0"),
("host9", 999, "prefix9")]
def test_add_string(setup):
gl = get_graphite_logger()
gl.add("name", "value")
assert gl._data == {}
def test_rewrite(setup):
gl = get_graphite_logger()
gl.add("name", 44)
gl.add("name", 55.66, 2222)
assert gl._data == {"name": (55.66, 2222)}
def test__generate_line(setup):
gl = get_graphite_logger()
assert gl._generate_line("my.prefix", "your.value", 55, 1111) == \
"my.prefix.your.value 55 1111\n"
def test__generate_line(setup):
gl = get_graphite_logger()
assert gl._generate_line("my.prefix", "your.value", 55, 1111) == \
"my.prefix.your.value 55 1111\n"
def test_rewrite(setup):
gl = get_graphite_logger()
gl.add("name", 44)
gl.add("name", 55.66, 2222)
assert gl._data == {"name": (55.66, 2222)}
def test_add_string(setup):
gl = get_graphite_logger()
gl.add("name", "value")
assert gl._data == {}
def test_add_no_timestamp(setup):
gl = get_graphite_logger()
with mock.patch("time.time") as m_time:
m_time.return_value = 1111
gl.add("name", 55)
assert gl._data == {"name": (55, 1111)}
def test_sendall_no_data(m_conn, setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.sendall()
assert m_conn.call_count == 0
def test_sendall_no_data(m_conn, setup):
gl = get_graphite_logger()
gl.add_destination("host1", 1111, "prefix1")
gl.sendall()
assert m_conn.call_count == 0
def test_collect(setup):
gl = get_graphite_logger()
gl.add("name", 44, collect=True)
gl.add("name", 55.66, 2222, collect=True)
assert gl._data == {"name": (99.66, 2222)}
def test_add_destination(setup):
gl = get_graphite_logger()
gl.add_destination("host0", 888, "prefix0")
gl.add_destination("host9", 999, "prefix9")
assert gl._servers == [("host0", 888, "prefix0"),
("host9", 999, "prefix9")]
