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
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Tests for the CLI argument helper manager."""
import argparse
import unittest
from plaso.cli.helpers import interface
from plaso.cli.helpers import manager
from plaso.lib import errors
from tests.cli.helpers import test_lib
class AnotherTestHelper(interface.ArgumentsHelper):
"""Another test CLI argument helper."""
NAME = 'another_test_helper'
DESCRIPTION = 'Another test helper that does nothing.'
@classmethod
def AddArguments(cls, argument_group):
"""Adds command line arguments to an argument group.
This function takes an argument parser or an argument group object and adds
to it all the command line arguments this helper supports.
Args:
argument_group (argparse._ArgumentGroup|argparse.ArgumentParser):
argparse group.
"""
argument_group.add_argument(
'-c', '--correcto', dest='correcto', action='store_true',
default=False, help='The correcto option.')
@classmethod
def ParseOptions(cls, options, configuration_object):
"""Parse and validate the configuration options.
Args:
options (argparse.Namespace): parser options.
configuration_object (object): object to be configured by the argument
helper.
Raises:
BadConfigOption: when a configuration parameter fails validation.
"""
if not hasattr(options, 'correcto'):
raise errors.BadConfigOption('Correcto not set.')
if not isinstance(getattr(options, 'correcto', None), bool):
raise errors.BadConfigOption('Correcto wrongly formatted.')
class HelperManagerTest(unittest.TestCase):
"""Tests the parsers manager."""
# pylint: disable=protected-access
def testHelperRegistration(self):
"""Tests the RegisterHelper and DeregisterHelper functions."""
number_of_helpers = len(manager.ArgumentHelperManager._helper_classes)
manager.ArgumentHelperManager.RegisterHelper(test_lib.TestHelper)
self.assertEqual(
len(manager.ArgumentHelperManager._helper_classes),
number_of_helpers + 1)
with self.assertRaises(KeyError):
manager.ArgumentHelperManager.RegisterHelper(test_lib.TestHelper)
manager.ArgumentHelperManager.DeregisterHelper(test_lib.TestHelper)
self.assertEqual(
len(manager.ArgumentHelperManager._helper_classes),
number_of_helpers)
def testCommandLineArguments(self):
"""Test the AddCommandLineArguments and function."""
manager.ArgumentHelperManager.RegisterHelpers([
test_lib.TestHelper, AnotherTestHelper])
arg_parser = argparse.ArgumentParser(conflict_handler='resolve')
manager.ArgumentHelperManager.AddCommandLineArguments(arg_parser)
# Assert the parameters have been set.
options = arg_parser.parse_args([])
self.assertTrue(hasattr(options, 'dynamic'))
self.assertTrue(hasattr(options, 'correcto'))
self.assertFalse(hasattr(options, 'foobar'))
# Make the parameters fail validation.
options.correcto = 'sfd'
with self.assertRaises(errors.BadConfigOption):
manager.ArgumentHelperManager.ParseOptions(options, None)
options.correcto = True
options.dynamic = 'now stuff'
manager.ArgumentHelperManager.ParseOptions(options, None)
manager.ArgumentHelperManager.DeregisterHelper(test_lib.TestHelper)
manager.ArgumentHelperManager.DeregisterHelper(AnotherTestHelper)
if __name__ == '__main__':
unittest.main()
|
