1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
|
from copy import deepcopy
import unittest
from typing import Generator
from types import ModuleType
import jc.lib
import jc.parsers.csv as csv_parser
class MyTests(unittest.TestCase):
def test_lib_get_parser_string(self):
p = jc.lib.get_parser('arp')
self.assertIsInstance(p, ModuleType)
def test_lib_get_parser_broken_parser(self):
"""get_parser substitutes the disabled_parser if a parser is broken"""
broken = jc.lib.get_parser('broken_parser')
disabled = jc.lib.get_parser('disabled_parser')
self.assertIs(broken, disabled)
def test_lib_get_parser_module(self):
p = jc.lib.get_parser(csv_parser)
self.assertIsInstance(p, ModuleType)
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
def parse():
pass
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
def parse():
pass
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()
|
