[Author Prev][Author Next][Thread Prev][Thread Next][Author Index][Thread Index]

[tor-commits] [onionperf/master] Fix unit tests.



commit 3e67cb32257ecbaec632b782603c162aa10a0af6
Author: Karsten Loesing <karsten.loesing@xxxxxxx>
Date:   Mon Jul 13 16:55:10 2020 +0200

    Fix unit tests.
---
 onionperf/tests/test_analysis.py | 265 ++++++++++++++++-----------------------
 1 file changed, 110 insertions(+), 155 deletions(-)

diff --git a/onionperf/tests/test_analysis.py b/onionperf/tests/test_analysis.py
index 6463458..82e2043 100644
--- a/onionperf/tests/test_analysis.py
+++ b/onionperf/tests/test_analysis.py
@@ -1,7 +1,8 @@
 import os
 import pkg_resources
 from nose.tools import *
-from onionperf import analysis, util
+from onionperf import util
+from tgentools import analysis
 
 
 def absolute_data_path(relative_path=""):
@@ -12,188 +13,142 @@ def absolute_data_path(relative_path=""):
                                            "tests/data/" + relative_path)
 
 DATA_DIR = absolute_data_path()
-LINE_ERROR = '2019-04-22 14:41:20 1555940480.647663 [message] [shd-tgen-transfer.c:1504] [_tgentransfer_log] [transfer-error] transport TCP,12,localhost:127.0.0.1:46878,localhost:127.0.0.1:43735,dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080,state=ERROR,error=READ transfer transfer5m,4,cyan,GET,5242880,(null),0,state=ERROR,error=PROXY total-bytes-read=0 total-bytes-write=0 payload-bytes-read=0/5242880 (0.00%) usecs-to-socket-create=11 usecs-to-socket-connect=210 usecs-to-proxy-init=283 usecs-to-proxy-choice=348 usecs-to-proxy-request=412 usecs-to-proxy-response=-1 usecs-to-command=-1 usecs-to-response=-1 usecs-to-first-byte=-1 usecs-to-last-byte=-1 usecs-to-checksum=-1'
-
-NO_PARSE_LINE = '2018-04-14 21:10:04 1523740204.809894 [message] [shd-tgen-transfer.c:803] [_tgentransfer_log] [transfer-error] transport TCP,17,NULL:37.218.247.40:26006,NULL:0.0.0.0:0,146.0.73.4:146.0.73.4:1313,state=SUCCESS,error=NONE transfer (null),26847,op-nl,NONE,0,(null),0,state=ERROR,error=AUTH total-bytes-read=1 total-bytes-write=0 payload-bytes-write=0/0 (-nan%) usecs-to-socket-create=0 usecs-to-socket-connect=8053676879205 usecs-to-proxy-init=-1 usecs-to-proxy-choice=-1 usecs-to-proxy-request=-1 usecs-to-proxy-response=-1 usecs-to-command=-1 usecs-to-response=-1 usecs-to-first-byte=-1 usecs-to-last-byte=-1 usecs-to-checksum=-1'
-
-def test_transfer_status_event():
-    transfer = analysis.TransferStatusEvent(LINE_ERROR)
-    assert_equals(transfer.is_success, False)
-    assert_equals(transfer.is_error, False)
-    assert_equals(transfer.is_complete, False)
-    assert_equals(transfer.unix_ts_end, 1555940480.647663)
-    assert_equals(transfer.endpoint_local, 'localhost:127.0.0.1:46878')
-    assert_equals(transfer.endpoint_proxy, 'localhost:127.0.0.1:43735')
-    assert_equals(
-        transfer.endpoint_remote,
-        'dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080'
-    )
+LINE_ERROR = '2019-04-22 14:41:20 1555940480.647663 [message] [tgen-stream.c:1618] [_tgenstream_log] [stream-error] transport [fd=12,local=localhost:127.0.0.1:46878,proxy=localhost:127.0.0.1:43735,remote=dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080,state=ERROR,error=READ] stream [id=4,vertexid=stream5m,name=cyan,peername=(null),sendsize=0,recvsize=5242880,sendstate=SEND_COMMAND,recvstate=RECV_NONE,error=PROXY] bytes [total-bytes-recv=0,total-bytes-send=0,payload-bytes-recv=0,payload-bytes-send=0,payload-progress-recv=0.00%,payload-progress-send=100.00%] times [created-ts=5948325159988,usecs-to-socket-create=11,usecs-to-socket-connect=210,usecs-to-proxy-init=283,usecs-to-proxy-choice=348,usecs-to-proxy-request=412,usecs-to-proxy-response=-1,usecs-to-command=-1,usecs-to-response=-1,usecs-to-first-byte-recv=-1,usecs-to-last-byte-recv=-1,usecs-to-checksum-recv=-1,usecs-to-first-byte-send=-1,usecs-to-last-byte-send=-1,usecs-to-checksum-send=-1,now-ts=594844
 6579043]'
+
+NO_PARSE_LINE = '2018-04-14 21:10:04 1523740204.809894 [message] [tgen-stream.c:1618] [_tgenstream_log] [stream-error] transport [fd=17,local=localhost:127.0.0.1:46878,proxy=localhost:127.0.0.1:43735,remote=dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080,state=SUCCESS,error=NONE] stream [id=4,vertexid=stream5m,name=cyan,peername=(null),sendsize=0,recvsize=5242880,sendstate=SEND_COMMAND,recvstate=RECV_NONE,error=PROXY] bytes [total-bytes-recv=1,total-bytes-send=0,payload-bytes-recv=0,payload-bytes-send=0,payload-progress-recv=0.00%,payload-progress-send=100.00%] times [created-ts=5948325159988,usecs-to-socket-create=0,usecs-to-socket-connect=210,usecs-to-proxy-init=-1,usecs-to-proxy-choice=-1,usecs-to-proxy-request=-1,usecs-to-proxy-response=-1,usecs-to-command=-1,usecs-to-response=-1,usecs-to-first-byte-recv=-1,usecs-to-last-byte-recv=-1,usecs-to-checksum-recv=-1,usecs-to-first-byte-send=-1,usecs-to-last-byte-send=-1,usecs-to-checksum-send=-1,now-ts=59484
 46579043]'
+
+
+def test_stream_status_event():
+    stream = analysis.StreamStatusEvent(LINE_ERROR)
+    assert_equals(stream.is_success, False)
+    assert_equals(stream.is_error, False)
+    assert_equals(stream.is_complete, False)
+    assert_equals(stream.unix_ts_end, 1555940480.647663)
+    assert_equals(stream.transport_info['local'], 'localhost:127.0.0.1:46878')
+    assert_equals(stream.transport_info['proxy'], 'localhost:127.0.0.1:43735')
     assert_equals(
-        transfer.endpoint_remote,
+        stream.transport_info['remote'],
         'dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080'
     )
-    assert_equals(transfer.transfer_id, 'transfer5m:4')
-    assert_equals(transfer.hostname_local, 'cyan')
-    assert_equals(transfer.method, 'GET')
-    assert_equals(transfer.filesize_bytes, 5242880)
-    assert_equals(transfer.hostname_remote, '(null)')
-    assert_equals(transfer.error_code, 'PROXY')
-    assert_equals(transfer.total_bytes_read, 0)
-    assert_equals(transfer.total_bytes_write, 0)
-    assert_equals(transfer.is_commander, True)
-    assert_equals(transfer.payload_bytes_status, 0)
-    assert_equals(transfer.unconsumed_parts, [
-        'usecs-to-socket-create=11', 'usecs-to-socket-connect=210',
-        'usecs-to-proxy-init=283', 'usecs-to-proxy-choice=348',
-        'usecs-to-proxy-request=412', 'usecs-to-proxy-response=-1',
-        'usecs-to-command=-1', 'usecs-to-response=-1',
-        'usecs-to-first-byte=-1', 'usecs-to-last-byte=-1',
-        'usecs-to-checksum=-1'
-    ])
-    assert_equals(transfer.elapsed_seconds, {})
-
-
-def test_transfer_complete_event_init():
-    complete = analysis.TransferCompleteEvent(LINE_ERROR)
+    assert_equals(stream.stream_id, '4:12:localhost:127.0.0.1:46878:dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080')
+    assert_equals(stream.stream_info['name'], 'cyan')
+    assert_equals(stream.stream_info['recvsize'], '5242880')
+    assert_equals(stream.stream_info['peername'], '(null)')
+    assert_equals(stream.stream_info['error'], 'PROXY')
+    assert_equals(stream.byte_info['total-bytes-recv'], '0')
+    assert_equals(stream.byte_info['total-bytes-send'], '0')
+    assert_equals(stream.byte_info['payload-bytes-recv'], '0')
+
+
+def test_stream_complete_event_init():
+    complete = analysis.StreamCompleteEvent(LINE_ERROR)
     assert_equals(complete.is_complete, True)
-    assert_equals(
-        complete.elapsed_seconds, {
-            'proxy_init': 0.000283,
-            'proxy_request': 0.000412,
-            'proxy_choice': 0.000348,
-            'socket_connect': 0.00021,
-            'socket_create': 1.1e-05
-        })
+    assert_equals(complete.time_info['usecs-to-proxy-init'], '283')
+    assert_equals(complete.time_info['usecs-to-proxy-request'], '412')
+    assert_equals(complete.time_info['usecs-to-proxy-choice'], '348')
+    assert_equals(complete.time_info['usecs-to-socket-connect'], '210')
+    assert_equals(complete.time_info['usecs-to-socket-create'], '11')
     assert_equals(complete.unix_ts_start, 1555940480.6472511)
 
 
-def test_transfer_error_event():
-    error = analysis.TransferErrorEvent(LINE_ERROR)
+def test_stream_error_event():
+    error = analysis.StreamErrorEvent(LINE_ERROR)
     assert_equals(error.is_error, True)
     assert_equals(error.is_success, False)
 
 
-def test_transfer_success_event_init():
-    success = analysis.TransferSuccessEvent(LINE_ERROR)
+def test_stream_success_event_init():
+    success = analysis.StreamSuccessEvent(LINE_ERROR)
     assert_equals(success.is_success, True)
 
 
-def test_transfer_object_init():
-    error = analysis.TransferErrorEvent(LINE_ERROR)
-    t = analysis.Transfer(error.transfer_id)
-    assert_equals(t.id, 'transfer5m:4')
-    assert_equals(t.last_event, None)
-    assert_equals(
-        t.payload_progress, {
-            0.0: None,
-            0.1: None,
-            0.2: None,
-            0.3: None,
-            0.4: None,
-            0.5: None,
-            0.6: None,
-            0.7: None,
-            0.8: None,
-            0.9: None,
-            1.0: None
-        })
+def test_stream_object_init():
+    error = analysis.StreamErrorEvent(LINE_ERROR)
+    s = analysis.Stream(error.stream_id)
+    assert_equals(s.id, '4:12:localhost:127.0.0.1:46878:dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080')
+    assert_equals(s.last_event, None)
 
 
-def test_transfer_object_add_event():
-    error = analysis.TransferErrorEvent(LINE_ERROR)
-    t = analysis.Transfer(error.transfer_id)
-    t.add_event(error)
-    assert_equals(t.last_event, error)
-    assert_equals(
-        t.payload_progress, {
-            0.0: 1555940480.647663,
-            0.1: None,
-            0.2: None,
-            0.3: None,
-            0.4: None,
-            0.5: None,
-            0.6: None,
-            0.7: None,
-            0.8: None,
-            0.9: None,
-            1.0: None
-        })
+def test_stream_object_add_event():
+    error = analysis.StreamErrorEvent(LINE_ERROR)
+    s = analysis.Stream(error.stream_id)
+    s.add_event(error)
+    assert_equals(s.last_event, error)
 
 
 @raises(KeyError)
-def test_transfer_object_get_data_error():
-    error = analysis.TransferErrorEvent(LINE_ERROR)
-    t = analysis.Transfer(error.transfer_id)
-    t.add_event(error)
-    t.get_data()['elapsed_seconds']['payload_progress']
+def test_stream_object_get_data_error():
+    error = analysis.StreamErrorEvent(LINE_ERROR)
+    s = analysis.Stream(error.stream_id)
+    s.add_event(error)
+    s.get_data()['elapsed_seconds']['payload_progress_recv']
 
 
-def test_transfer_object_get_data_no_error():
-    success = analysis.TransferSuccessEvent(LINE_ERROR)
-    t = analysis.Transfer(success.transfer_id)
-    t.add_event(success)
+def test_stream_object_get_data_no_error():
+    success = analysis.StreamSuccessEvent(LINE_ERROR)
+    s = analysis.Stream(success.stream_id)
+    s.add_event(success)
     assert_true(
-        t.get_data()['elapsed_seconds']['payload_progress'] is not None)
+        s.get_data()['elapsed_seconds']['payload_progress_recv'] is not None)
 
 
-def test_transfer_object_end_to_end():
-    error = analysis.TransferErrorEvent(LINE_ERROR)
-    t = analysis.Transfer(error.transfer_id)
-    t.add_event(error)
+def test_stream_object_end_to_end():
+    error = analysis.StreamErrorEvent(LINE_ERROR)
+    s = analysis.Stream(error.stream_id)
+    s.add_event(error)
     assert_equals(
-        t.get_data(), {
-            'is_error':
-            True,
-            'endpoint_local':
-            'localhost:127.0.0.1:46878',
-            'total_bytes_read':
-            0,
-            'error_code':
-            'PROXY',
-            'unix_ts_end':
-            1555940480.647663,
-            'hostname_local':
-            'cyan',
-            'endpoint_remote':
-            'dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080',
-            'elapsed_seconds': {
-                'proxy_init': 0.000283,
-                'proxy_request': 0.000412,
-                'proxy_choice': 0.000348,
-                'socket_connect': 0.00021,
-                'socket_create': 1.1e-05
+        s.get_data(), {
+            'is_success': False,
+            'is_error': True,
+            'is_complete': True,
+            'unix_ts_end': 1555940480.647663,
+            'transport_info': {'fd': '12',
+                'local': 'localhost:127.0.0.1:46878',
+                'proxy': 'localhost:127.0.0.1:43735',
+                'remote': 'dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080',
+                'state': 'ERROR',
+                'error': 'READ'},
+            'stream_info': {
+                'id': '4',
+                'vertexid': 'stream5m',
+                'name': 'cyan',
+                'peername': '(null)',
+                'sendsize': '0',
+                'recvsize': '5242880',
+                'sendstate': 'SEND_COMMAND',
+                'recvstate': 'RECV_NONE',
+                'error': 'PROXY'
             },
-            'method':
-            'GET',
-            'is_commander':
-            True,
-            'total_bytes_write':
-            0,
-            'unix_ts_start':
-            1555940480.6472511,
-            'hostname_remote':
-            '(null)',
-            'transfer_id':
-            'transfer5m:4',
-            'is_success':
-            False,
-            'payload_bytes_status':
-            0,
-            'endpoint_proxy':
-            'localhost:127.0.0.1:43735',
-            'is_complete':
-            True,
-            'filesize_bytes':
-            5242880
+            'byte_info': {
+                'total-bytes-recv': '0',
+                'total-bytes-send': '0',
+                'payload-bytes-recv': '0',
+                'payload-bytes-send': '0',
+                'payload-progress-recv': '0.00%',
+                'payload-progress-send': '100.00%'
+            },
+            'time_info': {
+                'created-ts': '5948325159988',
+                'usecs-to-socket-create': '11',
+                'usecs-to-socket-connect': '210',
+                'usecs-to-proxy-init': '283',
+                'usecs-to-proxy-choice': '348',
+                'usecs-to-proxy-request': '412',
+                'usecs-to-proxy-response': '-1',
+                'usecs-to-command': '-1',
+                'usecs-to-response': '-1',
+                'usecs-to-first-byte-recv': '-1',
+                'usecs-to-last-byte-recv': '-1',
+                'usecs-to-checksum-recv': '-1',
+                'usecs-to-first-byte-send': '-1',
+                'usecs-to-last-byte-send': '-1',
+                'usecs-to-checksum-send': '-1',
+                'now-ts': '5948446579043'
+            },
+            'stream_id': '4:12:localhost:127.0.0.1:46878:dc34og3c3aqdqntblnxkstzfvh7iy7llojd4fi5j23y2po32ock2k7ad.onion:0.0.0.0:8080',
+            'unix_ts_start': 1555940480.6472511
         })
 
-
-@raises(ZeroDivisionError)
-def test_transfer_status_parse_error():
-    transfer = analysis.TransferStatusEvent(NO_PARSE_LINE)
-    t = analysis.Transfer(transfer.transfer_id)
-    t.add_event(transfer)
-
 def test_parsing_parse_error():
     parser = analysis.TGenParser()
     parser.parse(util.DataSource(DATA_DIR + 'parse_error'))



_______________________________________________
tor-commits mailing list
tor-commits@xxxxxxxxxxxxxxxxxxxx
https://lists.torproject.org/cgi-bin/mailman/listinfo/tor-commits