Commit f6a0876e authored by juga's avatar juga
Browse files

Add test for trim_results_ip_changed

parent 4e7bc3e5
# -*- 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
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment