def test_countfiles_fe(self):
"""Testing def main_flow.
Recursive default counting.
Equivalent to
"count-files ~/.../tests/data_for_tests/django_staticfiles_for_test -fe {extension}"
Non recursive counting.
Equivalent to
"count-files ~/.../tests/data_for_tests/django_staticfiles_for_test -nr -fe {extension}"
:return:
"""
location = self.get_locations('data_for_tests')
extensions = {
'py': 2,
'json': 1,
'woff': 1,
'.': 2,
'TXT': 3,
'txt': 3
}
nr_extensions = {'css': 0, '.': 1, 'TXT': 1, 'txt': 1}
# case-insensitive and recursive
for k, v in extensions.items():
with self.subTest(k=k, v=v):
self.assertEqual(main_flow([location, '-fe', f'{k}']), v)
# case-sensitive and non-recursive
for k, v in nr_extensions.items():
with self.subTest(k=k, v=v):
self.assertEqual(
main_flow([location, '-nr', '-c', '-fe', f'{k}']), v)
def test_for_hidden(self):
"""Testing def main_flow.
Equivalent to
"count-files ~/.../tests/data_for_tests -nr -t .."
and "count-files ~/.../tests/data_for_tests -nr -a -t .."
:return:
"""
current_os = get_current_os()
if current_os.name == 'WinOS':
self.assertEqual(
main_flow([
self.get_locations('test_hidden_windows'), '-nr', '-t',
'..'
]), 1)
self.assertEqual(
main_flow([
self.get_locations('test_hidden_windows'), '-nr', '-t',
'..', '-a'
]), 2)
elif current_os.name == 'UnixOS':
self.assertEqual(
main_flow([
self.get_locations('test_hidden_linux'), '-nr', '-t', '..'
]), 1)
self.assertEqual(
main_flow([
self.get_locations('test_hidden_linux'), '-nr', '-t', '..',
'-a'
]), 2)
else:
# raise unittest exception
self.skipTest(
'For BaseOS, hidden file detection is currently not available.'
)
def test_parser_exit(self):
"""Check if all checks are performed in CLI.
1-2-3)The path does not exist, or there may be a typo in it.
4)Preview for an unsupported file type(no extension).
5)Preview for an unsupported file type(not in SUPPORTED_TYPES).
:return:
"""
args_dict = {
(self.get_locations('not_exists'), ):
1,
(self.get_locations('not_exists'), '-t', 'txt'):
1,
(self.get_locations('not_exists'), '-fe', '..'):
1,
# -fe used, no preview to count and -t
(self.get_locations('data_for_tests'), '-fe', '.', '-p'):
1,
(self.get_locations('data_for_tests'), '-fe', 'woff', '-p'):
1
}
for k, v in args_dict.items():
with self.subTest(k=k, v=v):
try:
main_flow(k)
# should return parser.exit(status=1)
except SystemExit as se:
# main_flow return parser.exit(status=0) for count_group if check is not effected in CLI
self.assertEqual(se.code, v)
except Exception as e:
self.fail(f'Unexpected exception raised: {e}')
else:
# main_flow return total_result for --total if check is not effected in CLI
# main_flow return len(files) for -fe .. if check is not effected in CLI
self.fail('SystemExit not raised')
def test_parser_exit_for_hidden_unix(self):
"""Check if all checks are performed in CLI.
If include_hidden=False and hidden folder in path:
In this case, the hidden folder is skipped(default settings)
and, accordingly, the files are not count at all.
TODO: revise the check for cases when:
count-files ~/Documents/not_hidden_subfolder/hidden_folder
You can count files in Documents, not_hidden_subfolder and not count them in a hidden_folder.
:return:
"""
# if not include_hidden and current_os.is_hidden_file_or_dir(location)
args_dict = {
(self.get_locations('test_hidden_linux', '.ebookreader'), ):
1,
(self.get_locations('test_hidden_linux', '.ebookreader'), '-t', 'txt'):
1,
(self.get_locations('test_hidden_linux', '.ebookreader'), '-fe', '..'):
1
}
for k, v in args_dict.items():
with self.subTest(k=k, v=v):
try:
main_flow(k)
# should return parser.exit(status=1)
except SystemExit as se:
# main_flow return parser.exit(status=0) for count_group if check is not effected in CLI
self.assertEqual(se.code, v)
except Exception as e:
self.fail(f'Unexpected exception raised: {e}')
else:
# main_flow return total_result for --total if check is not effected in CLI
# main_flow return len(files) for -fe .. if check is not effected in CLI
self.fail('SystemExit not raised')
def test_countfiles_all_t(self):
"""Testing def main_flow.
Recursive/non recursive counting.
Testing for hidden files is not carried out here.
Equivalent to "count-files ~/.../tests/data_for_tests -a -t .."
:return:
"""
self.assertEqual(
main_flow([self.get_locations('data_for_tests'), '-t', '..']), 16)
self.assertEqual(
main_flow(
[self.get_locations('data_for_tests'), '-t', '..', '-nr']), 6)
def test_parser_exit_no_data(self):
"""Testing def main_flow. Count group.
No files were found in the specified directory.
:return:
"""
try:
# count group, if not data
location = self.get_locations('data_for_tests',
'django_staticfiles_for_test',
'admin', 'css')
main_flow([location, '-nr'])
except SystemExit as e:
# not error, the folder is empty
self.assertEqual(e.code, 0)
def test_for_hidden_win(self):
"""Testing def main_flow.
Equivalent to
"python __main__.py ~/.../tests/data_for_tests -nr -t .."
and "python __main__.py ~/.../tests/data_for_tests -nr -a -t .."
:return:
"""
self.assertEqual(
main_flow(
[self.get_locations('test_hidden_windows'), '-nr', '-t',
'..']), 1)
self.assertEqual(
main_flow([
self.get_locations('test_hidden_windows'), '-nr', '-t', '..',
'-a'
]), 2)
def test_countfiles_fe_and_preview(self):
"""Testing def main_flow.
Passed: --preview for supported types.
Not a visual presentation(other tests), but the fact that there is no SystemExit.
:return:
"""
location = self.get_locations('data_for_tests')
extensions = {'json': 1, 'TXT': 3, '..': 16}
# case-insensitive and recursive + preview
for k, v in extensions.items():
with self.subTest(k=k, v=v):
self.assertEqual(
main_flow([location, '-fe', f'{k}', '-p', '-ps', '5']), v)
import sys
import os
from count_files.platforms import get_current_os
from count_files.__main__ import main_flow
from count_files.utils.viewing_modes import show_2columns, \
show_result_for_total, show_result_for_search_files
def get_locations(*args):
return os.path.normpath(os.path.join(os.path.expanduser('~/'), *args))
current_os = get_current_os()
main_c = """
main_flow([location])
"""
main_fe = """
main_flow([location, '-fe', 'txt'])
"""
main_t = """
main_flow([location, '-t', 'txt'])
"""
main_fm = """
main_flow([location, '-fm', '*'])
"""