LEFT | RIGHT |
1 #!/usr/bin/python | 1 #!/usr/bin/python |
2 # -*- coding: utf-8 -*- | 2 # -*- coding: utf-8 -*- |
3 """Tests for the preprocess plugins manager.""" | 3 """Tests for the preprocess plugins manager.""" |
4 | 4 |
5 import unittest | 5 import unittest |
6 | 6 |
7 from artifacts import reader as artifacts_reader | 7 from artifacts import reader as artifacts_reader |
8 from artifacts import registry as artifacts_registry | 8 from artifacts import registry as artifacts_registry |
9 | 9 |
10 from plaso.engine import knowledge_base | 10 from plaso.engine import knowledge_base |
(...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
44 | 44 |
45 knowledge_base_object = knowledge_base.KnowledgeBase() | 45 knowledge_base_object = knowledge_base.KnowledgeBase() |
46 | 46 |
47 _ = knowledge_base_object | 47 _ = knowledge_base_object |
48 | 48 |
49 # TODO: implement. | 49 # TODO: implement. |
50 # manager.PreprocessPluginsManager.CollectFromFileSystem( | 50 # manager.PreprocessPluginsManager.CollectFromFileSystem( |
51 # registry, knowledge_base_object, None, None) | 51 # registry, knowledge_base_object, None, None) |
52 | 52 |
53 # TODO: add tests for CollectFromWindowsRegistry | 53 # TODO: add tests for CollectFromWindowsRegistry |
| 54 # TODO: add tests for GetNames |
54 | 55 |
55 def testRegistrationPlugin(self): | 56 def testRegistrationPlugin(self): |
56 """Tests RegisterPlugin and DeregisterPlugin functions.""" | 57 """Tests RegisterPlugin and DeregisterPlugin functions.""" |
57 number_of_plugins = len(manager.PreprocessPluginsManager._plugins) | 58 number_of_plugins = len(manager.PreprocessPluginsManager._plugins) |
58 | 59 |
59 manager.PreprocessPluginsManager.RegisterPlugin( | 60 manager.PreprocessPluginsManager.RegisterPlugin( |
60 TestArtifactPreprocessorPlugin) | 61 TestArtifactPreprocessorPlugin) |
61 self.assertEqual( | 62 self.assertEqual( |
62 len(manager.PreprocessPluginsManager._plugins), number_of_plugins + 1) | 63 len(manager.PreprocessPluginsManager._plugins), number_of_plugins + 1) |
63 | 64 |
64 with self.assertRaises(KeyError): | 65 with self.assertRaises(KeyError): |
65 manager.PreprocessPluginsManager.RegisterPlugin( | 66 manager.PreprocessPluginsManager.RegisterPlugin( |
66 TestArtifactPreprocessorPlugin) | 67 TestArtifactPreprocessorPlugin) |
67 | 68 |
68 manager.PreprocessPluginsManager.DeregisterPlugin( | 69 manager.PreprocessPluginsManager.DeregisterPlugin( |
69 TestArtifactPreprocessorPlugin) | 70 TestArtifactPreprocessorPlugin) |
70 self.assertEqual( | 71 self.assertEqual( |
71 len(manager.PreprocessPluginsManager._plugins), number_of_plugins) | 72 len(manager.PreprocessPluginsManager._plugins), number_of_plugins) |
72 | 73 |
73 # TODO: add tests for RegisterPlugins | 74 # TODO: add tests for RegisterPlugins |
74 | 75 |
75 # TODO: add tests for RunPlugins | 76 # TODO: add tests for RunPlugins |
76 | 77 |
77 | 78 |
78 if __name__ == '__main__': | 79 if __name__ == '__main__': |
79 unittest.main() | 80 unittest.main() |
LEFT | RIGHT |