#NEWLINE# def _CheckDriveAndRoot( self ) : #NEWLINE# #INDENT# client_mock = action_mocks . ActionMock ( 'StatFile' , 'ListDirectory' ) #NEWLINE# for _ in test_lib . TestFlowHelper ( 'ArtifactCollectorFlow' , client_mock , artifact_list = [ 'SystemDriveEnvironmentVariable' ] , token = self . token , client_id = self . client_id , output = 'testsystemdrive' ) : #NEWLINE# #INDENT# pass #NEWLINE# #UNINDENT# fd = aff4 . FACTORY . Open ( rdfvalue . RDFURN ( self . client_id ) . Add ( 'testsystemdrive' ) , token = self . token ) #NEWLINE# self . assertEqual ( len ( fd ) , 1 ) #NEWLINE# self . assertEqual ( str ( fd [ 0 ] ) , 'C:' ) #NEWLINE# for _ in test_lib . TestFlowHelper ( 'ArtifactCollectorFlow' , client_mock , artifact_list = [ 'SystemRoot' ] , token = self . token , client_id = self . client_id , output = 'testsystemroot' ) : #NEWLINE# #INDENT# pass #NEWLINE# #UNINDENT# fd = aff4 . FACTORY . Open ( rdfvalue . RDFURN ( self . client_id ) . Add ( 'testsystemroot' ) , token = self . token ) #NEWLINE# self . assertEqual ( len ( fd ) , 1 ) #NEWLINE# self . assertTrue ( ( str ( fd [ 0 ] ) in [ 'C:\\Windows' , 'C:\\WINDOWS' ] ) )