[tor-commits] [sbws/master] Add unit tests for writing v3bw file

[pastly at torproject.org]

commit 6912a1fb2aa42cdd221b189fb0c1bf8e44043630
Author: juga0 <juga at riseup.net>
Date:   Mon Jun 25 16:12:29 2018 +0000

    Add unit tests for writing v3bw file
---
 tests/unit/lib/test_v3bwfile.py | 24 ++++++++++++++++++++++++
 1 file changed, 24 insertions(+)

diff --git a/tests/unit/lib/test_v3bwfile.py b/tests/unit/lib/test_v3bwfile.py
index bfe9d9b..bede4ff 100644
--- a/tests/unit/lib/test_v3bwfile.py
+++ b/tests/unit/lib/test_v3bwfile.py
@@ -1,6 +1,7 @@
 # -*- coding: utf-8 -*-
 """Test generation of bandwidth measurements document (v3bw)"""
 import json
+import os.path
 
 from sbws import __version__ as version
 from sbws.globals import SPEC_VERSION, RESULT_VERSION
@@ -8,6 +9,7 @@ from sbws.lib.resultdump import Result, load_result_file
 from sbws.lib.v3bwfile import (V3BWHeader, V3BWLine, TERMINATOR, LINE_SEP,
                                KEYVALUE_SEP_V110, num_results_of_type,
                                V3BWFile)
+from sbws.util.timestamp import now_fname
 
 timestamp = 1523974147
 timestamp_l = str(timestamp)
@@ -168,3 +170,25 @@ def test_v3bwfile(datadir, tmpdir):
     bwls = [V3BWLine.from_results(results[fp]) for fp in results]
     f = V3BWFile(header, bwls)
     assert v3bw == str(f)
+
+
+def test_from_arg_results(datadir, tmpdir, unittest_conf, unittest_args):
+    results = load_result_file(str(datadir.join("results.txt")))
+    expected_header = V3BWHeader(timestamp_l,
+                                 earliest_bandwidth=earliest_bandwidth,
+                                 latest_bandwidth=latest_bandwidth)
+    expected_bwls = [V3BWLine.from_results(results[fp]) for fp in results]
+    expected_f = V3BWFile(expected_header, expected_bwls)
+    v3bwfile = V3BWFile.from_arg_results(unittest_args, unittest_conf, results)
+    assert str(expected_f)[1:] == str(v3bwfile)[1:]
+    output = os.path.join(unittest_args.output, now_fname())
+    print(output)
+    v3bwfile.write(output)
+
+
+def test_from_arg_results_write(datadir, tmpdir, unittest_conf, unittest_args):
+    results = load_result_file(str(datadir.join("results.txt")))
+    v3bwfile = V3BWFile.from_arg_results(unittest_args, unittest_conf, results)
+    output = os.path.join(unittest_args.output, now_fname())
+    v3bwfile.write(output)
+    assert os.path.isfile(output)