Source Code for Module pike.test.querydirectory

 1  # 
 2  # Copyright (c) 2013, EMC Corporation 
 3  # All rights reserved. 
 4  # 
 5  # Redistribution and use in source and binary forms, with or without 
 6  # modification, are permitted provided that the following conditions are met: 
 7  # 
 8  # 1. Redistributions of source code must retain the above copyright notice, 
 9  # this list of conditions and the following disclaimer. 
10  # 2. Redistributions in binary form must reproduce the above copyright notice, 
11  # this list of conditions and the following disclaimer in the documentation 
12  # and/or other materials provided with the distribution. 
13  # 
14  # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" 
15  # AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE 
16  # IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE 
17  # ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE 
18  # LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 
19  # CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 
20  # SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 
21  # INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 
22  # CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 
23  # ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE 
24  # POSSIBILITY OF SUCH DAMAGE. 
25  # 
26  # Module Name: 
27  # 
28  #        querydirectory.py 
29  # 
30  # Abstract: 
31  # 
32  #        Query directory tests 
33  # 
34  # Authors: Brian Koropoff (brian.koropoff@emc.com) 
35  # 
36   
37  import pike.model 
38  import pike.smb2 
39  import pike.test 
40  import pike.ntstatus 
41   
42 -class QueryDirectoryTest(pike.test.PikeTest): 
43      # Enumerate directory at FILE_DIRECTORY_INFORMATION level. 
44      # Test for presence of . and .. entries 
45 -    def test_file_directory_info(self): 
46          chan, tree = self.tree_connect() 
47           
48          root = chan.create(tree, '', access=pike.smb2.GENERIC_READ, options=pike.smb2.FILE_DIRECTORY_FILE, share=pike.smb2.FILE_SHARE_READ).result() 
49          names = map(lambda info: info.file_name, chan.enum_directory(root)) 
50   
51          self.assertIn('.', names) 
52   
53          chan.close(root) 
54   
55      # Querying for a specific filename twice 
56      # on the same handle succeeds the first time and 
57      # fails with STATUS_NO_MORE_FILES the second. 
58 -    def test_specific_name(self): 
59          chan, tree = self.tree_connect() 
60          name = 'hello.txt' 
61           
62          hello = chan.create(tree, 
63                              name, 
64                              access=pike.smb2.GENERIC_WRITE | pike.smb2.GENERIC_READ | pike.smb2.DELETE, 
65                              disposition=pike.smb2.FILE_SUPERSEDE, 
66                              options=pike.smb2.FILE_DELETE_ON_CLOSE).result() 
67   
68          root = chan.create(tree, 
69                             '', 
70                             access=pike.smb2.GENERIC_READ, 
71                             options=pike.smb2.FILE_DIRECTORY_FILE, 
72                             share=pike.smb2.FILE_SHARE_READ).result() 
73   
74          query1 = chan.query_directory(root, file_name=name) 
75   
76          with self.assert_error(pike.ntstatus.STATUS_NO_MORE_FILES): 
77              chan.query_directory(root, file_name=name) 
78   
79          chan.close(hello) 
80          chan.close(root) 
81