From 053d1e951a3fe925575367b5a66b6d2b14b9044c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Antoine=20Beaupr=C3=A9?= <anarcat@debian.org>
Date: Tue, 8 Oct 2019 18:19:44 -0400
Subject: [PATCH] main test from micah data
---
tsa/howto/upgrades/predict-os.py | 33 +++++++++++++++++++++++++++++++-
1 file changed, 32 insertions(+), 1 deletion(-)
diff --git a/tsa/howto/upgrades/predict-os.py b/tsa/howto/upgrades/predict-os.py
index 8d141de9..3f28a0aa 100755
--- a/tsa/howto/upgrades/predict-os.py
+++ b/tsa/howto/upgrades/predict-os.py
@@ -25,9 +25,16 @@ import collections
from datetime import datetime
import io
import logging
+import logging.handlers
import os
import os.path
import sys
+import tempfile
+
+try:
+ import pytest
+except ImportError:
+ pytest = None
import pandas as pd
import matplotlib
@@ -107,11 +114,35 @@ def main(args):
try:
date = guess_completion_time(records, args.source)
print("completion time of %s major upgrades: %s" % (args.source, date))
- except ValueError as e:
+ except (TypeError, ValueError) as e:
logging.warning("cannot guess completion time: %s", e)
+ date = 'N/A'
plot_records(records, date, args)
+if pytest is not None:
+ @pytest.xfail('not sure why there is no output, but this should work')
+ @pytest.mark.parametrize("test_input,expected",
+ [(b'''Date,release,count
+2019-10-08,stretch,83
+2019-10-08,buster,3
+2019-10-08,sid,1
+2019-10-08,jessie,2''', 'cannot guess completion time')])
+ def test_main(test_input, expected):
+ with tempfile.NamedTemporaryFile() as csv:
+ csv.write(test_input)
+ csv.flush()
+ handler = logging.handlers.MemoryHandler(1000)
+ handler.setLevel('DEBUG')
+ logging.getLogger('').addHandler(handler)
+ with tempfile.NamedTemporaryFile(suffix='.png') as graph:
+ args = parse_args(['--path', csv.name, '--output', graph.name])
+ main(args)
+ output = "\n".join([record.getMessage()
+ for record in handler.buffer])
+ assert expected in output
+
+
def load_csv(fp):
'''load the data from the CSV'''
return pd.read_csv(fp)
--
GitLab