[tor-commits] [sbws/master] Add test for trim_results_ip_changed

[pastly at torproject.org]

commit f6a0876eebddc04d9158ca2f6f2fe80132c3def9
Author: juga0 <juga at 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