commit f6a0876eebddc04d9158ca2f6f2fe80132c3def9 Author: juga0 juga@riseup.net Date: Mon Jun 18 13:29:58 2018 +0000
Add test for trim_results_ip_changed --- tests/unit/lib/test_resultdump.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+)
diff --git a/tests/unit/lib/test_resultdump.py b/tests/unit/lib/test_resultdump.py new file mode 100644 index 0000000..05a21ae --- /dev/null +++ b/tests/unit/lib/test_resultdump.py @@ -0,0 +1,31 @@ +# -*- coding: utf-8 -*- +"""Unit tests for resultdump.""" +from sbws.lib.resultdump import trim_results_ip_changed, Result, ResultSuccess + + +TIME1 = 1529232277.9028733 +TIME2 = 1529232278.9028733 +FP1 = 'A' * 40 +FP2 = 'Z' * 40 +ED25519 = 'g+Shk00y9Md0hg1S6ptnuc/wWKbADBgdjT0Kg+TSF3s' +CIRC = [FP1, FP2] +DEST_URL = 'http://example.com/sbws.bin' +NICK = 'A' +RELAY_ip1 = '169.254.100.1' +RELAY_ip2 = '169.254.100.2' +RELAY1 = Result.Relay(FP1, NICK, RELAY_ip1, ED25519) +RELAY2 = Result.Relay(FP1, NICK, RELAY_ip2, ED25519) +RTTS = [5, 25] +DOWNLOADS = [{'duration': 4, 'amount': 40}] + +RESULTSUCCESS1 = ResultSuccess(RTTS, DOWNLOADS, RELAY1, CIRC, DEST_URL, + 'sbws', t=TIME1) +RESULTSUCCESS2 = ResultSuccess(RTTS, DOWNLOADS, RELAY2, CIRC, DEST_URL, + 'sbws', t=TIME2) +RESULTDICT = {FP1: [RESULTSUCCESS1, RESULTSUCCESS2]} + + +def test_trim_results_ip_changed(): + expected_results_dict = {FP1: [RESULTSUCCESS2]} + results_dict = trim_results_ip_changed(RESULTDICT) + assert expected_results_dict == results_dict