mirror of
https://github.com/kellyjonbrazil/jc.git
synced 2025-06-17 00:07:37 +02:00
136 lines
4.5 KiB
Python
136 lines
4.5 KiB
Python
from copy import deepcopy
|
|
import unittest
|
|
from typing import Generator
|
|
import jc.lib
|
|
|
|
|
|
class MyTests(unittest.TestCase):
|
|
def test_lib_parse_csv(self):
|
|
data = {
|
|
'': [],
|
|
'a,b,c\n1,2,3': [{'a':'1', 'b':'2', 'c':'3'}]
|
|
}
|
|
|
|
for test_data, expected_output in data.items():
|
|
self.assertEqual(jc.lib.parse('csv', test_data), expected_output)
|
|
|
|
def test_lib_parse_csv_s_is_generator(self):
|
|
self.assertIsInstance(jc.lib.parse('csv_s', 'a,b,c\n1,2,3'), Generator)
|
|
|
|
def test_lib_parse_kv(self):
|
|
data = {
|
|
'': {},
|
|
'a=1\nb=2\nc=3': {'a':'1', 'b':'2', 'c':'3'}
|
|
}
|
|
|
|
for test_data, expected_output in data.items():
|
|
self.assertEqual(jc.lib.parse('kv', test_data), expected_output)
|
|
|
|
def test_lib_parser_mod_list_is_list(self):
|
|
self.assertIsInstance(jc.lib.parser_mod_list(), list)
|
|
|
|
def test_lib_parser_mod_list_contains_csv(self):
|
|
self.assertTrue('csv' in jc.lib.parser_mod_list())
|
|
|
|
def test_lib_parser_mod_list_length(self):
|
|
self.assertGreaterEqual(len(jc.lib.parser_mod_list()), 80)
|
|
|
|
def test_lib_parser_info_is_dict(self):
|
|
self.assertIsInstance(jc.lib.parser_info('csv'), dict)
|
|
|
|
def test_lib_parser_info_csv(self):
|
|
self.assertTrue(jc.lib.parser_info('csv')['name'] == 'csv')
|
|
|
|
def test_lib_all_parser_info_is_list_of_dicts(self):
|
|
self.assertIsInstance(jc.lib.all_parser_info(), list)
|
|
self.assertIsInstance(jc.lib.all_parser_info()[0], dict)
|
|
|
|
def test_lib_all_parser_info_contains_csv(self):
|
|
p_list = []
|
|
for p in jc.lib.all_parser_info():
|
|
p_list.append(p['name'])
|
|
self.assertTrue('csv' in p_list)
|
|
|
|
def test_lib_all_parser_info_length(self):
|
|
self.assertGreaterEqual(len(jc.lib.all_parser_info()), 80)
|
|
|
|
def test_lib_all_parser_hidden_length(self):
|
|
reg_length = len(jc.lib.all_parser_info())
|
|
hidden_length = len(jc.lib.all_parser_info(show_hidden=True))
|
|
self.assertGreater(hidden_length, reg_length)
|
|
|
|
def test_lib_plugin_parser_mod_list_is_list(self):
|
|
self.assertIsInstance(jc.lib.plugin_parser_mod_list(), list)
|
|
|
|
def test_lib_plugin_parser_mod_list_length_is_zero(self):
|
|
"""Ensure there are no plugin parsers present during test/build."""
|
|
self.assertEqual(len(jc.lib.plugin_parser_mod_list()), 0)
|
|
|
|
def test_lib_cliname_to_modname(self):
|
|
self.assertEqual(jc.lib._cliname_to_modname('module-name'), 'module_name')
|
|
|
|
def test_lib_argumentname_to_modname(self):
|
|
self.assertEqual(jc.lib._cliname_to_modname('--module-name'), 'module_name')
|
|
|
|
def test_lib_modname_to_cliname(self):
|
|
self.assertEqual(jc.lib._modname_to_cliname('module_name'), 'module-name')
|
|
|
|
def test_lib_all_parser_info_show_deprecated(self):
|
|
# save old state
|
|
old_parsers = deepcopy(jc.lib.parsers)
|
|
old_get_parser = deepcopy(jc.lib._get_parser)
|
|
|
|
# mock data
|
|
class mock_parser_info:
|
|
version = "1.1"
|
|
description = "`deprecated` command parser"
|
|
author = "nobody"
|
|
author_email = "nobody@gmail.com"
|
|
compatible = ["linux", "darwin"]
|
|
magic_commands = ["deprecated"]
|
|
deprecated = True
|
|
|
|
class mock_parser:
|
|
info = mock_parser_info
|
|
|
|
jc.lib.parsers = ['deprecated']
|
|
jc.lib._get_parser = lambda x: mock_parser # type: ignore
|
|
result = jc.lib.all_parser_info(show_deprecated=True)
|
|
|
|
# reset
|
|
jc.lib.parsers = old_parsers
|
|
jc.lib._get_parser = old_get_parser
|
|
|
|
self.assertEqual(len(result), 1)
|
|
|
|
def test_lib_all_parser_info_show_hidden(self):
|
|
# save old state
|
|
old_parsers = deepcopy(jc.lib.parsers)
|
|
old_get_parser = deepcopy(jc.lib._get_parser)
|
|
|
|
# mock data
|
|
class mock_parser_info:
|
|
version = "1.1"
|
|
description = "`deprecated` command parser"
|
|
author = "nobody"
|
|
author_email = "nobody@gmail.com"
|
|
compatible = ["linux", "darwin"]
|
|
magic_commands = ["deprecated"]
|
|
hidden = True
|
|
|
|
class mock_parser:
|
|
info = mock_parser_info
|
|
|
|
jc.lib.parsers = ['deprecated']
|
|
jc.lib._get_parser = lambda x: mock_parser # type: ignore
|
|
result = jc.lib.all_parser_info(show_hidden=True)
|
|
|
|
# reset
|
|
jc.lib.parsers = old_parsers
|
|
jc.lib._get_parser = old_get_parser
|
|
|
|
self.assertEqual(len(result), 1)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |