niklas/tracetools_analysis
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add test for quiet loading option

Browse source 
Signed-off-by: Christophe Bedard <bedard.christophe@gmail.com>
This commit is contained in:
Christophe Bedard 2020-05-23 10:15:29 -04:00
parent 28b61587cf
commit 7d3eded902
1 changed files with 22 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

29
tracetools_analysis/test/tracetools_analysis/test_process_command.py
View file

@ -1,5 +1,6 @@
#!/usr/bin/env python3
# Copyright 2019 Apex.AI, Inc.
# Copyright 2020 Christophe Bedard
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
@ -13,6 +14,8 @@
# See the License for the specific language governing permissions and
# limitations under the License.
import contextlib
from io import StringIO
import os
import shutil
import tempfile
@ -60,31 +63,43 @@ class TestProcessCommand(unittest.TestCase):
def tearDown(self):
shutil.rmtree(self.test_dir_path)
def test_inspect_input_path(self) -> None:
def test_inspect_input_path(
self,
quiet: bool = False,
) -> None:
# Should find converted file under directory
file_path, create_file = inspect_input_path(self.with_converted_file_dir, False)
file_path, create_file = inspect_input_path(self.with_converted_file_dir, False, quiet)
self.assertEqual(self.converted_file_path, file_path)
self.assertFalse(create_file)
# Should find it but set it to be re-created
file_path, create_file = inspect_input_path(self.with_converted_file_dir, True)
file_path, create_file = inspect_input_path(self.with_converted_file_dir, True, quiet)
self.assertEqual(self.converted_file_path, file_path)
self.assertTrue(create_file)
# Should fail to find converted file under directory
file_path, create_file = inspect_input_path(self.without_converted_file_dir, False)
file_path, create_file = inspect_input_path(self.without_converted_file_dir, False, quiet)
self.assertIsNone(file_path)
self.assertFalse(create_file)
file_path, create_file = inspect_input_path(self.without_converted_file_dir, True)
file_path, create_file = inspect_input_path(self.without_converted_file_dir, True, quiet)
self.assertIsNone(file_path)
self.assertFalse(create_file)
# Should accept any file path if it exists
file_path, create_file = inspect_input_path(self.random_file_path, False)
file_path, create_file = inspect_input_path(self.random_file_path, False, quiet)
self.assertEqual(self.random_file_path, file_path)
self.assertFalse(create_file)
# Should set it to be re-created
file_path, create_file = inspect_input_path(self.random_file_path, True)
file_path, create_file = inspect_input_path(self.random_file_path, True, quiet)
self.assertEqual(self.random_file_path, file_path)
self.assertTrue(create_file)
# TODO try with a trace directory
def test_inspect_input_path_quiet(self) -> None:
temp_stdout = StringIO()
with contextlib.redirect_stdout(temp_stdout):
# Just run the other test again
self.test_inspect_input_path(quiet=True)
# Shouldn't be any output
output = temp_stdout.getvalue()
self.assertEqual(0, len(output), f'was not quiet: "{output}"')