def testCollectWithProgramData(self): """Tests the Collect function with the %ProgramData% variable.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() 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( 'allusersappdata') self.assertIsNotNone(environment_variable) self.assertEqual(environment_variable.value, '%SystemDrive%\\ProgramData')
def testCollectWithAllUsersProfile(self): """Tests the Collect function with the %AllUsersProfile% variable.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() 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( 'allusersappdata') self.assertIsNotNone(environment_variable) self.assertEqual( environment_variable.value, 'C:\\Documents and Settings\\All Users\\Application Data')
def testCollect(self): """Tests the Collect function.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() knowledge_base_object = knowledge_base.KnowledgeBase() plugin.Collect(knowledge_base_object) environment_variable = knowledge_base_object.GetEnvironmentVariable( 'allusersappdata') self.assertIsNone(environment_variable)
def testCollect(self): """Tests the Collect function.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() 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( 'allusersappdata') self.assertIsNone(environment_variable)
def testCollectWithProgramData(self): """Tests the Collect function with the %ProgramData% variable.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() 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( 'allusersappdata') self.assertIsNotNone(environment_variable) self.assertEqual(environment_variable.value, '%SystemDrive%\\ProgramData')
def testCollectWithAllUsersProfile(self): """Tests the Collect function with the %AllUsersProfile% variable.""" plugin = windows.WindowsAllUsersAppDataKnowledgeBasePlugin() 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( 'allusersappdata') self.assertIsNotNone(environment_variable) self.assertEqual( environment_variable.value, 'C:\\Documents and Settings\\All Users\\Application Data')