[tor-commits] [sbws/master] Some integration tests for measuring relays (probably will fail)

[pastly at torproject.org]

commit c6c7ed1a6c5db840a8ca59ede892874ae41e6d6d
Author: Matt Traudt <sirmatt at ksu.edu>
Date:   Tue Jun 19 22:19:53 2018 -0400

    Some integration tests for measuring relays (probably will fail)
---
 tests/integration/core/test_scanner.py | 87 ++++++++++++++++++++++++++++++++++
 1 file changed, 87 insertions(+)

diff --git a/tests/integration/core/test_scanner.py b/tests/integration/core/test_scanner.py
new file mode 100644
index 0000000..9f5bb8c
--- /dev/null
+++ b/tests/integration/core/test_scanner.py
@@ -0,0 +1,87 @@
+from sbws.core.scanner import measure_relay
+from sbws.lib.circuitbuilder import GapsCircuitBuilder as CB
+from sbws.util.config import get_config
+from sbws.lib.relaylist import RelayList
+from sbws.lib.destination import DestinationList
+from sbws.lib.resultdump import ResultSuccess
+
+
+def assert_within(value, target, radius):
+    '''
+    Assert that **value** is within **radius** of **target**
+
+    If target is 10 and radius is 2, value can be anywhere between 8 and 12
+    inclusive
+    '''
+    assert target - radius < value, 'Value is too small. {} is not within '\
+        '{} of {}'.format(value, radius, target)
+    assert target + radius > value, 'Value is too big. {} is not within '\
+        '{} of {}'.format(value, radius, target)
+
+
+def get_everything_to_measure(dotsbws, cont, parser):
+    args = parser.parse_args(
+        '-d {} --log-level DEBUG scanner'.format(dotsbws).split())
+    conf = get_config(args)
+    conf['destinations']['foo'] = 'on'
+    conf['destinations.foo'] = {}
+    conf['destinations.foo']['url'] = 'http://127.0.0.1:28888/sbws.bin'
+    rl = RelayList(args, conf, cont)
+    cb = CB(args, conf, cont)
+    dests, error_msg = DestinationList.from_config(conf, cb, rl, cont)
+    assert dests, error_msg
+    return {
+        'args': args,
+        'conf': conf,
+        'rl': rl,
+        'cb': cb,
+        'dests': dests
+    }
+
+
+def test_measure_relay_with_maxadvertisedbandwidth(
+        persistent_launch_tor, parser, persistent_empty_dotsbws):
+    cont = persistent_launch_tor
+    dotsbws = persistent_empty_dotsbws.name
+    d = get_everything_to_measure(dotsbws, cont, parser)
+    args = d['args']
+    conf = d['conf']
+    rl = d['rl']
+    dests = d['dests']
+    cb = d['cb']
+    # 117A456C911114076BEB4E757AC48B16CC0CCC5F is relay1mbyteMAB
+    relay = [r for r in rl.relays
+             if r.nickname == 'relay1mbyteMAB'][0]
+    result = measure_relay(args, conf, dests, cb, rl, relay)
+    assert len(result) == 1
+    result = result[0]
+    assert isinstance(result, ResultSuccess)
+    one_mbyte = 1 * 1024 * 1024
+    allowed_error = 5  # bytes per second
+    dls = result.downloads
+    for dl in dls:
+        assert_within(dl['amount'] / dl['duration'], one_mbyte, allowed_error)
+
+
+def test_measure_relay_with_relaybandwidthrate(
+        persistent_launch_tor, parser, persistent_empty_dotsbws):
+    cont = persistent_launch_tor
+    dotsbws = persistent_empty_dotsbws.name
+    d = get_everything_to_measure(dotsbws, cont, parser)
+    args = d['args']
+    conf = d['conf']
+    rl = d['rl']
+    dests = d['dests']
+    cb = d['cb']
+    # relay1mbyteRBR 934E06F38A391CB71DF83ECDE05DFF5CDE3AC49D
+    relay = [r for r in rl.relays
+             if r.nickname == 'relay1mbyteRBR'][0]
+    result = measure_relay(args, conf, dests, cb, rl, relay)
+    assert len(result) == 1
+    result = result[0]
+    assert isinstance(result, ResultSuccess)
+    one_mbyte = 1 * 1024 * 1024
+    allowed_error = 5  # bytes per second
+    dls = result.downloads
+    for dl in dls:
+        assert_within(dl['amount'] / dl['duration'], one_mbyte, allowed_error)