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
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""Tests for the CLI argument helper manager."""
from __future__ import unicode_literals
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()
|
