def testCollectWithProgramData(self):
"""Tests the Collect function with the %ProgramData% variable."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
session = sessions.Session()
storage_writer = self._CreateTestStorageWriter()
test_knowledge_base = knowledge_base.KnowledgeBase()
test_mediator = mediator.PreprocessMediator(session, storage_writer,
test_knowledge_base)
environment_variable = artifacts.EnvironmentVariableArtifact(
case_sensitive=False,
name='programdata',
value='%SystemDrive%\\ProgramData')
test_mediator.knowledge_base.AddEnvironmentVariable(
environment_variable)
plugin.Collect(test_mediator)
self.assertEqual(storage_writer.number_of_preprocessing_warnings, 0)
environment_variable = test_mediator.knowledge_base.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNotNone(environment_variable)
self.assertEqual(environment_variable.value,
'%SystemDrive%\\ProgramData')
def testCollectWithAllUsersProfile(self):
"""Tests the Collect function with the %AllUsersProfile% variable."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
session = sessions.Session()
storage_writer = self._CreateTestStorageWriter()
test_knowledge_base = knowledge_base.KnowledgeBase()
test_mediator = mediator.PreprocessMediator(session, storage_writer,
test_knowledge_base)
environment_variable = artifacts.EnvironmentVariableArtifact(
case_sensitive=False,
name='allusersprofile',
value='C:\\Documents and Settings\\All Users')
test_mediator.knowledge_base.AddEnvironmentVariable(
environment_variable)
plugin.Collect(test_mediator)
number_of_warnings = storage_writer.GetNumberOfAttributeContainers(
'preprocessing_warning')
self.assertEqual(number_of_warnings, 0)
environment_variable = test_mediator.knowledge_base.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNotNone(environment_variable)
self.assertEqual(environment_variable.value,
'C:\\Documents and Settings\\All Users')
def testCollect(self):
"""Tests the Collect function."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
knowledge_base_object = knowledge_base.KnowledgeBase()
plugin.Collect(knowledge_base_object)
environment_variable = knowledge_base_object.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNone(environment_variable)
def testCollect(self):
"""Tests the Collect function."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
session = sessions.Session()
storage_writer = self._CreateTestStorageWriter()
test_knowledge_base = knowledge_base.KnowledgeBase()
test_mediator = mediator.PreprocessMediator(session, storage_writer,
test_knowledge_base)
plugin.Collect(test_mediator)
self.assertEqual(storage_writer.number_of_preprocessing_warnings, 0)
environment_variable = test_mediator.knowledge_base.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNone(environment_variable)
def testCollectWithProgramData(self):
"""Tests the Collect function with the %ProgramData% variable."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
knowledge_base_object = knowledge_base.KnowledgeBase()
environment_variable = artifacts.EnvironmentVariableArtifact(
case_sensitive=False, name='programdata',
value='%SystemDrive%\\ProgramData')
knowledge_base_object.AddEnvironmentVariable(environment_variable)
plugin.Collect(knowledge_base_object)
environment_variable = knowledge_base_object.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNotNone(environment_variable)
self.assertEqual(environment_variable.value, '%SystemDrive%\\ProgramData')
def testCollectWithAllUsersProfile(self):
"""Tests the Collect function with the %AllUsersProfile% variable."""
plugin = windows.WindowsAllUsersAppProfileKnowledgeBasePlugin()
knowledge_base_object = knowledge_base.KnowledgeBase()
environment_variable = artifacts.EnvironmentVariableArtifact(
case_sensitive=False, name='allusersprofile',
value='C:\\Documents and Settings\\All Users')
knowledge_base_object.AddEnvironmentVariable(environment_variable)
plugin.Collect(knowledge_base_object)
environment_variable = knowledge_base_object.GetEnvironmentVariable(
'allusersprofile')
self.assertIsNotNone(environment_variable)
self.assertEqual(
environment_variable.value, 'C:\\Documents and Settings\\All Users')
