def test_load_file_same_profile_name(self):
loader = ProfileLoader()
with patch_stderr() as stderr, patch(printmsg, 'is_color_enabled',
lambda x: True):
loader.load_file(
os.path.join(MOCKED_DATA, 'load-file-same-profile-name.json'))
stderr.seek(0)
self.assertEqual(
'\x1b[93mwarn\x1b[0m: conflicting profiles with the name "test"\n',
stderr.read())
def test_check_arg_patterns_match_not(self):
entries = [
{
ARGS: ['-l'],
ARGPATTERNS: [{
'expression': '-l',
'match_not': True,
}],
RESULT: False
},
{
ARGS: ['-l'],
ARGPATTERNS: [{
'expression': '-a',
'match_not': True,
}],
RESULT: True
},
]
for entry in entries:
args = entry[ARGS]
argpatterns = entry[ARGPATTERNS]
with self.subTest(args=args, argpatterns=argpatterns):
argpats = list(map(lambda x: ArgPattern(x), argpatterns))
self.assertEqual(
entry[RESULT],
ProfileLoader.check_arg_patterns(args, argpats),
)
def test_check_arg_patterns(self):
entries = [
{
ARGS: ['-l'],
ARGPATTERNS: [{
'expression': '-l'
}],
RESULT: True
},
{
ARGS: ['-l'],
ARGPATTERNS: [{
'expression': '-a'
}],
RESULT: False
},
{
ARGS: ['-lA'],
ARGPATTERNS: [{
'expression': '-[A-Za-z]*l[A-Za-z]*'
}],
RESULT: True
},
{
ARGS: ['-l', '-a'],
ARGPATTERNS: [{
'expression': '-l',
'position': 1,
}],
RESULT: True
},
{
ARGS: ['-l', '-a'],
ARGPATTERNS: [{
'expression': '-l',
'position': 2,
}],
RESULT: False
},
{
ARGS: ['-l'],
ARGPATTERNS: [{
'expression': '-l'
}, {
'expression': '-a'
}],
RESULT: False
},
]
for entry in entries:
args = entry[ARGS]
argpatterns = entry[ARGPATTERNS]
with self.subTest(args=args, argpatterns=argpatterns):
argpats = list(map(lambda x: ArgPattern(x), argpatterns))
self.assertEqual(
entry[RESULT],
ProfileLoader.check_arg_patterns(args, argpats))
def test_from_profile_separate_file(self):
loader = ProfileLoader()
loader.load_file(
os.path.join(MOCKED_DATA, 'from-profile-separate-file1.json'))
loader.load_file(
os.path.join(MOCKED_DATA, 'from-profile-separate-file2.json'))
prof = loader.get_profile_by_name('test')
numbers = loader.get_profile_by_name('numbers')
self.assertEqual(numbers.patterns[0]['expression'],
prof.loaded_patterns[0].expression)
def test_get_profile_by_command_which(self):
loader = ProfileLoader()
loader.load_file(
os.path.join(MOCKED_DATA, 'get-profile-by-command-which.json'))
with patch(profileloader, 'which', lambda x: '/usr/bin/date'):
self.assertEqual(loader.get_profile_by_command('date', []),
loader.get_profile_by_name('which'))
self.assertIsNone(loader.get_profile_by_command('noexist', []))
def test_include_from_profile_fail(self):
loader = ProfileLoader()
with self.assertRaises(ConfigPropertyError):
loader.load_file(
os.path.join(MOCKED_DATA, 'include-from-profile-fail.json'))
loader.get_profile_by_name('asdf').load_patterns()
def test_check_arg_patterns_subcommand(self):
entries = [
{
ARGS: ['add', 'entry', '-v'],
ARGPATTERNS: [{
'subcommand': ['add']
}],
RESULT: True
},
{
ARGS: ['add', 'entry', '-v', 'last'],
ARGPATTERNS: [{
'subcommand': ['add', 'entry', 'last']
}],
RESULT: True
},
{
ARGS: ['add', 'entry', '-v'],
ARGPATTERNS: [{
'subcommand': ['add', 'entry', 'last']
}],
RESULT: False
},
{
ARGS: ['add', 'entry', '-v', 'last'],
ARGPATTERNS: [{
'subcommand': ['add', 'entry', 'first']
}],
RESULT: False
},
{
ARGS: ['add', 'entry', '-v', '--', 'last'],
ARGPATTERNS: [{
'subcommand': ['add', 'entry', 'last']
}],
RESULT: False
},
{
ARGS: ['remove', 'entry', '-v'],
ARGPATTERNS: [{
'subcommand': [None, 'entry']
}],
RESULT: True
},
{
ARGS: ['-v', 'add', 'entry'],
ARGPATTERNS: [{
'subcommand': ['add', 'entry']
}],
RESULT: True
},
{
ARGS: ['-v', 'add', 'entry'],
ARGPATTERNS: [{
'subcommand': 'add'
}],
RESULT: True
},
]
for entry in entries:
args = entry[ARGS]
argpatterns = entry[ARGPATTERNS]
with self.subTest(args=args, argpatterns=argpatterns):
argpats = list(map(lambda x: ArgPattern(x), argpatterns))
self.assertEqual(
entry[RESULT],
ProfileLoader.check_arg_patterns(args, argpats),
)