[Author Prev][Author Next][Thread Prev][Thread Next][Author Index][Thread Index]
[tor-commits] [nyx/master] Test _draw_accounting_stats()
commit 28f7b5967d1c99073af70176621340cc442569c2
Author: Damian Johnson <atagar@xxxxxxxxxxxxxx>
Date: Wed Apr 20 09:04:18 2016 -0700
Test _draw_accounting_stats()
Test the graph panel method that renders accounting stats. This also changes
the method to display user friendly units (ex. '5 MB') rather than the raw byte
count.
---
nyx/panel/graph.py | 42 ++++++++++++++++++++----------------------
test/panel/graph.py | 45 +++++++++++++++++++++++++++++++++++++++++++++
2 files changed, 65 insertions(+), 22 deletions(-)
diff --git a/nyx/panel/graph.py b/nyx/panel/graph.py
index 132a7d6..1b2a067 100644
--- a/nyx/panel/graph.py
+++ b/nyx/panel/graph.py
@@ -33,7 +33,7 @@ GraphStat = enum.Enum(('BANDWIDTH', 'bandwidth'), ('CONNECTIONS', 'connections')
Interval = enum.Enum(('EACH_SECOND', 'each second'), ('FIVE_SECONDS', '5 seconds'), ('THIRTY_SECONDS', '30 seconds'), ('MINUTELY', 'minutely'), ('FIFTEEN_MINUTE', '15 minute'), ('THIRTY_MINUTE', '30 minute'), ('HOURLY', 'hourly'), ('DAILY', 'daily'))
Bounds = enum.Enum(('GLOBAL_MAX', 'global_max'), ('LOCAL_MAX', 'local_max'), ('TIGHT', 'tight'))
-DrawAttributes = collections.namedtuple('DrawAttributes', ('stat', 'subgraph_height', 'subgraph_width', 'interval', 'bounds_type', 'accounting', 'right_to_left'))
+DrawAttributes = collections.namedtuple('DrawAttributes', ('stat', 'subgraph_height', 'subgraph_width', 'interval', 'bounds_type', 'right_to_left'))
INTERVAL_SECONDS = {
Interval.EACH_SECOND: 1,
@@ -552,7 +552,6 @@ class GraphPanel(nyx.panel.Panel):
subgraph_width = min(subwindow.width / 2, CONFIG['features.graph.max_width']),
interval = self.update_interval,
bounds_type = self.bounds_type,
- accounting = accounting_stats,
right_to_left = CONFIG['features.graph.right_to_left'],
)
@@ -565,8 +564,8 @@ class GraphPanel(nyx.panel.Panel):
if subwindow.width <= COLLAPSE_WIDTH:
self._draw_bandwidth_stats(subwindow, attr, subwindow.width)
- if attr.accounting:
- self._draw_accounting_stats(subwindow, attr)
+ if accounting_stats:
+ _draw_accounting_stats(subwindow, DEFAULT_CONTENT_HEIGHT + attr.subgraph_height - 2, accounting_stats)
def _draw_subgraph(self, subwindow, attr, data, x, color):
# Concering our subgraph colums, the y-axis label can be at most six
@@ -700,24 +699,6 @@ class GraphPanel(nyx.panel.Panel):
subwindow.addstr(1, labeling_line, primary_footer, PRIMARY_COLOR)
subwindow.addstr(attr.subgraph_width + 1, labeling_line, secondary_footer, SECONDARY_COLOR)
- def _draw_accounting_stats(self, subwindow, attr):
- y = DEFAULT_CONTENT_HEIGHT + attr.subgraph_height - 2
-
- if tor_controller().is_alive():
- hibernate_color = CONFIG['attr.hibernate_color'].get(attr.accounting.status, RED)
-
- x = subwindow.addstr(0, y, 'Accounting (', BOLD)
- x = subwindow.addstr(x, y, attr.accounting.status, BOLD, hibernate_color)
- x = subwindow.addstr(x, y, ')', BOLD)
-
- subwindow.addstr(35, y, 'Time to reset: %s' % str_tools.short_time_label(attr.accounting.time_until_reset))
-
- subwindow.addstr(2, y + 1, '%s / %s' % (attr.accounting.read_bytes, attr.accounting.read_limit), PRIMARY_COLOR)
- subwindow.addstr(37, y + 1, '%s / %s' % (attr.accounting.written_bytes, attr.accounting.write_limit), SECONDARY_COLOR)
- else:
- subwindow.addstr(0, y, 'Accounting:', BOLD)
- subwindow.addstr(12, y, 'Connection Closed...')
-
def _update_accounting(self, event):
if not CONFIG['features.graph.bw.accounting.show']:
self._accounting_stats = None
@@ -744,6 +725,23 @@ class GraphPanel(nyx.panel.Panel):
self.redraw(True)
+def _draw_accounting_stats(subwindow, y, accounting):
+ if tor_controller().is_alive():
+ hibernate_color = CONFIG['attr.hibernate_color'].get(accounting.status, RED)
+
+ x = subwindow.addstr(0, y, 'Accounting (', BOLD)
+ x = subwindow.addstr(x, y, accounting.status, BOLD, hibernate_color)
+ x = subwindow.addstr(x, y, ')', BOLD)
+
+ subwindow.addstr(35, y, 'Time to reset: %s' % str_tools.short_time_label(accounting.time_until_reset))
+
+ subwindow.addstr(2, y + 1, '%s / %s' % (str_tools.size_label(accounting.read_bytes), str_tools.size_label(accounting.read_limit)), PRIMARY_COLOR)
+ subwindow.addstr(37, y + 1, '%s / %s' % (str_tools.size_label(accounting.written_bytes), str_tools.size_label(accounting.write_limit)), SECONDARY_COLOR)
+ else:
+ subwindow.addstr(0, y, 'Accounting:', BOLD)
+ subwindow.addstr(12, y, 'Connection Closed...')
+
+
def _size_label(byte_count, decimal = 1):
"""
Alias for str_tools.size_label() that accounts for if the user prefers bits
diff --git a/test/panel/graph.py b/test/panel/graph.py
new file mode 100644
index 0000000..7ffc48e
--- /dev/null
+++ b/test/panel/graph.py
@@ -0,0 +1,45 @@
+"""
+Unit tests for nyx.panel.graph.
+"""
+
+import datetime
+import unittest
+
+import stem.control
+
+import nyx.panel.graph
+import test
+
+from test import require_curses
+from mock import patch
+
+EXPECTED_ACCOUNTING = """
+Accounting (awake) Time to reset: 01:02
+ 4 KB / 105 KB 2 KB / 9 KB
+""".strip()
+
+
+class TestGraph(unittest.TestCase):
+ @require_curses
+ @patch('nyx.panel.graph.tor_controller')
+ def test_draw_accounting_stats(self, tor_controller_mock):
+ tor_controller_mock().is_alive.return_value = True
+
+ stat = stem.control.AccountingStats(
+ 1410723598.276578,
+ 'awake',
+ datetime.datetime(2014, 9, 14, 19, 41),
+ 62,
+ 4837, 102944, 107781,
+ 2050, 7440, 9490,
+ )
+
+ rendered = test.render(nyx.panel.graph._draw_accounting_stats, 0, stat)
+ self.assertEqual(EXPECTED_ACCOUNTING, rendered.content)
+
+ @require_curses
+ @patch('nyx.panel.graph.tor_controller')
+ def test_draw_accounting_stats_disconnected(self, tor_controller_mock):
+ tor_controller_mock().is_alive.return_value = False
+ rendered = test.render(nyx.panel.graph._draw_accounting_stats, 0, None)
+ self.assertEqual('Accounting: Connection Closed...', rendered.content)
_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits