

def testKnowledgeBaseRootAttributesGetMappedCorrectly(self):
    kb = rdf_client.KnowledgeBase(environ_path='the_path', environ_temp='the_temp', environ_systemroot='the_systemroot', environ_windir='the_windir', environ_programfiles='the_programfiles', environ_programfilesx86='the_programfilesx86', environ_systemdrive='the_systemdrive', environ_allusersprofile='the_allusersprofile', environ_allusersappdata='the_allusersappdata')
    mapping = artifact_utils.GetWindowsEnvironmentVariablesMap(kb)
    self.assertEqual(mapping, {
        'allusersappdata': 'the_allusersappdata',
        'allusersprofile': 'the_allusersprofile',
        'path': 'the_path',
        'programdata': 'the_allusersprofile',
        'programfiles': 'the_programfiles',
        'programfiles(x86)': 'the_programfilesx86',
        'programw6432': 'the_programfiles',
        'systemdrive': 'the_systemdrive',
        'systemroot': 'the_systemroot',
        'temp': 'the_temp',
        'windir': 'the_windir',
    })

dataset/ETHPy150Open google/grr/grr/lib/artifact_utils_test.py GetWindowsEnvironmentVariablesMapTest.testKnowledgeBaseRootAttributesGetMappedCorrectly/original